add SIGTERM/SIGINT handler for gracefull shutdown

go-graphite · Sep 14, 2021 · 5691753 · 5691753
1 parent aef60a0
commit 5691753
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/carbon-clickhouse.go b/carbon-clickhouse.go
@@ -137,7 +137,7 @@ func main() {
 
 	go func() {
 		c := make(chan os.Signal, 1)
-		signal.Notify(c, syscall.SIGUSR1, syscall.SIGUSR2, syscall.SIGHUP)
+		signal.Notify(c, syscall.SIGUSR1, syscall.SIGUSR2, syscall.SIGHUP, syscall.SIGTERM, syscall.SIGINT)
 
 		for {
 			s := <-c
@@ -149,6 +149,9 @@ func main() {
 				mainLogger.Info("SIGUSR2 received. Ignoring")
 			case syscall.SIGHUP:
 				mainLogger.Info("SIGHUP received. Ignoring")
+			case syscall.SIGTERM, syscall.SIGINT:
+				mainLogger.Info("shutting down")
+				app.Stop()
 			}
 		}
 	}()