fusenapi/server/websocket/internal/logic/ws_reuse_last_connect.go

72 lines
2.0 KiB
Go
Raw Normal View History

2023-08-15 10:14:44 +00:00
package logic
2023-08-23 07:08:45 +00:00
//复用websocket连接标识
2023-08-15 10:14:44 +00:00
import (
"encoding/json"
2023-08-18 09:33:42 +00:00
"fmt"
2023-08-15 10:14:44 +00:00
"fusenapi/constants"
2023-08-23 08:45:01 +00:00
"fusenapi/utils/encryption_decryption"
2023-08-15 10:14:44 +00:00
"github.com/zeromicro/go-zero/core/logx"
)
2023-08-30 09:40:40 +00:00
// 复用连接处理器
2023-08-30 09:46:47 +00:00
type reuseConnProcessor struct {
2023-08-30 10:00:26 +00:00
w wsConnectItem
2023-08-30 09:40:40 +00:00
}
2023-08-30 09:46:47 +00:00
func (r *reuseConnProcessor) allocationMessage(data []byte) {
2023-08-15 10:14:44 +00:00
logx.Info("收到请求恢复上次连接标识数据:", string(data))
2023-08-24 09:23:41 +00:00
var wid string
if err := json.Unmarshal(data, &wid); err != nil {
logx.Error(" invalid format of wid :", wid)
2023-08-30 09:40:40 +00:00
r.w.incomeDataFormatErrResponse("invalid format of wid")
2023-08-15 10:14:44 +00:00
return
}
2023-08-23 08:45:01 +00:00
//解密
2023-08-24 09:23:41 +00:00
decryptionWid, err := encryption_decryption.CBCDecrypt(wid)
2023-08-23 08:45:01 +00:00
if err != nil {
2023-08-30 09:40:40 +00:00
r.w.reuseLastConnErrResponse("invalid wid")
2023-08-15 10:14:44 +00:00
return
}
2023-08-24 09:23:41 +00:00
lendecryptionWid := len(decryptionWid)
2023-08-15 10:38:33 +00:00
//合成client后缀,不是同个后缀的不能复用
2023-08-30 09:40:40 +00:00
userPart := getUserJoinPart(r.w.userId, r.w.guestId, r.w.userAgent)
2023-08-15 10:38:33 +00:00
lenUserPart := len(userPart)
2023-08-24 09:23:41 +00:00
if lendecryptionWid <= lenUserPart {
2023-08-30 09:40:40 +00:00
r.w.reuseLastConnErrResponse("length of client id is to short")
2023-08-15 10:38:33 +00:00
return
}
//尾部不同不能复用
2023-08-24 09:23:41 +00:00
if decryptionWid[lendecryptionWid-lenUserPart:] != userPart {
2023-08-30 09:40:40 +00:00
r.w.reuseLastConnErrResponse("the client id is not belong to you before")
2023-08-15 10:38:33 +00:00
return
}
2023-08-15 10:14:44 +00:00
//存在是不能给他申请重新绑定
2023-08-24 09:23:41 +00:00
if v, ok := mapConnPool.Load(wid); ok {
2023-08-18 03:05:18 +00:00
obj, ok := v.(wsConnectItem)
if !ok {
logx.Error("连接断言失败")
}
2023-08-24 06:11:17 +00:00
//是当前自己占用(无需处理)
2023-08-30 09:40:40 +00:00
if obj.uniqueId == r.w.uniqueId {
rsp := r.w.respondDataFormat(constants.WEBSOCKET_CONNECT_SUCCESS, wid)
r.w.sendToOutChan(rsp)
2023-08-18 03:05:18 +00:00
return
} else {
2023-08-30 09:40:40 +00:00
r.w.reuseLastConnErrResponse("the wid is used by other people")
2023-08-18 03:05:18 +00:00
return
}
2023-08-15 10:14:44 +00:00
}
//重新绑定
2023-08-30 09:40:40 +00:00
r.w.uniqueId = wid
mapConnPool.Store(wid, *r.w)
rsp := r.w.respondDataFormat(constants.WEBSOCKET_CONNECT_SUCCESS, wid)
r.w.sendToOutChan(rsp)
2023-08-15 10:14:44 +00:00
return
}
2023-08-18 09:33:42 +00:00
// 获取用户拼接部分(复用标识用到)
2023-08-30 03:25:20 +00:00
func getUserJoinPart(userId, guestId int64, userAgent string) string {
return fmt.Sprintf("|_%d_%d_|_%s_|", userId, guestId, userAgent)
2023-08-18 09:33:42 +00:00
}