mirror of
https://github.com/diamondburned/arikawa.git
synced 2024-12-11 07:54:58 +00:00
diamondburned
c6679dc52c
This commit refactors the Store interface in State into smaller interfaces in package store. These interfaces are combined into one structure called a "Cabinet". The default implementation of those interfaces have been rewritten in package defaultstore, while the old no-op implementation stays with the store package. This commit also omitted several state handlers for user events, as it is unclear what they are actually structured like.
74 lines
1.2 KiB
Go
74 lines
1.2 KiB
Go
package defaultstore
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/diamondburned/arikawa/v2/discord"
|
|
"github.com/diamondburned/arikawa/v2/state/store"
|
|
)
|
|
|
|
type Guild struct {
|
|
mut sync.RWMutex
|
|
guilds map[discord.GuildID]discord.Guild
|
|
}
|
|
|
|
var _ store.GuildStore = (*Guild)(nil)
|
|
|
|
func NewGuild() *Guild {
|
|
return &Guild{
|
|
guilds: map[discord.GuildID]discord.Guild{},
|
|
}
|
|
}
|
|
|
|
func (s *Guild) Reset() error {
|
|
s.mut.Lock()
|
|
defer s.mut.Unlock()
|
|
|
|
s.guilds = map[discord.GuildID]discord.Guild{}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *Guild) Guild(id discord.GuildID) (*discord.Guild, error) {
|
|
s.mut.RLock()
|
|
defer s.mut.RUnlock()
|
|
|
|
ch, ok := s.guilds[id]
|
|
if !ok {
|
|
return nil, store.ErrNotFound
|
|
}
|
|
|
|
// implicit copy
|
|
return &ch, nil
|
|
}
|
|
|
|
func (s *Guild) Guilds() ([]discord.Guild, error) {
|
|
s.mut.RLock()
|
|
defer s.mut.RUnlock()
|
|
|
|
if len(s.guilds) == 0 {
|
|
return nil, store.ErrNotFound
|
|
}
|
|
|
|
var gs = make([]discord.Guild, 0, len(s.guilds))
|
|
for _, g := range s.guilds {
|
|
gs = append(gs, g)
|
|
}
|
|
|
|
return gs, nil
|
|
}
|
|
|
|
func (s *Guild) GuildSet(guild discord.Guild) error {
|
|
s.mut.Lock()
|
|
s.guilds[guild.ID] = guild
|
|
s.mut.Unlock()
|
|
return nil
|
|
}
|
|
|
|
func (s *Guild) GuildRemove(id discord.GuildID) error {
|
|
s.mut.Lock()
|
|
delete(s.guilds, id)
|
|
s.mut.Unlock()
|
|
return nil
|
|
}
|