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

package logic

import (
	"encoding/json"
	"fusenapi/constants"
	"github.com/zeromicro/go-zero/core/logx"
)

// 刷新重连请求恢复上次连接的标识
func (w *wsConnectItem) reuseLastConnect(data []byte) {
	logx.Info("收到请求恢复上次连接标识数据:", string(data))
	var clientId string
	if err := json.Unmarshal(data, &clientId); err != nil {
		logx.Error(" invalid format of client id :", clientId)
		w.sendToOutChan(w.respondDataFormat(constants.WEBSOCKET_REQUEST_RESUME_LAST_CONNECT_ERR, "invalid format of client id"))
		return
	}
	lenClientId := len(clientId)
	//id长度不对
	if lenClientId > 100 {
		w.sendToOutChan(w.respondDataFormat(constants.WEBSOCKET_REQUEST_RESUME_LAST_CONNECT_ERR, "length of client id is to long"))
		return
	}
	//合成client后缀,不是同个后缀的不能复用
	userPart := getUserPart(w.userId, w.guestId)
	lenUserPart := len(userPart)
	if lenClientId <= lenUserPart {
		w.sendToOutChan(w.respondDataFormat(constants.WEBSOCKET_REQUEST_RESUME_LAST_CONNECT_ERR, "length of client id is to short"))
		return
	}
	//尾部不同不能复用
	if clientId[lenClientId-lenUserPart:] != userPart {
		w.sendToOutChan(w.respondDataFormat(constants.WEBSOCKET_REQUEST_RESUME_LAST_CONNECT_ERR, "the client id is not belong you before"))
		return
	}
	publicMutex.Lock()
	defer publicMutex.Unlock()
	//存在是不能给他申请重新绑定
	if _, ok := mapConnPool.Load(clientId); ok {
		rsp := w.respondDataFormat(constants.WEBSOCKET_REQUEST_RESUME_LAST_CONNECT_ERR, "id has bound by other connect ")
		w.sendToOutChan(rsp)
		return
	}
	//重新绑定
	w.uniqueId = clientId
	rsp := w.respondDataFormat(constants.WEBSOCKET_CONNECT_SUCCESS, clientId)
	w.sendToOutChan(rsp)
	return
}