Shutdown gracefully if the terminal was closed(sighup)

ssvlabs · Aug 29, 2024 · 3d20f2a · 3d20f2a
1 parent 0b17972
commit 3d20f2a
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/internal/platform/lifecycle/lifecycle.go b/internal/platform/lifecycle/lifecycle.go
@@ -12,16 +12,13 @@ import (
 const terminationDelay = time.Second * 5
 
 func ListenForApplicationShutDown(ctx context.Context, shutdownFunc func(), signalChannel chan os.Signal) {
-	signal.Notify(signalChannel, os.Interrupt, syscall.SIGTERM)
+	signal.Notify(signalChannel, os.Interrupt, syscall.SIGTERM, syscall.SIGHUP, syscall.SIGINT)
 
 	select {
 	case sig := <-signalChannel:
-		switch sig {
-		case os.Interrupt, syscall.SIGTERM:
-			slog.Warn("shutdown signal received")
-			shutdownFunc()
-			time.Sleep(terminationDelay)
-		}
+		slog.With("sig", sig.String()).Warn("shutdown signal received")
+		shutdownFunc()
+		time.Sleep(terminationDelay)
 	case <-ctx.Done():
 		slog.Warn("context deadline exceeded or canceled")
 		shutdownFunc()