136 lines
2.9 KiB
Go
136 lines
2.9 KiB
Go
package cache
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/zelenin/go-tdlib/client"
|
|
)
|
|
|
|
// Status stores formatted data for XMPP presence
|
|
type Status struct {
|
|
ID int64
|
|
XMPP string
|
|
Description string
|
|
}
|
|
|
|
// Cache allows operating the chats and users cache in
|
|
// a thread-safe manner
|
|
type Cache struct {
|
|
chats map[int64]*client.Chat
|
|
users map[int32]*client.User
|
|
statuses map[int64]*Status
|
|
chatsLock sync.Mutex
|
|
usersLock sync.Mutex
|
|
statusesLock sync.Mutex
|
|
}
|
|
|
|
// NewCache initializes a cache
|
|
func NewCache() *Cache {
|
|
return &Cache{
|
|
chats: map[int64]*client.Chat{},
|
|
users: map[int32]*client.User{},
|
|
statuses: map[int64]*Status{},
|
|
}
|
|
}
|
|
|
|
// ChatsKeys grabs chat ids synchronously to avoid lockups
|
|
// while they are used
|
|
func (cache *Cache) ChatsKeys() []int64 {
|
|
cache.chatsLock.Lock()
|
|
defer cache.chatsLock.Unlock()
|
|
|
|
var keys []int64
|
|
for id := range cache.chats {
|
|
keys = append(keys, id)
|
|
}
|
|
return keys
|
|
}
|
|
|
|
// UsersKeys grabs user ids synchronously to avoid lockups
|
|
// while they are used
|
|
func (cache *Cache) UsersKeys() []int32 {
|
|
cache.usersLock.Lock()
|
|
defer cache.usersLock.Unlock()
|
|
|
|
var keys []int32
|
|
for id := range cache.users {
|
|
keys = append(keys, id)
|
|
}
|
|
return keys
|
|
}
|
|
|
|
// StatusesRange loops through the map in a thread-safe manner
|
|
func (cache *Cache) StatusesRange() chan *Status {
|
|
cache.statusesLock.Lock()
|
|
|
|
statusChan := make(chan *Status, 1)
|
|
|
|
go func() {
|
|
defer func() {
|
|
cache.statusesLock.Unlock()
|
|
close(statusChan)
|
|
}()
|
|
|
|
for _, status := range cache.statuses {
|
|
statusChan <- status
|
|
}
|
|
}()
|
|
|
|
return statusChan
|
|
}
|
|
|
|
// GetChat retrieves chat by id if it's present in the cache
|
|
func (cache *Cache) GetChat(id int64) (*client.Chat, bool) {
|
|
cache.chatsLock.Lock()
|
|
defer cache.chatsLock.Unlock()
|
|
|
|
chat, ok := cache.chats[id]
|
|
return chat, ok
|
|
}
|
|
|
|
// GetUser retrieves user by id if it's present in the cache
|
|
func (cache *Cache) GetUser(id int32) (*client.User, bool) {
|
|
cache.usersLock.Lock()
|
|
defer cache.usersLock.Unlock()
|
|
|
|
user, ok := cache.users[id]
|
|
return user, ok
|
|
}
|
|
|
|
// GetStatus retrieves status by id if it's present in the cache
|
|
func (cache *Cache) GetStatus(id int64) (*Status, bool) {
|
|
cache.statusesLock.Lock()
|
|
defer cache.statusesLock.Unlock()
|
|
|
|
status, ok := cache.statuses[id]
|
|
return status, ok
|
|
}
|
|
|
|
// SetChat stores a chat in the cache
|
|
func (cache *Cache) SetChat(id int64, chat *client.Chat) {
|
|
cache.chatsLock.Lock()
|
|
defer cache.chatsLock.Unlock()
|
|
|
|
cache.chats[id] = chat
|
|
}
|
|
|
|
// SetUser stores a user in the cache
|
|
func (cache *Cache) SetUser(id int32, user *client.User) {
|
|
cache.usersLock.Lock()
|
|
defer cache.usersLock.Unlock()
|
|
|
|
cache.users[id] = user
|
|
}
|
|
|
|
// SetStatus stores a status in the cache
|
|
func (cache *Cache) SetStatus(id int64, show string, status string) {
|
|
cache.statusesLock.Lock()
|
|
defer cache.statusesLock.Unlock()
|
|
|
|
cache.statuses[id] = &Status{
|
|
ID: id,
|
|
XMPP: show,
|
|
Description: status,
|
|
}
|
|
}
|