forked from pneymrl2f/nightingale
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
42 lines
739 B
42 lines
739 B
package cache
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/didi/nightingale/v5/models"
|
|
)
|
|
|
|
type UserGroupMap struct {
|
|
sync.RWMutex
|
|
Data map[int64]*models.UserGroup
|
|
}
|
|
|
|
var UserGroupCache = &UserGroupMap{Data: make(map[int64]*models.UserGroup)}
|
|
|
|
func (s *UserGroupMap) GetBy(id int64) *models.UserGroup {
|
|
s.RLock()
|
|
defer s.RUnlock()
|
|
|
|
return s.Data[id]
|
|
}
|
|
|
|
func (s *UserGroupMap) GetByIds(ids []int64) []*models.UserGroup {
|
|
s.RLock()
|
|
defer s.RUnlock()
|
|
var userGroups []*models.UserGroup
|
|
for _, id := range ids {
|
|
if s.Data[id] == nil {
|
|
continue
|
|
}
|
|
userGroups = append(userGroups, s.Data[id])
|
|
}
|
|
|
|
return userGroups
|
|
}
|
|
|
|
func (s *UserGroupMap) SetAll(userGroups map[int64]*models.UserGroup) {
|
|
s.Lock()
|
|
defer s.Unlock()
|
|
s.Data = userGroups
|
|
}
|