initialize sqlTextCounts where missed

Signed-off-by: Harshit Gangal <[email protected]>
timvaillancourt · May 14, 2024 · 1b17de6 · 1b17de6
1 parent e8ba229
commit 1b17de6
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 10 deletions.
diff --git a/go/vt/vtgate/plugin_mysql_server_test.go b/go/vt/vtgate/plugin_mysql_server_test.go
@@ -349,7 +349,7 @@ func TestKillMethods(t *testing.T) {
 func TestGracefulShutdown(t *testing.T) {
 	executor, _, _, _, _ := createExecutorEnv(t)
 
-	vh := newVtgateHandler(&VTGate{executor: executor, timings: timings, rowsReturned: rowsReturned, rowsAffected: rowsAffected})
+	vh := newVtgateHandler(&VTGate{executor: executor, timings: timings, rowsReturned: rowsReturned, rowsAffected: rowsAffected, sqlTextCounts: sqlTextCounts})
 	th := &testHandler{}
 	listener, err := mysql.NewListener("tcp", "127.0.0.1:", mysql.NewAuthServerNone(), th, 0, 0, false, false, 0, 0)
 	require.NoError(t, err)
@@ -379,7 +379,7 @@ func TestGracefulShutdown(t *testing.T) {
 func TestGracefulShutdownWithTransaction(t *testing.T) {
 	executor, _, _, _, _ := createExecutorEnv(t)
 
-	vh := newVtgateHandler(&VTGate{executor: executor, timings: timings, rowsReturned: rowsReturned, rowsAffected: rowsAffected})
+	vh := newVtgateHandler(&VTGate{executor: executor, timings: timings, rowsReturned: rowsReturned, rowsAffected: rowsAffected, sqlTextCounts: sqlTextCounts})
 	th := &testHandler{}
 	listener, err := mysql.NewListener("tcp", "127.0.0.1:", mysql.NewAuthServerNone(), th, 0, 0, false, false, 0, 0)
 	require.NoError(t, err)

diff --git a/go/vt/vtgate/vtgate.go b/go/vt/vtgate/vtgate.go
@@ -676,14 +676,15 @@ func (vtg *VTGate) HandlePanic(err *error) {
 
 func newVTGate(executor *Executor, resolver *Resolver, vsm *vstreamManager, tc *TxConn, gw *TabletGateway) *VTGate {
 	return &VTGate{
-		executor:     executor,
-		resolver:     resolver,
-		vsm:          vsm,
-		txConn:       tc,
-		gw:           gw,
-		timings:      timings,
-		rowsReturned: rowsReturned,
-		rowsAffected: rowsAffected,
+		executor:      executor,
+		resolver:      resolver,
+		vsm:           vsm,
+		txConn:        tc,
+		gw:            gw,
+		timings:       timings,
+		rowsReturned:  rowsReturned,
+		rowsAffected:  rowsAffected,
+		sqlTextCounts: sqlTextCounts,
 
 		logExecute:       logutil.NewThrottledLogger("Execute", 5*time.Second),
 		logPrepare:       logutil.NewThrottledLogger("Prepare", 5*time.Second),