fix: shutdown gRPC server gracefully (#4626)

fix: add graceful shutdown for gRPC server
This commit is contained in:
RichardYe
2025-04-16 06:58:25 +08:00
committed by GitHub
parent 789a9cc5a2
commit ebccef6033

View File

@@ -132,6 +132,9 @@ func (s *Server) Shutdown(ctx context.Context) {
slog.Error("failed to shutdown server", slog.String("error", err.Error())) slog.Error("failed to shutdown server", slog.String("error", err.Error()))
} }
// Shutdown gRPC server.
s.grpcServer.GracefulStop()
// Close database connection. // Close database connection.
if err := s.Store.Close(); err != nil { if err := s.Store.Close(); err != nil {
slog.Error("failed to close database", slog.String("error", err.Error())) slog.Error("failed to close database", slog.String("error", err.Error()))