Update termination handling

main.go

@@ -15,6 +15,7 @@ import (
 	"path"
 	"runtime"
 	"strings"
+	"syscall"
 	"time"
 
 	"github.com/hashicorp/go-hclog"
@@ -284,7 +285,7 @@ func main() {
 		var err error
 		err = httpServer.Serve(httpListener)
 
-		if err != nil && err != cmux.ErrServerClosed {
+		if err != nil && err != http.ErrServerClosed {
 			logger.Log().Error("Error starting http server", "error", err)
 			os.Exit(1)
 		}
@@ -306,8 +307,7 @@ func main() {
 
 	// trap sigterm or interupt and gracefully shutdown the server
 	c := make(chan os.Signal, 1)
-	signal.Notify(c, os.Interrupt)
-	signal.Notify(c, os.Kill)
+	signal.Notify(c, os.Interrupt, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT)
 
 	// Block until a signal is received.
 	sig := <-c