blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-04-06 04:15:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: push log add

Browse Source
This commit is contained in:
Gordon 2023-06-06 17:37:23 +08:00
parent f661245d9f
commit 35ceb7542a
2 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internal/msggateway/n_ws_server.go
View File

@ -124,11 +124,13 @@ func (ws *WsServer) registerClient(client *Client) {
)
cli, userOK, clientOK = ws.clients.Get(client.userID, client.platformID)
if !userOK {
log.ZDebug(client.ctx, "user not exist", "userID", client.userID, "platformID", client.platformID)
ws.clients.Set(client.userID, client)
atomic.AddInt64(&ws.onlineUserNum, 1)
atomic.AddInt64(&ws.onlineUserConnNum, 1)
} else {
log.ZDebug(client.ctx, "user exist", "userID", client.userID, "platformID", client.platformID)
if clientOK { //已经有同平台的连接存在
ws.clients.Set(client.userID, client)
ws.multiTerminalLoginChecker(cli)

8
internal/msggateway/user_map.go
View File

@ -1,6 +1,10 @@
package msggateway
import "sync"
import (
"context"
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/log"
"sync"
)
type UserMap struct {
m sync.Map
@ -36,10 +40,12 @@ func (u *UserMap) Get(key string, platformID int) ([]*Client, bool, bool) {
func (u *UserMap) Set(key string, v *Client) {
allClients, existed := u.m.Load(key)
if existed {
log.ZDebug(context.Background(), "Set existed", "user_id", key, "client", v)
oldClients := allClients.([]*Client)
oldClients = append(oldClients, v)
u.m.Store(key, oldClients)
} else {
log.ZDebug(context.Background(), "Set not existed", "user_id", key, "client", v)
var clients []*Client
clients = append(clients, v)
u.m.Store(key, clients)