79 lines
2.2 KiB
Go
79 lines
2.2 KiB
Go
// Package verify 校验
|
||
package verify
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"go_dreamfactory/comm"
|
||
"go_dreamfactory/pb"
|
||
"go_dreamfactory/stress/model"
|
||
|
||
"google.golang.org/protobuf/proto"
|
||
)
|
||
|
||
// WebSocketResponseJSON 返回数据结构体，返回值为json
|
||
type WebSocketResponseJSON struct {
|
||
Seq string `json:"seq"`
|
||
Cmd string `json:"cmd"`
|
||
Response struct {
|
||
Code int `json:"code"`
|
||
CodeMsg string `json:"codeMsg"`
|
||
Data interface{} `json:"data"`
|
||
} `json:"response"`
|
||
}
|
||
|
||
// WebSocketJSON 通过返回的Body 判断
|
||
// 返回示例: {"seq":"1566276523281-585638","cmd":"heartbeat","response":{"code":200,"codeMsg":"Success","data":null}}
|
||
// code 取body中的返回code
|
||
func WebSocketJSON(request *model.Request, seq string, msg []byte) (code int, isSucceed bool) {
|
||
responseJSON := &WebSocketResponseJSON{}
|
||
err := json.Unmarshal(msg, responseJSON)
|
||
if err != nil {
|
||
code = model.ParseError
|
||
fmt.Printf("请求结果 json.Unmarshal msg:%s err:%v", string(msg), err)
|
||
} else {
|
||
|
||
if seq != responseJSON.Seq {
|
||
code = model.ParseError
|
||
fmt.Println("请求和返回seq不一致 ~请求:", seq, responseJSON.Seq, string(msg))
|
||
} else {
|
||
code = responseJSON.Response.Code
|
||
// body 中code返回200为返回数据成功
|
||
if code == 200 {
|
||
isSucceed = true
|
||
}
|
||
}
|
||
}
|
||
// 开启调试模式
|
||
if request.GetDebug() {
|
||
fmt.Printf("请求结果 seq:%s body:%s \n", seq, string(msg))
|
||
}
|
||
return
|
||
}
|
||
|
||
func WebSocketProto(request *model.Request, seq string, data []byte) (code int, isSucceed bool) {
|
||
|
||
var msg *pb.UserMessage = &pb.UserMessage{}
|
||
|
||
if err := proto.Unmarshal(data, msg); err != nil {
|
||
fmt.Printf("unmarshal err:%v", err)
|
||
}
|
||
//fmt.Printf("接收消息=====msg.MainType = %s,msg.SubType = %s\n", msg.MainType, msg.SubType)
|
||
if msg.MainType == "chat" && msg.SubType == "message" {
|
||
resp := &pb.ChatMessagePush{}
|
||
if !comm.ProtoUnmarshal(msg, resp) {
|
||
code = model.ParseError
|
||
}
|
||
return
|
||
//fmt.Printf("接收消息=====resp:%v\n", resp.Chat)
|
||
} else if msg.MainType == "user" && msg.SubType == "login" {
|
||
resp := &pb.UserLoginResp{}
|
||
if !comm.ProtoUnmarshal(msg, resp) {
|
||
code = model.ParseError
|
||
}
|
||
//fmt.Printf("接收消息=====loginData:%v,userExpand:%v\n", resp.Data, resp.Ex)
|
||
return
|
||
}
|
||
return
|
||
}
|