telegabber/telegram/cache/cache.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,
}
}