залил

internal/userstats/userstats.go

@@ -0,0 +1,62 @@
+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
+}