diff --git a/cmd/main.go b/cmd/main.go index ff55d16..6f2a74b 100644 --- a/cmd/main.go +++ b/cmd/main.go @@ -25,7 +25,7 @@ func main() { signalChannel := make(chan os.Signal, 1) signal.Notify(signalChannel, syscall.SIGINT) signal.Notify(signalChannel, syscall.SIGTERM) - defer stop(signalChannel) + defer stop(signalChannel, s) // Запуск сервера go start(config, s, srv) @@ -42,10 +42,11 @@ func main() { } // Остановка сервера -func stop(signalChannel chan os.Signal) { - log.Println("authDBServer stopping ...") +func stop(signalChannel chan os.Signal, s *grpc.Server) { + defer s.GracefulStop() + defer signal.Stop(signalChannel) - signal.Stop(signalChannel) + log.Println("authDBServer stopping ...") } // Запуск сервера