Fix bug in graceful shutdown

medibloc · Oct 6, 2019 · f6280e7 · f6280e7
1 parent 83d30bf
commit f6280e7
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/server/util.go b/server/util.go
@@ -213,14 +213,17 @@ func TrapSignal(cleanupFunc func()) {
 	signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
 	go func() {
 		sig := <-sigs
+		if cleanupFunc != nil {
+			cleanupFunc()
+		}
+		exitCode := 128
 		switch sig {
-		case syscall.SIGTERM:
-			defer cleanupFunc()
-			os.Exit(128 + int(syscall.SIGTERM))
 		case syscall.SIGINT:
-			defer cleanupFunc()
-			os.Exit(128 + int(syscall.SIGINT))
+			exitCode += int(syscall.SIGINT)
+		case syscall.SIGTERM:
+			exitCode += int(syscall.SIGINT)
 		}
+		os.Exit(exitCode)
 	}()
 }