im 管理后台

server/internal/logic/tcpserver/server.go

@@ -0,0 +1,74 @@
+// Package tcpserver
+// @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 tcpserver
+
+import (
+	"context"
+	"github.com/gogf/gf/v2/frame/g"
+	"hotgo/internal/library/network/tcp"
+	"hotgo/internal/service"
+	"hotgo/utility/simple"
+)
+
+type sTCPServer struct {
+	serv *tcp.Server
+}
+
+func init() {
+	service.RegisterTCPServer(newTCPServer())
+}
+
+func newTCPServer() *sTCPServer {
+	return &sTCPServer{}
+}
+
+// Instance 获取实例
+func (s *sTCPServer) Instance() *tcp.Server {
+	return s.serv
+}
+
+// Start 启动服务
+func (s *sTCPServer) Start(ctx context.Context) {
+	simple.SafeGo(ctx, func(ctx context.Context) {
+		g.Log().Debug(ctx, "TCPServer start..")
+
+		s.serv = tcp.NewServer(&tcp.ServerConfig{
+			Name: simple.AppName(ctx),
+			Addr: g.Cfg().MustGet(ctx, "tcp.server.address").String(),
+		})
+
+		// 注册路由
+		s.serv.RegisterRouter(
+			s.onServerLogin,     // 服务登录
+			s.onServerHeartbeat, // 心跳
+			s.OnAuthSummary,     // 获取授权信息
+			s.OnExampleHello,    // 一个tcp请求例子
+		)
+
+		// 注册RPC路由
+		s.serv.RegisterRPCRouter(
+			s.OnExampleRPCHello, // 一个rpc请求例子
+		)
+
+		// 注册拦截器
+		s.serv.RegisterInterceptor(s.DefaultInterceptor, s.PreFilterInterceptor)
+
+		// 服务监听
+		if err := s.serv.Listen(); err != nil {
+			if !s.serv.IsClose() {
+				g.Log().Warningf(ctx, "TCPServer Listen err:%v", err)
+			}
+		}
+	})
+}
+
+// Stop 关闭服务
+func (s *sTCPServer) Stop(ctx context.Context) {
+	if s.serv != nil && !s.serv.IsClose() {
+		s.serv.Close()
+		g.Log().Debug(ctx, "TCPServer stop..")
+	}
+}