Fix gofmt formatting in telemetry files

Co-authored-by: samikshya-chand_data

Author: samikshya-db

telemetry/config.go

@@ -42,8 +42,8 @@ type Config struct {
 // Note: Telemetry is disabled by default and requires explicit opt-in.
 func DefaultConfig() *Config {
 	return &Config{
-		Enabled:         false, // Disabled by default, requires explicit opt-in
-		EnableTelemetry: false,
+		Enabled:                 false, // Disabled by default, requires explicit opt-in
+		EnableTelemetry:         false,
 		BatchSize:               100,
 		FlushInterval:           5 * time.Second,
 		MaxRetries:              3,
@@ -121,4 +121,3 @@ func isTelemetryEnabled(ctx context.Context, cfg *Config, host string, driverVer
 
 	return serverEnabled
 }
-

telemetry/config_test.go

@@ -200,7 +200,7 @@ func TestIsTelemetryEnabled_ExplicitOptOut(t *testing.T) {
 	defer server.Close()
 
 	cfg := &Config{
-		EnableTelemetry:      false, // Priority 2: Explicit opt-out
+		EnableTelemetry: false, // Priority 2: Explicit opt-out
 	}
 
 	ctx := context.Background()
@@ -223,7 +223,7 @@ func TestIsTelemetryEnabled_UserOptInServerEnabled(t *testing.T) {
 	defer server.Close()
 
 	cfg := &Config{
-		EnableTelemetry:      true, // User wants telemetry
+		EnableTelemetry: true, // User wants telemetry
 	}
 
 	ctx := context.Background()
@@ -251,7 +251,7 @@ func TestIsTelemetryEnabled_UserOptInServerDisabled(t *testing.T) {
 	defer server.Close()
 
 	cfg := &Config{
-		EnableTelemetry:      true, // User wants telemetry
+		EnableTelemetry: true, // User wants telemetry
 	}
 
 	ctx := context.Background()
@@ -279,7 +279,7 @@ func TestIsTelemetryEnabled_ServerFlagOnly(t *testing.T) {
 	defer server.Close()
 
 	cfg := &Config{
-		EnableTelemetry:      false, // Default: no explicit user preference
+		EnableTelemetry: false, // Default: no explicit user preference
 	}
 
 	ctx := context.Background()
@@ -321,7 +321,7 @@ func TestIsTelemetryEnabled_ServerError(t *testing.T) {
 	defer server.Close()
 
 	cfg := &Config{
-		EnableTelemetry:      true, // User wants telemetry
+		EnableTelemetry: true, // User wants telemetry
 	}
 
 	ctx := context.Background()
@@ -343,7 +343,7 @@ func TestIsTelemetryEnabled_ServerError(t *testing.T) {
 // TestIsTelemetryEnabled_ServerUnreachable tests unreachable server
 func TestIsTelemetryEnabled_ServerUnreachable(t *testing.T) {
 	cfg := &Config{
-		EnableTelemetry:      true, // User wants telemetry
+		EnableTelemetry: true, // User wants telemetry
 	}
 
 	ctx := context.Background()

telemetry/exporter.go

@@ -20,23 +20,23 @@ const (
 
 // telemetryExporter exports metrics to Databricks telemetry service.
 type telemetryExporter struct {
-	host          string
-	driverVersion string
-	httpClient    *http.Client
+	host           string
+	driverVersion  string
+	httpClient     *http.Client
 	circuitBreaker *circuitBreaker
-	cfg           *Config
+	cfg            *Config
 }
 
 // telemetryMetric represents a metric to export.
 type telemetryMetric struct {
-	metricType      string
-	timestamp       time.Time
-	workspaceID     string
-	sessionID       string
-	statementID     string
-	latencyMs       int64
-	errorType       string
-	tags            map[string]interface{}
+	metricType  string
+	timestamp   time.Time
+	workspaceID string
+	sessionID   string
+	statementID string
+	latencyMs   int64
+	errorType   string
+	tags        map[string]interface{}
 }
 
 // telemetryPayload is the JSON structure sent to Databricks.

telemetry/integration_test.go

@@ -151,10 +151,10 @@ func TestIntegration_CircuitBreakerOpening(t *testing.T) {
 func TestIntegration_OptInPriority_ExplicitOptOut(t *testing.T) {
 	cfg := &Config{
 		EnableTelemetry: false, // Priority 1 (client): Explicit opt-out
-		BatchSize:            100,
-		FlushInterval:        5 * time.Second,
-		MaxRetries:           3,
-		RetryDelay:           100 * time.Millisecond,
+		BatchSize:       100,
+		FlushInterval:   5 * time.Second,
+		MaxRetries:      3,
+		RetryDelay:      100 * time.Millisecond,
 	}
 
 	httpClient := &http.Client{Timeout: 5 * time.Second}

telemetry/interceptor.go

@@ -188,11 +188,11 @@ func (i *Interceptor) RecordOperation(ctx context.Context, sessionID string, ope
 	}()
 
 	metric := &telemetryMetric{
-		metricType:  "operation",
-		timestamp:   time.Now(),
-		sessionID:   sessionID,
-		latencyMs:   latencyMs,
-		tags:        map[string]interface{}{"operation_type": operationType},
+		metricType: "operation",
+		timestamp:  time.Now(),
+		sessionID:  sessionID,
+		latencyMs:  latencyMs,
+		tags:       map[string]interface{}{"operation_type": operationType},
 	}
 
 	i.aggregator.recordMetric(ctx, metric)

telemetry/request.go

@@ -14,10 +14,10 @@ type TelemetryRequest struct {
 
 // TelemetryFrontendLog represents a single telemetry log entry.
 type TelemetryFrontendLog struct {
-	WorkspaceID         int64               `json:"workspace_id,omitempty"`
-	FrontendLogEventID  string              `json:"frontend_log_event_id,omitempty"`
-	Context             *FrontendLogContext `json:"context,omitempty"`
-	Entry               *FrontendLogEntry   `json:"entry,omitempty"`
+	WorkspaceID        int64               `json:"workspace_id,omitempty"`
+	FrontendLogEventID string              `json:"frontend_log_event_id,omitempty"`
+	Context            *FrontendLogContext `json:"context,omitempty"`
+	Entry              *FrontendLogEntry   `json:"entry,omitempty"`
 }
 
 // FrontendLogContext contains the client context.
@@ -72,19 +72,19 @@ type DriverConnectionParameters struct {
 
 // SQLExecutionEvent contains SQL execution details.
 type SQLExecutionEvent struct {
-	ResultFormat      string           `json:"result_format,omitempty"`
-	ChunkCount        int              `json:"chunk_count,omitempty"`
-	BytesDownloaded   int64            `json:"bytes_downloaded,omitempty"`
-	PollCount         int              `json:"poll_count,omitempty"`
-	OperationDetail   *OperationDetail `json:"operation_detail,omitempty"`
+	ResultFormat    string           `json:"result_format,omitempty"`
+	ChunkCount      int              `json:"chunk_count,omitempty"`
+	BytesDownloaded int64            `json:"bytes_downloaded,omitempty"`
+	PollCount       int              `json:"poll_count,omitempty"`
+	OperationDetail *OperationDetail `json:"operation_detail,omitempty"`
 }
 
 // OperationDetail contains operation-specific details.
 type OperationDetail struct {
-	OperationType              string `json:"operation_type,omitempty"`
-	NOperationStatusCalls      int64  `json:"n_operation_status_calls,omitempty"`
-	OperationStatusLatencyMs   int64  `json:"operation_status_latency_millis,omitempty"`
-	IsInternalCall             bool   `json:"is_internal_call,omitempty"`
+	OperationType            string `json:"operation_type,omitempty"`
+	NOperationStatusCalls    int64  `json:"n_operation_status_calls,omitempty"`
+	OperationStatusLatencyMs int64  `json:"operation_status_latency_millis,omitempty"`
+	IsInternalCall           bool   `json:"is_internal_call,omitempty"`
 }
 
 // DriverErrorInfo contains error information.