引入并集成服务平滑关闭

router/server.go

@@ -140,7 +140,11 @@ func (s *server) Start() {
 	// 注册文档
 	s.uiInstance.RegisterHandler(s.router, s.option.swaggerBaseUri)
 	gracefulServer := graceful.NewServer(fmt.Sprintf(":%d", s.port), s.Router())
-	if err := gracefulServer.Serve(); err != nil {
+	if err := gracefulServer.ListenAndServe(); err != nil {
+		if strings.Contains(err.Error(), "use of closed network connection") {
+			fmt.Println("接收到退出指令, 服务平滑关闭")
+			return
+		}
 		panic("服务启动监听失败" + err.Error())
 	}
 }

router/server_test.go

@@ -8,10 +8,12 @@
 package router
 
 import (
+	"github.com/gin-gonic/gin"
 	"testing"
 )
 
 func TestNewServer(t *testing.T) {
 	s := NewServer(9087)
+	s.Router().GET("/ping", func(c *gin.Context) {})
 	s.Start()
 }