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, } }