43 lines
622 B
Go
43 lines
622 B
Go
package ws_handler
|
|
|
|
import (
|
|
"sync"
|
|
)
|
|
|
|
var UserMgr *userManager
|
|
|
|
type userManager struct {
|
|
userMap map[int]*Client
|
|
sync.RWMutex
|
|
}
|
|
|
|
func init() {
|
|
UserMgr = &userManager{
|
|
userMap: make(map[int]*Client),
|
|
}
|
|
}
|
|
|
|
func (m *userManager) Add(uid int, client *Client) {
|
|
m.Lock()
|
|
defer m.Unlock()
|
|
m.userMap[uid] = client
|
|
}
|
|
|
|
func (m *userManager) Delete(uid int) {
|
|
m.Lock()
|
|
defer m.Unlock()
|
|
delete(m.userMap, uid)
|
|
}
|
|
|
|
func (m *userManager) GetAll() map[int]*Client {
|
|
m.RLock()
|
|
defer m.RUnlock()
|
|
return m.userMap
|
|
}
|
|
|
|
func (m *userManager) GetByUID(uid int) *Client {
|
|
m.RLock()
|
|
defer m.RUnlock()
|
|
return m.userMap[uid]
|
|
}
|