Apply suggestions from code review

internal/query/execute_query.go

@@ -24,7 +24,7 @@ import (
 type executeSettings interface {
 	ExecMode() options.ExecMode
 	StatsMode() options.StatsMode
-	StatsCallback() func(stats *stats.QueryStats)
+	StatsCallback() func(stats stats.QueryStats)
 	TxControl() *query.TransactionControl
 	Syntax() options.Syntax
 	Params() params.Parameters

internal/query/options/execute.go

@@ -32,7 +32,7 @@ type (
 		execMode               ExecMode
 		statsMode              StatsMode
 		resourcePool           string
-		statsCallback          func(queryStats *stats.QueryStats)
+		statsCallback          func(queryStats stats.QueryStats)
 		callOptions            []grpc.CallOption
 		txControl              *tx.Control
 		retryOptions           []retry.Option
@@ -59,7 +59,7 @@ type (
 	syntaxOption    = Syntax
 	statsModeOption struct {
 		mode     StatsMode
-		callback func(*stats.QueryStats)
+		callback func(stats.QueryStats)
 	}
 	execModeOption         = ExecMode
 	responsePartLimitBytes int64
@@ -73,7 +73,7 @@ func (s *executeSettings) RetryOpts() []retry.Option {
 	return s.retryOptions
 }
 
-func (s *executeSettings) StatsCallback() func(*stats.QueryStats) {
+func (s *executeSettings) StatsCallback() func(stats.QueryStats) {
 	return s.statsCallback
 }
 
@@ -225,7 +225,7 @@ func (opt statsModeOption) applyExecuteOption(s *executeSettings) {
 	s.statsCallback = opt.callback
 }
 
-func WithStatsMode(mode StatsMode, callback func(*stats.QueryStats)) statsModeOption {
+func WithStatsMode(mode StatsMode, callback func(stats.QueryStats)) statsModeOption {
 	return statsModeOption{
 		mode:     mode,
 		callback: callback,

internal/query/options/execute_script.go

@@ -34,7 +34,7 @@ type (
 			Query  string
 		}
 		Mode           ExecMode
-		Stats          *stats.QueryStats
+		Stats          stats.QueryStats
 		ResultSetsMeta []struct {
 			Columns []struct {
 				Name string

internal/query/result.go

@@ -38,7 +38,7 @@ type (
 		resultSetIndex int64
 		closed         chan struct{}
 		trace          *trace.Query
-		statsCallback  func(queryStats *stats.QueryStats)
+		statsCallback  func(queryStats stats.QueryStats)
 		onClose        []func()
 		onNextPartErr  []func(err error)
 		onTxMeta       []func(txMeta *Ydb_Query.TransactionMeta)
@@ -93,7 +93,7 @@ func withTrace(t *trace.Query) resultOption {
 	}
 }
 
-func withStatsCallback(callback func(queryStats *stats.QueryStats)) resultOption {
+func withStatsCallback(callback func(queryStats stats.QueryStats)) resultOption {
 	return func(s *streamResult) {
 		s.statsCallback = callback
 	}

internal/query/result_test.go

@@ -1926,8 +1926,8 @@ func TestResultStats(t *testing.T) {
 				},
 			}, nil)
 			stream.EXPECT().Recv().Return(nil, io.EOF)
-			var s *stats.QueryStats
-			result, err := newResult(ctx, stream, withStatsCallback(func(queryStats *stats.QueryStats) {
+			var s stats.QueryStats
+			result, err := newResult(ctx, stream, withStatsCallback(func(queryStats stats.QueryStats) {
 				s = queryStats
 			}))
 			require.NoError(t, err)
@@ -2287,8 +2287,8 @@ func TestResultStats(t *testing.T) {
 				},
 			}, nil)
 			stream.EXPECT().Recv().Return(nil, io.EOF)
-			var s *stats.QueryStats
-			result, err := newResult(ctx, stream, withStatsCallback(func(queryStats *stats.QueryStats) {
+			var s stats.QueryStats
+			result, err := newResult(ctx, stream, withStatsCallback(func(queryStats stats.QueryStats) {
 				s = queryStats
 			}))
 			require.NoError(t, err)
@@ -2649,8 +2649,8 @@ func TestResultStats(t *testing.T) {
 				},
 			}, nil)
 			stream.EXPECT().Recv().Return(nil, io.EOF)
-			var s *stats.QueryStats
-			result, err := newResult(ctx, stream, withStatsCallback(func(queryStats *stats.QueryStats) {
+			var s stats.QueryStats
+			result, err := newResult(ctx, stream, withStatsCallback(func(queryStats stats.QueryStats) {
 				s = queryStats
 			}))
 			require.NoError(t, err)
@@ -2986,8 +2986,8 @@ func TestResultStats(t *testing.T) {
 				},
 			}, nil)
 			stream.EXPECT().Recv().Return(nil, io.EOF)
-			var s *stats.QueryStats
-			result, err := newResult(ctx, stream, withStatsCallback(func(queryStats *stats.QueryStats) {
+			var s stats.QueryStats
+			result, err := newResult(ctx, stream, withStatsCallback(func(queryStats stats.QueryStats) {
 				s = queryStats
 			}))
 			require.NoError(t, err)
@@ -3358,8 +3358,8 @@ func TestMaterializedResultStats(t *testing.T) {
 				},
 			}, nil)
 			stream.EXPECT().Recv().Return(nil, io.EOF)
-			var s *stats.QueryStats
-			result, err := newResult(ctx, stream, withStatsCallback(func(queryStats *stats.QueryStats) {
+			var s stats.QueryStats
+			result, err := newResult(ctx, stream, withStatsCallback(func(queryStats stats.QueryStats) {
 				s = queryStats
 			}))
 			require.NoError(t, err)
@@ -3719,8 +3719,8 @@ func TestMaterializedResultStats(t *testing.T) {
 				},
 			}, nil)
 			stream.EXPECT().Recv().Return(nil, io.EOF)
-			var s *stats.QueryStats
-			result, err := newResult(ctx, stream, withStatsCallback(func(queryStats *stats.QueryStats) {
+			var s stats.QueryStats
+			result, err := newResult(ctx, stream, withStatsCallback(func(queryStats stats.QueryStats) {
 				s = queryStats
 			}))
 			require.NoError(t, err)
@@ -4081,8 +4081,8 @@ func TestMaterializedResultStats(t *testing.T) {
 				},
 			}, nil)
 			stream.EXPECT().Recv().Return(nil, io.EOF)
-			var s *stats.QueryStats
-			result, err := newResult(ctx, stream, withStatsCallback(func(queryStats *stats.QueryStats) {
+			var s stats.QueryStats
+			result, err := newResult(ctx, stream, withStatsCallback(func(queryStats stats.QueryStats) {
 				s = queryStats
 			}))
 			require.NoError(t, err)
@@ -4418,8 +4418,8 @@ func TestMaterializedResultStats(t *testing.T) {
 				},
 			}, nil)
 			stream.EXPECT().Recv().Return(nil, io.EOF)
-			var s *stats.QueryStats
-			result, err := newResult(ctx, stream, withStatsCallback(func(queryStats *stats.QueryStats) {
+			var s stats.QueryStats
+			result, err := newResult(ctx, stream, withStatsCallback(func(queryStats stats.QueryStats) {
 				s = queryStats
 			}))
 			require.NoError(t, err)

internal/table/scanner/result.go

@@ -220,7 +220,7 @@ func (r *baseResult) CurrentResultSet() result.Set {
 }
 
 // Stats returns query execution queryStats.
-func (r *baseResult) Stats() *stats.QueryStats {
+func (r *baseResult) Stats() stats.QueryStats {
 	r.statsMtx.RLock()
 	defer r.statsMtx.RUnlock()
 

internal/xsql/propose/conn.go

@@ -82,7 +82,7 @@ func (c *Conn) Explain(ctx context.Context, sql string, _ *params.Params) (ast s
 	_, err := c.session.Query(
 		ctx, sql,
 		options.WithExecMode(options.ExecModeExplain),
-		options.WithStatsMode(options.StatsModeNone, func(stats *stats.QueryStats) {
+		options.WithStatsMode(options.StatsModeNone, func(stats stats.QueryStats) {
 			ast = stats.QueryAST()
 			plan = stats.QueryPlan()
 		}),

query/example_test.go

@@ -246,7 +246,7 @@ func Example_resultStats() {
 		id    int32  // required value
 		myStr string // required value
 	)
-	var s query.Stats
+	var stats query.Stats
 	// Do retry operation on errors with best effort
 	row, err := db.Query().QueryRow(ctx, // context manage exiting from Do
 		`SELECT CAST($id AS Uint64) AS id, CAST($myStr AS Text) AS myStr`,
@@ -256,8 +256,8 @@ func Example_resultStats() {
 				Param("$myStr").Text("123").
 				Build(),
 		),
-		query.WithStatsMode(query.StatsModeFull, func(stats query.Stats) {
-			s = stats
+		query.WithStatsMode(query.StatsModeFull, func(s query.Stats) {
+			stats = s
 		}),
 		query.WithIdempotent(),
 	)
@@ -275,14 +275,14 @@ func Example_resultStats() {
 
 	fmt.Printf("id=%v, myStr='%s'\n", id, myStr)
 	fmt.Println("Stats:")
-	fmt.Printf("- Compilation='%v'\n", s.Compilation())
-	fmt.Printf("- TotalCPUTime='%v'\n", s.TotalCPUTime())
-	fmt.Printf("- ProcessCPUTime='%v'\n", s.ProcessCPUTime())
-	fmt.Printf("- QueryAST='%v'\n", s.QueryAST())
-	fmt.Printf("- QueryPlan='%v'\n", s.QueryPlan())
+	fmt.Printf("- Compilation='%v'\n", stats.Compilation())
+	fmt.Printf("- TotalCPUTime='%v'\n", stats.TotalCPUTime())
+	fmt.Printf("- ProcessCPUTime='%v'\n", stats.ProcessCPUTime())
+	fmt.Printf("- QueryAST='%v'\n", stats.QueryAST())
+	fmt.Printf("- QueryPlan='%v'\n", stats.QueryPlan())
 	fmt.Println("- Phases:")
 	for {
-		phase, ok := s.NextPhase()
+		phase, ok := stats.NextPhase()
 		if !ok {
 			break
 		}