移动入口文件

cmd/main.go

@@ -0,0 +1,42 @@
+package main
+
+import (
+	"fmt"
+	"go-desk-service/config"
+	"go-desk-service/libs"
+	"go-desk-service/router"
+	"log"
+	"os"
+	"os/signal"
+	"strconv"
+	"syscall"
+
+	"github.com/gin-gonic/gin"
+)
+
+func main() {
+	appConfig, err := config.LoadConfig()
+	if err != nil {
+		fmt.Println(err)
+		fmt.Println("读取配置失败")
+	}
+
+	go handleShutdownSignals()
+	libs.Connect()
+	app := gin.Default()
+	router.Init(app)
+	app.Run(":" + strconv.Itoa(appConfig.Port))
+}
+
+// 处理关闭信号的函数
+func handleShutdownSignals() {
+	sigChan := make(chan os.Signal, 1)
+	signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT)
+
+	// 阻塞等待信号
+	sig := <-sigChan
+	log.Printf("接收到信号: %v, 系统关闭", sig)
+
+	// 强制退出程序
+	os.Exit(0)
+}

main.go

@@ -1,22 +0,0 @@
-package main
-
-import (
-	"fmt"
-	"go-desk-service/config"
-	"go-desk-service/router"
-	"strconv"
-
-	"github.com/gin-gonic/gin"
-)
-
-func main() {
-	appConfig, err := config.LoadConfig()
-	if err != nil {
-		fmt.Println(err)
-		fmt.Println("读取配置失败")
-	}
-
-	app := gin.Default()
-	router.Init(app)
-	app.Run(":" + strconv.Itoa(appConfig.Port))
-}