7679f51e3e
We use the same name for hit/miss/invalidation metrics. The downside is that now we have conflated a full cache purge and a single cache key invalidation into a single label, but the advantage is that it's easier to monitor the usage of a single cache. And moreover, duplicating the same string across multiple places increases the chances of making a mistake. Centralizing the cache name only at instantiation simplifies the code. After this, we would be ready to use Redis. https://mattermost.atlassian.net/browse/MM-57193 ```release-note NONE ```
84 строки
2.3 KiB
Go
84 строки
2.3 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package localcachelayer
|
|
|
|
import (
|
|
"bytes"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
"github.com/mattermost/mattermost/server/v8/channels/store"
|
|
)
|
|
|
|
type LocalCacheTeamStore struct {
|
|
store.TeamStore
|
|
rootStore *LocalCacheStore
|
|
}
|
|
|
|
func (s *LocalCacheTeamStore) handleClusterInvalidateTeam(msg *model.ClusterMessage) {
|
|
if bytes.Equal(msg.Data, clearCacheMessageData) {
|
|
s.rootStore.teamAllTeamIdsForUserCache.Purge()
|
|
} else {
|
|
s.rootStore.teamAllTeamIdsForUserCache.Remove(string(msg.Data))
|
|
}
|
|
}
|
|
|
|
func (s LocalCacheTeamStore) ClearCaches() {
|
|
s.rootStore.teamAllTeamIdsForUserCache.Purge()
|
|
if s.rootStore.metrics != nil {
|
|
s.rootStore.metrics.IncrementMemCacheInvalidationCounter(s.rootStore.teamAllTeamIdsForUserCache.Name())
|
|
}
|
|
}
|
|
|
|
func (s LocalCacheTeamStore) InvalidateAllTeamIdsForUser(userId string) {
|
|
s.rootStore.doInvalidateCacheCluster(s.rootStore.teamAllTeamIdsForUserCache, userId, nil)
|
|
if s.rootStore.metrics != nil {
|
|
s.rootStore.metrics.IncrementMemCacheInvalidationCounter(s.rootStore.teamAllTeamIdsForUserCache.Name())
|
|
}
|
|
}
|
|
|
|
func (s LocalCacheTeamStore) GetUserTeamIds(userID string, allowFromCache bool) ([]string, error) {
|
|
if !allowFromCache {
|
|
return s.TeamStore.GetUserTeamIds(userID, allowFromCache)
|
|
}
|
|
|
|
var userTeamIds []string
|
|
if err := s.rootStore.doStandardReadCache(s.rootStore.teamAllTeamIdsForUserCache, userID, &userTeamIds); err == nil {
|
|
return userTeamIds, nil
|
|
}
|
|
|
|
userTeamIds, err := s.TeamStore.GetUserTeamIds(userID, allowFromCache)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if len(userTeamIds) > 0 {
|
|
s.rootStore.doStandardAddToCache(s.rootStore.teamAllTeamIdsForUserCache, userID, userTeamIds)
|
|
}
|
|
|
|
return userTeamIds, nil
|
|
}
|
|
|
|
func (s LocalCacheTeamStore) Update(team *model.Team) (*model.Team, error) {
|
|
var oldTeam *model.Team
|
|
var err error
|
|
if team.DeleteAt != 0 {
|
|
oldTeam, err = s.TeamStore.Get(team.Id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
tm, err := s.TeamStore.Update(team)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer s.rootStore.doClearCacheCluster(s.rootStore.rolePermissionsCache)
|
|
|
|
if oldTeam != nil && oldTeam.DeleteAt == 0 {
|
|
s.rootStore.doClearCacheCluster(s.rootStore.teamAllTeamIdsForUserCache)
|
|
}
|
|
|
|
return tm, err
|
|
}
|