Add 优雅地重启或停止

main.go

@@ -1,26 +1,52 @@
 package main
 
 import (
+	"context"
+	"fmt"
 	"github.com/gin-gonic/gin"
 	"go-gin-api/app/config"
 	"go-gin-api/app/route"
 	"log"
+	"net/http"
 	"os"
+	"os/signal"
+	"time"
 )
 
 func main() {
-
-	//gin.DebugMode   测试模式
-	//gin.ReleaseMode 发布模式
-	gin.SetMode(gin.DebugMode)
+	gin.SetMode(config.AppMode)
 	engine := gin.New()
 
 	// 设置路由
 	route.SetupRouter(engine)
 
-	// 启动服务
-	if err := engine.Run(config.AppPort); err != nil {
-		log.Println(err)
-		os.Exit(1)
+	server := &http.Server{
+		Addr         : config.AppPort,
+		Handler      : engine,
+		ReadTimeout  : config.AppReadTimeout * time.Second,
+		WriteTimeout : config.AppWriteTimeout * time.Second,
 	}
+
+	info := fmt.Sprintf("HTTP server listening %s, Pid is %v ", config.AppPort, os.Getpid())
+	fmt.Println(info)
+
+	go func() {
+		if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
+			log.Fatalf("HTTP server listen: %s\n", err)
+		}
+	}()
+
+	// 等待中断信号以优雅地关闭服务器（设置 5 秒的超时时间）
+	signalChan := make(chan os.Signal)
+	signal.Notify(signalChan, os.Interrupt)
+	sig := <-signalChan
+	log.Println("Get Signal:", sig)
+	log.Println("Shutdown Server ...")
+
+	ctx, cancel := context.WithTimeout(context.Background(), 5 * time.Second)
+	defer cancel()
+	if err := server.Shutdown(ctx); err != nil {
+		log.Fatal("Server Shutdown:", err)
+	}
+	log.Println("Server exiting")
 }