fix: graceful shutdown

w-h-a · Jul 4, 2024 · b8f432b · b8f432b
1 parent 8edefb3
commit b8f432b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/server/grpcserver/grpc_server.go b/server/grpcserver/grpc_server.go
@@ -151,9 +151,12 @@ func (s *grpcServer) start() error {
 			close(exit)
 		}()
 
+		countdown := time.NewTimer(10 * time.Second)
+
 		select {
 		case <-exit:
-		case <-time.After(time.Minute):
+			countdown.Stop()
+		case <-countdown.C:
 			s.server.Stop()
 		}