graceful shutdown allows to server to wait for 10 sec to finish work

main.go

@@ -1,17 +1,58 @@
 package main
 
 import (
+	"context"
+	"log"
+	"net/http"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
 	"ukiyo/pkg/apilayer"
 	"ukiyo/pkg/webhooks"
 
 	"github.com/gin-gonic/gin"
 )
 
 func main() {
-	r := gin.Default()
-	webhooks.HooksListener(r)
-	webhooks.HealthCheck(r)
-	apilayer.AddContainer(r)
-	apilayer.EditContainerToken(r)
-	r.Run(":8080")
+	router := gin.Default()
+
+	webhooks.HooksListener(router)
+	webhooks.HealthCheck(router)
+	apilayer.AddContainer(router)
+	apilayer.EditContainerToken(router)
+
+	srv := &http.Server{
+		Addr:    ":8080",
+		Handler: router,
+	}
+
+	// Initializing the server in a goroutine so that
+	// it won't block the graceful shutdown handling below
+	go func() {
+		if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
+			log.Fatalf("listen: %s\n", err)
+		}
+	}()
+
+	// Wait for interrupt signal to gracefully shutdown the server with
+	// a timeout of 5 seconds.
+	quit := make(chan os.Signal)
+	// kill (no param) default send syscall.SIGTERM
+	// kill -2 is syscall.SIGINT
+	// kill -9 is syscall.SIGKILL but can't be catch, so don't need add it
+	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
+	<-quit
+	log.Println("ukiyo Shutting down server...")
+
+	// The context is used to inform the server it has 10 seconds to finish
+	// the request it is currently handling
+	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+	defer cancel()
+	if err := srv.Shutdown(ctx); err != nil {
+		log.Fatal("ukiyo Server forced to shutdown:", err)
+	}
+
+	log.Println("ukiyo Server exiting")
+
 }