You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
1.4 KiB
66 lines
1.4 KiB
package new
|
|
|
|
import "sync"
|
|
|
|
type UserMap struct {
|
|
m sync.Map
|
|
}
|
|
|
|
func newUserMap() *UserMap {
|
|
return &UserMap{}
|
|
}
|
|
func (u *UserMap) GetAll(key string) ([]*Client, bool) {
|
|
allClients, ok := u.m.Load(key)
|
|
if ok {
|
|
return allClients.([]*Client), ok
|
|
}
|
|
return nil, ok
|
|
}
|
|
func (u *UserMap) Get(key string, platformID int) (*Client, bool, bool) {
|
|
allClients, userExisted := u.m.Load(key)
|
|
if userExisted {
|
|
for _, client := range allClients.([]*Client) {
|
|
if client.platformID == platformID {
|
|
return client, userExisted, true
|
|
}
|
|
}
|
|
return nil, userExisted, false
|
|
}
|
|
return nil, userExisted, false
|
|
}
|
|
func (u *UserMap) Set(key string, v *Client) {
|
|
allClients, existed := u.m.Load(key)
|
|
if existed {
|
|
oldClients := allClients.([]*Client)
|
|
oldClients = append(oldClients, v)
|
|
u.m.Store(key, oldClients)
|
|
} else {
|
|
clients := make([]*Client, 3)
|
|
clients = append(clients, v)
|
|
u.m.Store(key, clients)
|
|
}
|
|
}
|
|
func (u *UserMap) delete(key string, platformID int) (isDeleteUser bool) {
|
|
allClients, existed := u.m.Load(key)
|
|
if existed {
|
|
oldClients := allClients.([]*Client)
|
|
a := make([]*Client, 3)
|
|
for _, client := range oldClients {
|
|
if client.platformID != platformID {
|
|
a = append(a, client)
|
|
}
|
|
}
|
|
if len(a) == 0 {
|
|
u.m.Delete(key)
|
|
return true
|
|
} else {
|
|
u.m.Store(key, a)
|
|
return false
|
|
}
|
|
}
|
|
return existed
|
|
}
|
|
func (u *UserMap) DeleteAll(key string) {
|
|
u.m.Delete(key)
|
|
}
|