Add graceful shutdown.

oliver006 · Jan 5, 2024 · d7c7511 · d7c7511
1 parent 769064a
commit d7c7511
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/main.go b/main.go
@@ -229,11 +229,7 @@ func main() {
 
 	s := make(chan os.Signal, 1)
 	signal.Notify(s, syscall.SIGINT, syscall.SIGTERM)
-	go func() {
-		select {
-		case _s := <-s:
-			log.Infof("Receive exit signal, bye!    signal: %s", _s.String())
-			os.Exit(0)
-		}
-	}()
+	_sig := <-s
+	log.Infof("Receive exit signal, bye! Signal: %s", _sig.String())
+	os.Exit(0)
 }