发布v2.3.5版本，更新内容请查看：https://github.com/bufanyun/hotgo/blob/v2.0/docs/guide-zh-CN/start-update-log.md

2025-08-28 13:03:22 +08:00 · 2023-03-13 17:00:46 +08:00
parent ab912d0ba6
commit 1acc6d17c4
51 changed files with 2777 additions and 1024 deletions
--- a/server/internal/library/network/tcp/server.go
+++ b/server/internal/library/network/tcp/server.go
@@ -0,0 +1,278 @@
+package tcp
+
+import (
+	"context"
+	"fmt"
+	"github.com/gogf/gf/v2/encoding/gjson"
+	"github.com/gogf/gf/v2/errors/gerror"
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/net/gtcp"
+	"github.com/gogf/gf/v2/os/gctx"
+	"github.com/gogf/gf/v2/os/glog"
+	"reflect"
+	"sync"
+	"time"
+)
+
+type ClientConn struct {
+	Conn      *gtcp.Conn
+	Auth      *AuthMeta
+	heartbeat int64
+	mutex     sync.Mutex
+}
+
+type ServerConfig struct {
+	Name string // 服务名称
+	Addr string // 监听地址
+}
+
+type Server struct {
+	Ctx          context.Context
+	Logger       *glog.Logger
+	addr         string
+	name         string
+	ln           *gtcp.Server
+	wgLn         sync.WaitGroup
+	mutex        sync.Mutex
+	closeFlag    bool
+	clients      map[string]*ClientConn // 已登录的认证客户端
+	mutexConns   sync.Mutex
+	wgConns      sync.WaitGroup
+	cronRouters  map[string]RouterHandler // 路由
+	queueRouters map[string]RouterHandler
+	authRouters  map[string]RouterHandler
+}
+
+func NewServer(config *ServerConfig) (server *Server, err error) {
+	if config == nil {
+		err = gerror.New("config is nil")
+		return
+	}
+
+	if config.Addr == "" {
+		err = gerror.New("server address is not set")
+		return
+	}
+
+	if config.Name == "" {
+		config.Name = "hotgo"
+	}
+
+	server = new(Server)
+	server.Ctx = gctx.New()
+	server.addr = config.Addr
+	server.name = config.Name
+	server.ln = gtcp.NewServer(server.addr, server.accept, config.Name)
+	server.clients = make(map[string]*ClientConn)
+	server.closeFlag = false
+
+	logger := glog.New()
+	path := g.Cfg().MustGet(server.Ctx, "logger.path", "logs/logger").String()
+	if err = logger.SetPath(fmt.Sprintf("%s/tcp.server/%s", path, config.Name)); err != nil {
+		return
+	}
+	server.Logger = logger
+
+	server.startCron()
+
+	return
+}
+
+func (server *Server) accept(conn *gtcp.Conn) {
+	defer func() {
+		server.mutexConns.Lock()
+		conn.Close()
+		// 从登录列表中移除
+		if _, ok := server.clients[conn.RemoteAddr().String()]; ok {
+			delete(server.clients, conn.RemoteAddr().String())
+		}
+		server.mutexConns.Unlock()
+	}()
+
+	for {
+		msg, err := RecvPkg(conn)
+		if err != nil {
+			server.Logger.Debugf(server.Ctx, "RecvPkg err:%+v, client closed.", err)
+			break
+		}
+
+		client := server.getLoginConn(conn)
+
+		switch msg.Router {
+		case "ServerLogin": // 服务登录
+			server.onServerLogin(msg.Data, conn)
+		case "ServerHeartbeat": // 心跳
+			if client == nil {
+				server.Logger.Infof(server.Ctx, "conn not connected, ignore the heartbeat, msg:%+v", msg)
+				continue
+			}
+			server.onServerHeartbeat(msg, client)
+		default: // 通用路由消息处理
+			if client == nil {
+				server.Logger.Warningf(server.Ctx, "conn is not logged in but sends a routing message. actively conn disconnect, msg:%+v", msg)
+				time.Sleep(time.Second)
+				conn.Close()
+				return
+			}
+			server.handleRouterMsg(msg, client)
+		}
+	}
+}
+
+// handleRouterMsg 处理路由消息
+func (server *Server) handleRouterMsg(msg *Message, client *ClientConn) {
+
+	// 验证签名
+	err := VerifySign(msg.Data, client.Auth.AppId, client.Auth.SecretKey)
+	if err != nil {
+		server.Logger.Warningf(server.Ctx, "handleRouterMsg VerifySign err:%+v message: %+v", err, msg)
+		return
+	}
+
+	handle := func(routers map[string]RouterHandler, group string) {
+		if routers == nil {
+			server.Logger.Debugf(server.Ctx, "handleRouterMsg route is not initialized %v message: %+v", group, msg)
+			return
+		}
+		f, ok := routers[msg.Router]
+		if !ok {
+			server.Logger.Debugf(server.Ctx, "handleRouterMsg invalid %v message: %+v", group, msg)
+			return
+		}
+		f(msg.Data, client)
+	}
+
+	switch client.Auth.Group {
+	case ClientGroupCron:
+		handle(server.cronRouters, client.Auth.Group)
+	case ClientGroupQueue:
+		handle(server.queueRouters, client.Auth.Group)
+	case ClientGroupAuth:
+		handle(server.authRouters, client.Auth.Group)
+	default:
+		server.Logger.Warningf(server.Ctx, "group is not registered: %+v", client.Auth.Group)
+	}
+}
+
+// getLoginConn 获取指定已登录的连接
+func (server *Server) getLoginConn(conn *gtcp.Conn) *ClientConn {
+	client, ok := server.clients[conn.RemoteAddr().String()]
+	if !ok {
+		return nil
+	}
+	return client
+}
+
+// getLoginConn 获取指定appid的所有连接
+func (server *Server) getAppIdClients(appid string) (list []*ClientConn) {
+	for _, v := range server.clients {
+		if v.Auth.AppId == appid {
+			list = append(list, v)
+		}
+	}
+	return
+}
+
+// RegisterAuthRouter 注册授权路由
+func (server *Server) RegisterAuthRouter(routers map[string]RouterHandler) {
+	server.mutex.Lock()
+	defer server.mutex.Unlock()
+
+	if server.authRouters == nil {
+		server.authRouters = make(map[string]RouterHandler)
+	}
+
+	for i, router := range routers {
+		_, ok := server.authRouters[i]
+		if ok {
+			server.Logger.Debugf(server.Ctx, "server authRouters duplicate registration:%v", i)
+			continue
+		}
+		server.authRouters[i] = router
+	}
+}
+
+// RegisterCronRouter 注册任务路由
+func (server *Server) RegisterCronRouter(routers map[string]RouterHandler) {
+	server.mutex.Lock()
+	defer server.mutex.Unlock()
+
+	if server.cronRouters == nil {
+		server.cronRouters = make(map[string]RouterHandler)
+	}
+
+	for i, router := range routers {
+		_, ok := server.cronRouters[i]
+		if ok {
+			server.Logger.Debugf(server.Ctx, "server cronRouters duplicate registration:%v", i)
+			continue
+		}
+		server.cronRouters[i] = router
+	}
+}
+
+// RegisterQueueRouter 注册队列路由
+func (server *Server) RegisterQueueRouter(routers map[string]RouterHandler) {
+	server.mutex.Lock()
+	defer server.mutex.Unlock()
+
+	if server.queueRouters == nil {
+		server.queueRouters = make(map[string]RouterHandler)
+	}
+
+	for i, router := range routers {
+		_, ok := server.queueRouters[i]
+		if ok {
+			server.Logger.Debugf(server.Ctx, "server queueRouters duplicate registration:%v", i)
+			continue
+		}
+		server.queueRouters[i] = router
+	}
+}
+
+func (server *Server) Listen() (err error) {
+	server.wgLn.Add(1)
+	defer server.wgLn.Done()
+	return server.ln.Run()
+}
+
+// Close 关闭服务
+func (server *Server) Close() {
+	if server.closeFlag {
+		return
+	}
+	server.closeFlag = true
+
+	server.stopCron()
+
+	server.mutexConns.Lock()
+	for _, client := range server.clients {
+		client.Conn.Close()
+	}
+	server.clients = nil
+	server.mutexConns.Unlock()
+	server.wgConns.Wait()
+
+	if server.ln != nil {
+		server.ln.Close()
+	}
+	server.wgLn.Wait()
+}
+
+// Write 向指定客户端发送消息
+func (server *Server) Write(conn *gtcp.Conn, data interface{}) (err error) {
+	if server.closeFlag {
+		return gerror.New("service is down")
+	}
+
+	msgType := reflect.TypeOf(data)
+	if msgType == nil || msgType.Kind() != reflect.Ptr {
+		return gerror.Newf("json message pointer required: %+v", data)
+	}
+
+	msg := &Message{Router: msgType.Elem().Name(), Data: data}
+
+	server.Logger.Debugf(server.Ctx, "server Write Router:%v, data:%+v", msg.Router, gjson.New(data).String())
+
+	return SendPkg(conn, msg)
+}