fusen/fusenapi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

11

Browse Source
This commit is contained in:
laodaming 2023-10-07 16:10:51 +08:00
parent 67fe103ac1
commit 7fa7b868ca
2 changed files with 20 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/websocket/internal/logic/datatransferlogic.go
View File

@ -85,7 +85,7 @@ type userConnPoolCtlChanItem struct {
guestId int64 //必须（两个用户id任意一个不为0） guestId int64 //必须（两个用户id任意一个不为0）
uniqueId string //主连接池唯一标识（添加/删除时候必须） uniqueId string //主连接池唯一标识（添加/删除时候必须）
message []byte //消息（发送消息传的,格式是经过标准输出序列化后的数据） message []byte //消息（发送消息传的,格式是经过标准输出序列化后的数据）
messageType constants.Websocket //消息类型（发送消息传的） //messageType constants.Websocket //消息类型（发送消息传的）
option int64 //操作 2发消息 1增加 0删除 option int64 //操作 2发消息 1增加 0删除
} }

13
server/websocket/internal/logic/ws_user_connect_pool.go
View File

@ -23,7 +23,6 @@ func createUserConnPoolElement(userId, guestId int64, uniqueId string) {
guestId: guestId, guestId: guestId,
uniqueId: uniqueId, uniqueId: uniqueId,
message: nil, message: nil,
messageType: "",
option: 1, option: 1,
} }
select { select {
@ -42,13 +41,9 @@ func deleteUserConnPoolElement(userId, guestId int64, uniqueId string) {
guestId: guestId, guestId: guestId,
uniqueId: uniqueId, uniqueId: uniqueId,
message: nil, message: nil,
messageType: "",
option: 0, option: 0,
} }
select { userConnPoolCtlChan <- data
case userConnPoolCtlChan <- data:
return
}
} }
// 根据用户索引发现链接并发送(广播)消息到出口队列 // 根据用户索引发现链接并发送(广播)消息到出口队列
@ -89,7 +84,7 @@ func ConsumeUserConnPoolCtlChanData(ctx context.Context) {
userKey = getmapUserConnPoolUniqueId(data.userId, data.guestId) userKey = getmapUserConnPoolUniqueId(data.userId, data.guestId)
switch data.option { switch data.option {
case 2: //发送消息 case 2: //发送消息
logx.Info("通过用户id索引发送消息") //logx.Info("通过用户id索引发送消息")
mapUserUniqueId, ok := mapUserConnPool[userKey] mapUserUniqueId, ok := mapUserConnPool[userKey]
if !ok { if !ok {
logx.Info("通过用户id索引发送消息,连接不存在，用户索引key:", userKey) logx.Info("通过用户id索引发送消息,连接不存在，用户索引key:", userKey)
@ -110,7 +105,7 @@ func ConsumeUserConnPoolCtlChanData(ctx context.Context) {
originConn.sendToOutChan(data.message) originConn.sendToOutChan(data.message)
} }
case 1: //添加 case 1: //添加
logx.Info("添加用户id索引标识：", data.uniqueId) //logx.Info("添加用户id索引标识：", data.uniqueId)
//存在这个用户的map池子 //存在这个用户的map池子
if mapUserUniqueId, ok := mapUserConnPool[userKey]; ok { if mapUserUniqueId, ok := mapUserConnPool[userKey]; ok {
mapUserUniqueId[data.uniqueId] = struct{}{} mapUserUniqueId[data.uniqueId] = struct{}{}
@ -119,7 +114,7 @@ func ConsumeUserConnPoolCtlChanData(ctx context.Context) {
mapUserConnPool[userKey][data.uniqueId] = struct{}{} mapUserConnPool[userKey][data.uniqueId] = struct{}{}
} }
case 0: //删除 case 0: //删除
logx.Info("删除用户id索引标识：", data.uniqueId) //logx.Info("删除用户id索引标识：", data.uniqueId)
if mapUserUniqueId, ok := mapUserConnPool[userKey]; ok { if mapUserUniqueId, ok := mapUserConnPool[userKey]; ok {
delete(mapUserUniqueId, data.uniqueId) delete(mapUserUniqueId, data.uniqueId)
} }