mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-04-04 21:09:51 +08:00
9ed6200e45
* pb * fix: Modifying other fields while setting IsPrivateChat does not take effect * fix: quote message error revoke * refactoring scheduled tasks * refactoring scheduled tasks * refactoring scheduled tasks * refactoring scheduled tasks * refactoring scheduled tasks * refactoring scheduled tasks * upgrading pkg tools * fix * fix * optimize log output * feat: support GetLastMessage * feat: support GetLastMessage * feat: s3 switch * feat: s3 switch * fix: GetUsersOnline * feat: SendBusinessNotification supported configuration parameters * feat: SendBusinessNotification supported configuration parameters * feat: SendBusinessNotification supported configuration parameters * feat: seq conversion failed without exiting * monolithic * fix: DeleteDoc crash * fix: DeleteDoc crash * fix: monolithic * fix: monolithic * fix: fill send time * fix: fill send time * fix: crash caused by withdrawing messages from users who have left the group * fix: mq * fix: mq * fix: user msg timestamp * fix: mq * 1 * 1 * 1 * 1 * 1 * 1 * 1 * seq read config * seq read config * 1 * 1 * fix: the source message of the reference is withdrawn, and the referenced message is deleted * 1 * 1 * 1 * 1 * 1 * 1 * 1 * 1 * 1 * 1 * 1 * 1 * 1 * 1
83 lines
1.6 KiB
Go
83 lines
1.6 KiB
Go
package mcache
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/storage/cache"
|
|
)
|
|
|
|
var (
|
|
globalOnlineCache cache.OnlineCache
|
|
globalOnlineOnce sync.Once
|
|
)
|
|
|
|
func NewOnlineCache() cache.OnlineCache {
|
|
globalOnlineOnce.Do(func() {
|
|
globalOnlineCache = &onlineCache{
|
|
user: make(map[string]map[int32]struct{}),
|
|
}
|
|
})
|
|
return globalOnlineCache
|
|
}
|
|
|
|
type onlineCache struct {
|
|
lock sync.RWMutex
|
|
user map[string]map[int32]struct{}
|
|
}
|
|
|
|
func (x *onlineCache) GetOnline(ctx context.Context, userID string) ([]int32, error) {
|
|
x.lock.RLock()
|
|
defer x.lock.RUnlock()
|
|
pSet, ok := x.user[userID]
|
|
if !ok {
|
|
return nil, nil
|
|
}
|
|
res := make([]int32, 0, len(pSet))
|
|
for k := range pSet {
|
|
res = append(res, k)
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
func (x *onlineCache) SetUserOnline(ctx context.Context, userID string, online, offline []int32) error {
|
|
x.lock.Lock()
|
|
defer x.lock.Unlock()
|
|
pSet, ok := x.user[userID]
|
|
if ok {
|
|
for _, p := range offline {
|
|
delete(pSet, p)
|
|
}
|
|
}
|
|
if len(online) > 0 {
|
|
if !ok {
|
|
pSet = make(map[int32]struct{})
|
|
x.user[userID] = pSet
|
|
}
|
|
for _, p := range online {
|
|
pSet[p] = struct{}{}
|
|
}
|
|
}
|
|
if len(pSet) == 0 {
|
|
delete(x.user, userID)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (x *onlineCache) GetAllOnlineUsers(ctx context.Context, cursor uint64) (map[string][]int32, uint64, error) {
|
|
if cursor != 0 {
|
|
return nil, 0, nil
|
|
}
|
|
x.lock.RLock()
|
|
defer x.lock.RUnlock()
|
|
res := make(map[string][]int32)
|
|
for k, v := range x.user {
|
|
pSet := make([]int32, 0, len(v))
|
|
for p := range v {
|
|
pSet = append(pSet, p)
|
|
}
|
|
res[k] = pSet
|
|
}
|
|
return res, 0, nil
|
|
}
|