chore: 并入 iotlcm-app 历史到 app/（原提交 f40bda9）

2026-11-02 13:28:00 +08:00
parent 03ef151f5b f40bda99a8
commit 226495365e
1107 changed files with 147143 additions and 0 deletions
@@ -0,0 +1,58 @@
+package cmd
+
+import (
+	"context"
+	"os"
+	"sync"
+	"git.5iot.com/iotlcm-public/iotlcm-app/internal/consts"
+	"git.5iot.com/iotlcm-public/iotlcm-app/internal/library/event"
+
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/os/gctx"
+	"github.com/gogf/gf/v2/os/gproc"
+)
+
+var (
+	serverCloseSignal = make(chan struct{}, 1)
+	serverWg          = sync.WaitGroup{}
+	once              sync.Once
+)
+
+// 等待接收服务关闭信号
+func waitServerClose(ctx context.Context) {
+	signalListen(ctx, handleSignalClose)
+	<-serverCloseSignal
+}
+
+// 等待所有服务关闭
+func waitAllServerClose(ctx context.Context) {
+	serverWg.Wait()
+}
+
+// 处理服务关闭信号
+func handleSignalClose(sig os.Signal) {
+	serverCloseEvent(gctx.GetInitCtx())
+	serverCloseSignal <- struct{}{}
+}
+
+// 监听信号
+func signalListen(ctx context.Context, handler ...gproc.SigHandler) {
+	g.Go(ctx, func(ctx context.Context) {
+		// syscall.SIGINT,
+		// syscall.SIGQUIT,
+		// syscall.SIGKILL,
+		// syscall.SIGTERM,
+		// syscall.SIGABRT.
+		gproc.AddSigHandlerShutdown(handler...)
+		gproc.Listen()
+	}, func(ctx context.Context, exception error) {
+		g.Log().Fatalf(ctx, "信号监听异常 err: %+v", exception)
+	})
+}
+
+// serverCloseEvent 关闭事件
+func serverCloseEvent(ctx context.Context) {
+	once.Do(func() {
+		event.EventsInstance().Emit(ctx, consts.EventKeyServerClose, nil)
+	})
+}