引入并集成服务平滑关闭

This commit is contained in:
2025-05-29 10:45:06 +08:00
parent 558587ec68
commit 1da666d6fc
4 changed files with 34 additions and 99 deletions

View File

@ -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())
}
}