server

Server/Gateway/handler/ws_handler/client.go

@@ -3,14 +3,18 @@ package ws_handler
 import (
 	"common/log"
 	"common/net/socket"
+	"common/proto/gen/cs"
 	"context"
 	"fmt"
 	"go.uber.org/zap"
+	"google.golang.org/protobuf/proto"
 	"runtime/debug"
 	"sync"
 	"time"
 )
 
+var GatewaySID int64
+
 type Client struct {
 	sync.WaitGroup
 	conn      socket.ISocketConn // Socket
@@ -20,8 +24,10 @@ type Client struct {
 	cancel    context.CancelFunc // 取消上下文
 	heartBeat time.Time          // 最后一次心跳
 
-	UID      int
-	SceneSID int64 // 场景服ID
+	UID        int
+	SceneSID   int64 // 场景服ID
+	InstanceID int
+	UniqueNo   int64
 }
 
 func NewClient(uid int, conn socket.ISocketConn) *Client {
@@ -78,6 +84,39 @@ func (c *Client) OnEvent(event Event) {
 	}
 }
 
+func (c *Client) WriteMessage(id cs.MessageID, data proto.Message) {
+	d, err := proto.Marshal(data)
+	if err != nil {
+		c.logger.Errorf("WriteMessage proto.Marshal err: %v", err)
+		return
+	}
+	m, err := proto.Marshal(&cs.Message{
+		ID:      id,
+		Payload: d,
+	})
+	if err != nil {
+		c.logger.Errorf("WriteMessage proto.Marshal err: %v", err)
+		return
+	}
+	if err = c.conn.Write(m); err != nil {
+		c.logger.Errorf("WriteMessage err: %v", err)
+	}
+}
+
+func (c *Client) WriteBytes(id cs.MessageID, data []byte) {
+	m, err := proto.Marshal(&cs.Message{
+		ID:      id,
+		Payload: data,
+	})
+	if err != nil {
+		c.logger.Errorf("WriteBytes proto.Marshal err: %v", err)
+		return
+	}
+	if err = c.conn.Write(m); err != nil {
+		c.logger.Errorf("WriteBytes err: %v", err)
+	}
+}
+
 // CloseClient 关闭客户端（同步，会等待onClose执行完成）
 func (c *Client) CloseClient() {
 	if c.cancel != nil {