This commit is contained in:
laodaming
2023-07-25 18:04:01 +08:00
parent d0fc91b61b
commit b0f598af68
4 changed files with 14 additions and 15 deletions

View File

@@ -97,8 +97,8 @@ func DataTransferHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
mapConnPool.Store(flag, ws)
defer ws.close()
//把连接成功消息发回去
rsp.MsgType = constants.WEBSOCKET_CONNECT_SUCCESS
rsp.Message = flag
rsp.T = constants.WEBSOCKET_CONNECT_SUCCESS
rsp.D = flag
b, _ := json.Marshal(rsp)
conn.WriteMessage(websocket.TextMessage, b)
//循环读客户端信息
@@ -116,8 +116,8 @@ func DataTransferHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
// 心跳
func (w *wsConnectItem) heartbeat() {
rsp := types.DataTransferRsp{
MsgType: constants.WEBSOCKET_HEARTBEAT,
Message: "",
T: constants.WEBSOCKET_HEARTBEAT,
D: "",
}
for {
time.Sleep(time.Second * 10)
@@ -202,11 +202,11 @@ func (w *wsConnectItem) dealwithReciveData(data []byte) {
logx.Error("invalid format of websocket message")
return
}
switch parseInfo.MsgType {
switch parseInfo.T {
//图片渲染
case constants.WEBSOCKET_RENDER_IMAGE:
var renderImageData []types.RenderImageReqMsg
if err := json.Unmarshal([]byte(parseInfo.Message), &renderImageData); err != nil {
if err := json.Unmarshal([]byte(parseInfo.D), &renderImageData); err != nil {
logx.Error("invalid format of websocket render image message", err)
return
}

View File

@@ -50,7 +50,6 @@ func RenderNotifyHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
h.Write([]byte(fmt.Sprintf(constants.RENDER_NOTIFY_SIGN_KEY, string(notifyByte), req.Time)))
signHex := h.Sum(nil)
sign := hex.EncodeToString(signHex)
fmt.Println(sign)
if req.Sign != sign {
httpx.OkJsonCtx(r.Context(), w, basic.Response{
Code: basic.CodeRequestParamsErr.Code,