完善服务器

Server/Gateway/handler/ws_handler/handler.go

@@ -0,0 +1,71 @@
+package ws_handler
+
+import (
+	"common/net/grpc/service"
+	"common/proto/gen/grpc_pb"
+	"encoding/json"
+	"gateway/grpc_server/stream_client"
+	"time"
+)
+
+func (c *Client) handle(event Event) {
+	switch e := event.(type) {
+	case *ClientEvent:
+		m, err := parseMsg(e.Msg)
+		if err != nil {
+			c.logger.Errorf("handle event json.Unmarshal err: %v", err)
+			c.cancel()
+		}
+		c.logger.Infof("收到客户端消息：%+v", *m)
+		switch m.Type {
+		case "enter":
+			c.onEnter(m)
+		case "action":
+			c.onAction(m)
+		}
+	case *PongEvent:
+		c.heartBeat = time.Now()
+	}
+}
+
+func (c *Client) onEnter(msg *tempMsg) {
+	//_ = c.conn.Write(wapMsg(&tempMsg{
+	//	Type: "init",
+	//	Data: fmt.Sprintf("[%v,%v]", utils.RandInt(1, 100), utils.RandInt(1, 100)),
+	//}))
+	client, err := service.SceneNewClient()
+	if err != nil {
+		c.logger.Errorf("SceneNewClient err: %v", err)
+		return
+	}
+	resp, err := client.Enter(c.ctx, &grpc_pb.EnterReq{
+		UID:        int32(c.UID),
+		SID:        0,
+		InstanceID: 1,
+	})
+	if err != nil {
+		c.logger.Errorf("enter err: %v", err)
+		return
+	}
+	c.SceneSID = resp.SID
+}
+
+func (c *Client) onAction(msg *tempMsg) {
+	if c.SceneSID == 0 {
+		return
+	}
+	d := &tempAction{}
+	if err := json.Unmarshal([]byte(msg.Data), d); err != nil {
+		return
+	}
+	m := &tempActionMove{}
+	if err := json.Unmarshal([]byte(d.Data), m); err != nil {
+		return
+	}
+	stream := stream_client.FindSceneBySID(c.SceneSID, stream_client.FunAction)
+	if err := stream.SendMsg(&grpc_pb.ActionReq{
+		Action: int32(m.Move),
+	}); err != nil {
+		c.logger.Errorf("send action err: %v", err)
+	}
+}