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) } }