Add terminating state

Signed-off-by: David Piegza <[email protected]>

go/vt/vtgate/plugin_mysql_server.go

@@ -183,6 +183,11 @@ func startSpan(ctx context.Context, query, label string) (trace.Span, context.Co
 
 func (vh *vtgateHandler) ComQuery(c *mysql.Conn, query string, callback func(*sqltypes.Result) error) error {
 	ctx := context.Background()
+
+	if terminating {
+		return vterrors.New(vtrpcpb.Code_ABORTED, "Terminating vtgate")
+	}
+
 	var cancel context.CancelFunc
 	if *mysqlQueryTimeout != 0 {
 		ctx, cancel = context.WithTimeout(ctx, *mysqlQueryTimeout)

go/vt/vtgate/vtgate.go

@@ -83,6 +83,8 @@ var (
 	// healthCheckTimeout is the timeout on the RPC call to tablets
 	healthCheckTimeout = time.Minute
 
+	terminating bool
+
 	// System settings related flags
 	sysVarSetEnabled = true
 	setVarEnabled    = true
@@ -328,6 +330,8 @@ func Init(
 		}
 	})
 	servenv.OnTerm(func() {
+		terminating = true
+
 		if st != nil && enableSchemaChangeSignal {
 			st.Stop()
 		}
@@ -405,6 +409,10 @@ func (vtg *VTGate) registerDebugHealthHandler() {
 // IsHealthy returns nil if server is healthy.
 // Otherwise, it returns an error indicating the reason.
 func (vtg *VTGate) IsHealthy() error {
+	if terminating {
+		return errors.New("Terminating")
+	}
+
 	return nil
 }