chore: shutdown server with timeout context

xinpianchang · Dec 27, 2021 · c36a456 · c36a456
1 parent 7075928
commit c36a456
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/core/xservice/server.go b/core/xservice/server.go
@@ -151,7 +151,9 @@ func (t *serverImpl) Serve() error {
 	t.registerGrpcServiceEtcd()
 
 	signalx.AddShutdownHook(func(os.Signal) {
-		_ = server.Shutdown(context.Background())
+		ctx, cancel := context.WithTimeout(context.Background(), time.Minute)
+		defer cancel()
+		_ = server.Shutdown(ctx)
 		t.grpc.GracefulStop()
 		sentry.Flush(time.Second * 2)
 		log.Info("shutdown", zap.Int("pid", os.Getpid()))