63 lines
1.2 KiB
Go
63 lines
1.2 KiB
Go
package userstats
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
type UserStats struct {
|
|
Username string `json:"username"`
|
|
MessageCount int `json:"message_count"`
|
|
LastActive time.Time `json:"last_active"`
|
|
IsMod bool `json:"is_mod"`
|
|
IsVip bool `json:"is_vip"`
|
|
IsSubscriber bool `json:"is_subscriber"`
|
|
IsMarked bool `json:"is_marked"`
|
|
LastMarkedDate time.Time `json:"-"`
|
|
}
|
|
|
|
type Store struct {
|
|
mu sync.RWMutex
|
|
users map[string]*UserStats // key = username (lowercase)
|
|
}
|
|
|
|
func NewStore() *Store {
|
|
return &Store{
|
|
users: make(map[string]*UserStats),
|
|
}
|
|
}
|
|
|
|
func (s *Store) GetOrCreate(username string) *UserStats {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
lower := username
|
|
if u, ok := s.users[lower]; ok {
|
|
return u
|
|
}
|
|
u := &UserStats{Username: username}
|
|
s.users[lower] = u
|
|
return u
|
|
}
|
|
|
|
func (s *Store) Update(username string, fn func(*UserStats)) {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
lower := username
|
|
u, ok := s.users[lower]
|
|
if !ok {
|
|
u = &UserStats{Username: username}
|
|
s.users[lower] = u
|
|
}
|
|
fn(u)
|
|
}
|
|
|
|
func (s *Store) GetAll() []*UserStats {
|
|
s.mu.RLock()
|
|
defer s.mu.RUnlock()
|
|
res := make([]*UserStats, 0, len(s.users))
|
|
for _, u := range s.users {
|
|
res = append(res, u)
|
|
}
|
|
return res
|
|
}
|