1 Star 0 Fork 0

player1 / restful-api-demo

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
manager.go 1.65 KB
一键复制 编辑 原始数据 按行查看 历史
player1 提交于 2024-02-17 22:54 . 调整优雅关闭的代码
package core
import (
"fmt"
"os"
"os/signal"
"sync"
"syscall"
"github.com/infraboard/mcube/logger"
"github.com/infraboard/mcube/logger/zap"
"gitee.com/player1111/restful-api-demo.git/http"
)
type Manager struct {
httpService *http.HttpService
handlerService http.HandlerService
wg sync.WaitGroup
log logger.Logger
}
func NewManager() *Manager {
return &Manager{
handlerService: http.NewHandlerService(),
httpService: http.NewHttpService(),
log: zap.L().Named("Manager"),
}
}
func (m *Manager) Start() error {
/* m.handlerService = http.NewHandlerService()
m.httpService = http.NewHttpService()*/
// 启动handlerService
if err := m.handlerService.Start(); err != nil {
fmt.Printf("Failed to start HandlerService: %v\n", err)
return fmt.Errorf("Failed to start HandlerService: %v\n", err)
}
// 启动httpService
if err := m.httpService.Start(); err != nil {
fmt.Printf("Failed to start HttpService: %v\n", err)
return fmt.Errorf("Failed to start HttpService: %v\n", err)
}
return nil
}
func (m *Manager) Stop() error {
var err error
if err = m.handlerService.Stop(); err != nil {
fmt.Printf("Failed to stop HandlerService: %v\n", err)
}
if err = m.httpService.Stop(); err != nil {
fmt.Printf("Failed to stop HttpService: %v\n", err)
}
return err
}
// 处理来自外部的中断信号
func (m *Manager) WaitStop() {
ch := make(chan os.Signal, 1)
signal.Notify(ch, syscall.SIGTERM, syscall.SIGQUIT, syscall.SIGHUP, syscall.SIGINT)
sig := <-ch
m.log.Infof("received signal:%s\n", sig)
if err := m.httpService.Stop(); err != nil {
m.log.Warnf("http Service shutdown failed:%v", err)
}
}
Go
1
https://gitee.com/player1111/restful-api-demo.git
git@gitee.com:player1111/restful-api-demo.git
player1111
restful-api-demo
restful-api-demo
19fd67857982

搜索帮助