liulei/jx-callback
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

aa

Browse Source
This commit is contained in:
suyl
2021-07-29 14:53:07 +08:00
parent 17dfe9e9df
commit bc299677ad
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
business/jxstore/event/event.go
View File

@@ -45,7 +45,7 @@ func init() {
clientsHeart := make(map[string]*websocket.Conn) clientsHeart := make(map[string]*websocket.Conn)
wsClient.Clients = clients wsClient.Clients = clients
wsClient.ClientsHeart = clientsHeart wsClient.ClientsHeart = clientsHeart
wsClient = new(sync.RWMutex) wsClient.s = new(sync.RWMutex)
go handleMessages() go handleMessages()
} }
@@ -100,7 +100,7 @@ func ImMessage(userID string, ws *websocket.Conn) (err error) {
if len(messageGroups) == 0 { if len(messageGroups) == 0 {
return return
} }
wsClient.Lock() wsClient.s.Lock()
clientUser[userID] = ws clientUser[userID] = ws
wsClient.ClientsHeart[userID] = ws wsClient.ClientsHeart[userID] = ws
for _, v := range messageGroups { for _, v := range messageGroups {
@@ -110,7 +110,7 @@ func ImMessage(userID string, ws *websocket.Conn) (err error) {
wsClient.Clients[v.GroupID] = clientUser wsClient.Clients[v.GroupID] = clientUser
} }
} }
wsClient.Unlock() wsClient.s.Unlock()
globals.SugarLogger.Debugf("userID :%v ,clients :%v", userID, utils.Format4Output(wsClient.Clients, false)) globals.SugarLogger.Debugf("userID :%v ,clients :%v", userID, utils.Format4Output(wsClient.Clients, false))
var s *model.ImMessageRecord var s *model.ImMessageRecord
@@ -173,9 +173,9 @@ func ImMessage(userID string, ws *websocket.Conn) (err error) {
} }
func GetOnlineUserCount() (count int) { func GetOnlineUserCount() (count int) {
wsClient.RLock() wsClient.s.RLock()
count = len(wsClient.ClientsHeart) count = len(wsClient.ClientsHeart)
wsClient.RUnlock() wsClient.s.RUnlock()
return count return count
} }