Exit with code 1 when gotten second signal

chapsuk · Dec 12, 2018 · 5451e57 · 5451e57
1 parent ea0e779
commit 5451e57
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/shutdown.go b/shutdown.go
@@ -8,7 +8,8 @@ import (
 )
 
 // ShutdownContext returns child context from passed context which will be canceled
-// on incoming signals: SIGINT, SIGTERM, SIGHUP
+// on incoming signals: SIGINT, SIGTERM, SIGHUP.
+// Ends immediately by os.Exit(1) after second signal
 func ShutdownContext(c context.Context) context.Context {
 	ctx, cancel := context.WithCancel(c)
 	go func() {
@@ -19,6 +20,8 @@ func ShutdownContext(c context.Context) context.Context {
 			return
 		case <-ch:
 			cancel()
+			<-ch
+			os.Exit(1)
 		}
 	}()
 	return ctx