im 管理后台

2026-08-11 23:13:51 +08:00
parent 0397fe5f22
commit 331a55751d
1407 changed files with 162069 additions and 2 deletions
@@ -0,0 +1,44 @@
+// 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)
+	})
+}