feat: Implement graceful shutdown

Close DropKit#20

main.go

@@ -1,9 +1,16 @@
 package main
 
 import (
+	"context"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
 	"github.com/DropKit/DropKit-Adapter/logger"
 	routes "github.com/DropKit/DropKit-Adapter/router"
 	"github.com/DropKit/DropKit-Adapter/services"
+	"github.com/ethereum/go-ethereum/log"
 	"github.com/spf13/viper"
 )
 
@@ -25,6 +32,19 @@ func init() {
 }
 
 func main() {
-	router := routes.SetupRouter()
-	router.Run(":" + viper.GetString(`DROPKIT.PORT`))
+	// router := routes.SetupRouter()
+	_, srv := routes.SetupRouter()
+	// router.Run(":" + viper.GetString(`DROPKIT.PORT`))
+
+	shutdown := make(chan os.Signal)
+	signal.Notify(shutdown, syscall.SIGINT, syscall.SIGTERM)
+	<-shutdown
+	log.Info("Shutdown Server")
+
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+	if err := srv.Shutdown(ctx); err != nil {
+		log.Error("Forced to shutdown server")
+	}
+
 }

router/routers.go

@@ -86,5 +86,6 @@ func SetupRouter() *gin.Engine {
 	{
 		roleRoute.POST("/create", role.CreateColumnRole)
 	}
-	return r
+
+	return r, srv
 }