引入平滑重启，有BUG待调试

router/server.go

@@ -9,6 +9,7 @@ package router
 
 import (
 	"fmt"
+	"git.zhangdeman.cn/zhangdeman/graceful"
 	"net/http"
 	"strings"
 
@@ -138,7 +139,8 @@ type server struct {
 func (s *server) Start() {
 	// 注册文档
 	s.uiInstance.RegisterHandler(s.router, s.option.swaggerBaseUri)
-	if err := s.router.Run(fmt.Sprintf(":%d", s.port)); err != nil {
+	gracefulServer := graceful.NewServer(fmt.Sprintf(":%d", s.port), s.Router())
+	if err := gracefulServer.Serve(); err != nil {
 		panic("服务启动监听失败" + err.Error())
 	}
 }

router/server_test.go

@@ -0,0 +1,17 @@
+// Package router ...
+//
+// Description : router ...
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 2025-05-28 20:58
+package router
+
+import (
+	"testing"
+)
+
+func TestNewServer(t *testing.T) {
+	s := NewServer(9087)
+	s.Start()
+}