加入网络层

Server/Gateway/handler/ws_handler/client.go

@@ -0,0 +1,89 @@
+package ws_handler
+
+import (
+	"common/log"
+	"common/net/socket"
+	"context"
+	"go.uber.org/zap"
+	"runtime/debug"
+	"sync"
+	"time"
+)
+
+type Client struct {
+	sync.WaitGroup
+	conn      socket.ISocketConn // Socket
+	mailChan  chan Event         // 邮箱队列
+	logger    *zap.SugaredLogger
+	ctx       context.Context
+	cancel    context.CancelFunc
+	heartBeat time.Time
+
+	UID int32
+}
+
+func NewClient(uid int32, conn socket.ISocketConn) *Client {
+	client := &Client{}
+	client.UID = uid
+	client.conn = conn
+	client.logger = log.GetLogger().Named("uid").With("uid", client.UID)
+	client.logger.Errorf("错误日志 %v", 1)
+
+	client.heartBeat = time.Now()
+	client.mailChan = make(chan Event, 1024)
+	client.ctx, client.cancel = context.WithCancel(context.Background())
+	go client.Loop()
+	return client
+}
+
+// CloseClient 关闭客户端
+func (c *Client) CloseClient() {
+	if c.cancel != nil {
+		c.cancel()
+		c.Wait()
+	}
+}
+
+func (c *Client) onClose() {
+	if c.conn != nil {
+		_ = c.conn.Close()
+		c.conn = nil
+	}
+	if c.mailChan != nil {
+		close(c.mailChan)
+		c.mailChan = nil
+	}
+	UserMgr.Delete(c.UID)
+	c.Done()
+}
+
+func (c *Client) Loop() {
+	defer func() {
+		if err := recover(); err != nil {
+			c.logger.Errorf("Client Loop err: %v", err)
+			debug.PrintStack()
+		}
+	}()
+	defer c.onClose()
+	c.Add(1)
+	//心跳检测
+	hearBeatTicker := time.NewTicker(3000 * time.Millisecond)
+	for {
+		select {
+		case <-c.ctx.Done():
+			return
+		case _, _ = <-c.mailChan:
+
+		case <-hearBeatTicker.C:
+			// 心跳超时直接关掉连接
+			if c.checkHeartBeatTimeout() {
+				return
+			}
+		}
+	}
+}
+
+func (c *Client) checkHeartBeatTimeout() bool {
+	sub := time.Now().Sub(c.heartBeat)
+	return sub > 60*time.Second
+}