Add nil check to prevent panics (#14902)

Signed-off-by: Manan Gupta <[email protected]>
vitessio · Jan 9, 2024 · 12e75b1 · 12e75b1
1 parent 9e3b332
commit 12e75b1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/go/vt/vtgate/vtgate.go b/go/vt/vtgate/vtgate.go
@@ -364,8 +364,10 @@ func Init(
 			st.Start()
 		}
 		srv := initMySQLProtocol(vtgateInst)
-		servenv.OnTermSync(srv.shutdownMysqlProtocolAndDrain)
-		servenv.OnClose(srv.rollbackAtShutdown)
+		if srv != nil {
+			servenv.OnTermSync(srv.shutdownMysqlProtocolAndDrain)
+			servenv.OnClose(srv.rollbackAtShutdown)
+		}
 	})
 	servenv.OnTerm(func() {
 		if st != nil && enableSchemaChangeSignal {