mirror of
https://github.com/hanzoai/team.git
synced 2026-11-01 07:55:31 +00:00
Devin BinnieandGitHub
d629c01e61
* [MM-68648] Implement GetForGroup to get fields in the Property System, add caching for fields * Re-add CRUD functions for the cache to invalidate when updates happen
97 lines
3.1 KiB
Go
97 lines
3.1 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package localcachelayer
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
"github.com/mattermost/mattermost/server/public/shared/mlog"
|
|
"github.com/mattermost/mattermost/server/v8/channels/store"
|
|
)
|
|
|
|
func (s LocalCachePropertyFieldStore) Create(field *model.PropertyField) (*model.PropertyField, error) {
|
|
created, err := s.PropertyFieldStore.Create(field)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
s.InvalidateFieldsForGroup(created.GroupID)
|
|
return created, nil
|
|
}
|
|
|
|
func (s LocalCachePropertyFieldStore) Update(groupID string, fields []*model.PropertyField, expectedUpdateAts map[string]int64) ([]*model.PropertyField, error) {
|
|
updated, err := s.PropertyFieldStore.Update(groupID, fields, expectedUpdateAts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// The returned slice includes both the requested fields and any linked
|
|
// fields the store propagated to. Invalidate each distinct group so all
|
|
// affected GetForGroup caches are cleared.
|
|
invalidated := make(map[string]bool, len(updated))
|
|
for _, field := range updated {
|
|
if invalidated[field.GroupID] {
|
|
continue
|
|
}
|
|
invalidated[field.GroupID] = true
|
|
s.InvalidateFieldsForGroup(field.GroupID)
|
|
}
|
|
return updated, nil
|
|
}
|
|
|
|
func (s LocalCachePropertyFieldStore) Delete(groupID string, id string) error {
|
|
if err := s.PropertyFieldStore.Delete(groupID, id); err != nil {
|
|
return err
|
|
}
|
|
|
|
s.InvalidateFieldsForGroup(groupID)
|
|
return nil
|
|
}
|
|
|
|
type LocalCachePropertyFieldStore struct {
|
|
store.PropertyFieldStore
|
|
rootStore *LocalCacheStore
|
|
}
|
|
|
|
func (s *LocalCachePropertyFieldStore) handleClusterInvalidatePropertyField(msg *model.ClusterMessage) {
|
|
if bytes.Equal(msg.Data, clearCacheMessageData) {
|
|
if err := s.rootStore.propertyFieldCache.Purge(); err != nil {
|
|
s.rootStore.logger.Warn("failed to purge property field cache", mlog.Err(err))
|
|
}
|
|
} else if err := s.rootStore.propertyFieldCache.Remove(string(msg.Data)); err != nil {
|
|
s.rootStore.logger.Warn("failed to remove property field cache entry", mlog.Err(err))
|
|
}
|
|
}
|
|
|
|
func (s LocalCachePropertyFieldStore) InvalidateFieldsForGroup(groupID string) {
|
|
s.rootStore.doInvalidateCacheCluster(s.rootStore.propertyFieldCache, groupID, nil)
|
|
if s.rootStore.metrics != nil {
|
|
s.rootStore.metrics.IncrementMemCacheInvalidationCounter(s.rootStore.propertyFieldCache.Name())
|
|
}
|
|
}
|
|
|
|
func (s *LocalCachePropertyFieldStore) getFieldsForGroupFromCache(groupID string) ([]*model.PropertyField, bool) {
|
|
var fields []*model.PropertyField
|
|
if err := s.rootStore.doStandardReadCache(s.rootStore.propertyFieldCache, groupID, &fields); err == nil {
|
|
return fields, true
|
|
}
|
|
return nil, false
|
|
}
|
|
|
|
func (s LocalCachePropertyFieldStore) GetForGroup(ctx context.Context, groupID string) ([]*model.PropertyField, error) {
|
|
if fields, ok := s.getFieldsForGroupFromCache(groupID); ok {
|
|
return fields, nil
|
|
}
|
|
|
|
fields, err := s.PropertyFieldStore.GetForGroup(ctx, groupID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
s.rootStore.doStandardAddToCache(s.rootStore.propertyFieldCache, groupID, fields)
|
|
return fields, nil
|
|
}
|