mirror of
https://github.com/hanzoai/team.git
synced 2026-10-30 14:49:35 +00:00
4fc75801be
There were 3 remaining caches which were there in the store layer. We migrate them to make the store layer fully free from any caches. https://mattermost.atlassian.net/browse/MM-56879 ```release-note NONE ``` Co-authored-by: Mattermost Build <[email protected]>
118 lines
3.4 KiB
Go
118 lines
3.4 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package localcachelayer
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"sort"
|
|
"strings"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
"github.com/mattermost/mattermost/server/v8/channels/store"
|
|
)
|
|
|
|
type LocalCacheRoleStore struct {
|
|
store.RoleStore
|
|
rootStore *LocalCacheStore
|
|
}
|
|
|
|
func (s *LocalCacheRoleStore) handleClusterInvalidateRole(msg *model.ClusterMessage) {
|
|
if bytes.Equal(msg.Data, clearCacheMessageData) {
|
|
s.rootStore.roleCache.Purge()
|
|
} else {
|
|
s.rootStore.roleCache.Remove(string(msg.Data))
|
|
}
|
|
}
|
|
|
|
func (s *LocalCacheRoleStore) handleClusterInvalidateRolePermissions(msg *model.ClusterMessage) {
|
|
if bytes.Equal(msg.Data, clearCacheMessageData) {
|
|
s.rootStore.rolePermissionsCache.Purge()
|
|
} else {
|
|
s.rootStore.rolePermissionsCache.Remove(string(msg.Data))
|
|
}
|
|
}
|
|
|
|
func (s LocalCacheRoleStore) Save(role *model.Role) (*model.Role, error) {
|
|
if role.Name != "" {
|
|
defer s.rootStore.doInvalidateCacheCluster(s.rootStore.roleCache, role.Name, nil)
|
|
defer s.rootStore.doClearCacheCluster(s.rootStore.rolePermissionsCache)
|
|
}
|
|
return s.RoleStore.Save(role)
|
|
}
|
|
|
|
func (s LocalCacheRoleStore) GetByName(ctx context.Context, name string) (*model.Role, error) {
|
|
var role *model.Role
|
|
if err := s.rootStore.doStandardReadCache(s.rootStore.roleCache, name, &role); err == nil {
|
|
return role, nil
|
|
}
|
|
|
|
role, err := s.RoleStore.GetByName(ctx, name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
s.rootStore.doStandardAddToCache(s.rootStore.roleCache, name, role)
|
|
return role, nil
|
|
}
|
|
|
|
func (s LocalCacheRoleStore) GetByNames(names []string) ([]*model.Role, error) {
|
|
var foundRoles []*model.Role
|
|
var rolesToQuery []string
|
|
|
|
for _, roleName := range names {
|
|
var role *model.Role
|
|
if err := s.rootStore.doStandardReadCache(s.rootStore.roleCache, roleName, &role); err == nil {
|
|
foundRoles = append(foundRoles, role)
|
|
} else {
|
|
rolesToQuery = append(rolesToQuery, roleName)
|
|
}
|
|
}
|
|
|
|
roles, err := s.RoleStore.GetByNames(rolesToQuery)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, role := range roles {
|
|
s.rootStore.doStandardAddToCache(s.rootStore.roleCache, role.Name, role)
|
|
}
|
|
|
|
return append(foundRoles, roles...), nil
|
|
}
|
|
|
|
func (s LocalCacheRoleStore) Delete(roleId string) (*model.Role, error) {
|
|
role, err := s.RoleStore.Delete(roleId)
|
|
|
|
if err == nil {
|
|
s.rootStore.doInvalidateCacheCluster(s.rootStore.roleCache, role.Name, nil)
|
|
defer s.rootStore.doClearCacheCluster(s.rootStore.rolePermissionsCache)
|
|
}
|
|
return role, err
|
|
}
|
|
|
|
func (s LocalCacheRoleStore) PermanentDeleteAll() error {
|
|
defer s.rootStore.roleCache.Purge()
|
|
defer s.rootStore.doClearCacheCluster(s.rootStore.roleCache)
|
|
defer s.rootStore.doClearCacheCluster(s.rootStore.rolePermissionsCache)
|
|
|
|
return s.RoleStore.PermanentDeleteAll()
|
|
}
|
|
|
|
func (s LocalCacheRoleStore) ChannelHigherScopedPermissions(roleNames []string) (map[string]*model.RolePermissions, error) {
|
|
sort.Strings(roleNames)
|
|
cacheKey := strings.Join(roleNames, "/")
|
|
var rolePermissionsMap map[string]*model.RolePermissions
|
|
if err := s.rootStore.doStandardReadCache(s.rootStore.rolePermissionsCache, cacheKey, &rolePermissionsMap); err == nil {
|
|
return rolePermissionsMap, nil
|
|
}
|
|
|
|
rolePermissionsMap, err := s.RoleStore.ChannelHigherScopedPermissions(roleNames)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
s.rootStore.doStandardAddToCache(s.rootStore.rolePermissionsCache, cacheKey, rolePermissionsMap)
|
|
return rolePermissionsMap, nil
|
|
}
|