added docker shutdown logic

Author: CodeShellDev

main.go

@@ -7,6 +7,7 @@ import (
 
 	proxy "github.com/codeshelldev/secured-signal-api/internals/proxy"
 	middlewares "github.com/codeshelldev/secured-signal-api/internals/proxy/middlewares"
+	docker "github.com/codeshelldev/secured-signal-api/utils/docker"
 	env "github.com/codeshelldev/secured-signal-api/utils/env"
 	log "github.com/codeshelldev/secured-signal-api/utils/logger"
 )
@@ -56,5 +57,20 @@ func main() {
 
 	log.Info("Server Listening on ", addr)
 
-	http.ListenAndServe(addr, log_m0.Use())
-}
+	server := &http.Server{
+		Addr:    addr,
+		Handler: log_m0.Use(),
+	}
+
+	stop := docker.Run(func(){
+		err := server.ListenAndServe()
+		
+		if err != nil && err != http.ErrServerClosed {
+			log.Fatal("Server error: ", err.Error())
+		}
+	})
+
+	<-stop
+
+	docker.Shutdown(server)
+}

utils/docker/docker.go

@@ -0,0 +1,50 @@
+package docker
+
+import (
+	"context"
+	"net/http"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	log "github.com/codeshelldev/secured-signal-api/utils/logger"
+)
+
+var stop chan os.Signal
+
+func Run(main func()) chan os.Signal {
+	stop = make(chan os.Signal, 1)
+	signal.Notify(stop, syscall.SIGINT, syscall.SIGTERM)
+
+	go main()
+
+	return stop
+}
+
+func Exit(code int) {
+	log.Info("Exiting...")
+
+	os.Exit(code)
+
+	stop <- syscall.SIGTERM
+}
+
+func Shutdown(server *http.Server) {
+	log.Info("Shutdown signal received")
+
+	log.Sync()
+
+	ctx, cancel := context.WithTimeout(context.Background(), 5 * time.Second)
+	defer cancel()
+
+	err := server.Shutdown(ctx); 
+
+	if err != nil {
+		log.Fatal("Server shutdown failed: ", err.Error())
+
+		log.Info("Server exited forcefully")
+	} else {
+		log.Info("Server exited gracefully")
+	}
+}