mirror of
https://github.com/bufanyun/hotgo.git
synced 2025-01-23 19:00:24 +08:00
45 lines
1.1 KiB
Go
45 lines
1.1 KiB
Go
// Package cmd
|
|
// @Link https://github.com/bufanyun/hotgo
|
|
// @Copyright Copyright (c) 2023 HotGo CLI
|
|
// @Author Ms <133814250@qq.com>
|
|
// @License https://github.com/bufanyun/hotgo/blob/master/LICENSE
|
|
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"github.com/gogf/gf/v2/os/gctx"
|
|
"github.com/gogf/gf/v2/os/gproc"
|
|
"hotgo/internal/consts"
|
|
"hotgo/utility/simple"
|
|
"os"
|
|
"sync"
|
|
)
|
|
|
|
var (
|
|
serverCloseSignal = make(chan struct{}, 1)
|
|
serverWg = sync.WaitGroup{}
|
|
once sync.Once
|
|
)
|
|
|
|
// signalHandlerForOverall 关闭信号处理
|
|
func signalHandlerForOverall(sig os.Signal) {
|
|
serverCloseEvent(gctx.GetInitCtx())
|
|
serverCloseSignal <- struct{}{}
|
|
}
|
|
|
|
// signalListen 信号监听
|
|
func signalListen(ctx context.Context, handler ...gproc.SigHandler) {
|
|
simple.SafeGo(ctx, func(ctx context.Context) {
|
|
gproc.AddSigHandlerShutdown(handler...)
|
|
gproc.Listen()
|
|
})
|
|
}
|
|
|
|
// serverCloseEvent 关闭事件
|
|
// 区别于服务收到退出信号后的处理，只会执行一次
|
|
func serverCloseEvent(ctx context.Context) {
|
|
once.Do(func() {
|
|
simple.Event().Call(consts.EventServerClose, ctx)
|
|
})
|
|
}
|