mirror of
https://github.com/hanzoai/team.git
synced 2026-10-30 14:49:35 +00:00
zeekay
8feb24ea1f
The server module named github.com/hanzoai/team/server/v8 — a path that resolves to the retired workspace's repository, not this one — while server/public still published github.com/mattermost/mattermost/server/public. Both are github.com/hanzoteam/server/... now, which is where this code actually lives. That second half is what made the binary unbootable. Upstream's public path stayed claimed by upstream, mattermost-plugin-ai imported it, and the build linked public/model twice; public/plugin's gob.Register then ran twice under one name and panicked at init, before main. Reverting the rename traded the brand for a working binary. Rebranding the plugin alongside it needs no such trade: the agents fork is github.com/hanzoteam/agents/v2 against our public module, so no module named mattermost/mattermost is in the graph at all. Measured: `go list -deps ./cmd/mattermost` names exactly one public/model, and it is ours.
140 lines
3.9 KiB
Go
140 lines
3.9 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package localcachelayer
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
|
|
"github.com/hanzoteam/server/server/public/model"
|
|
"github.com/hanzoteam/server/server/v8/channels/store"
|
|
)
|
|
|
|
type LocalCacheFileInfoStore struct {
|
|
store.FileInfoStore
|
|
rootStore *LocalCacheStore
|
|
}
|
|
|
|
func (s *LocalCacheFileInfoStore) handleClusterInvalidateFileInfo(msg *model.ClusterMessage) {
|
|
if bytes.Equal(msg.Data, clearCacheMessageData) {
|
|
s.rootStore.fileInfoCache.Purge()
|
|
return
|
|
}
|
|
s.rootStore.fileInfoCache.Remove(string(msg.Data))
|
|
}
|
|
|
|
func (s LocalCacheFileInfoStore) GetForPost(postId string, readFromMaster, includeDeleted, allowFromCache bool) ([]*model.FileInfo, error) {
|
|
if !allowFromCache {
|
|
return s.FileInfoStore.GetForPost(postId, readFromMaster, includeDeleted, allowFromCache)
|
|
}
|
|
|
|
cacheKey := postId
|
|
if includeDeleted {
|
|
cacheKey += "_deleted"
|
|
}
|
|
|
|
var fileInfo []*model.FileInfo
|
|
if err := s.rootStore.doStandardReadCache(s.rootStore.fileInfoCache, cacheKey, &fileInfo); err == nil {
|
|
return fileInfo, nil
|
|
}
|
|
|
|
fileInfos, err := s.FileInfoStore.GetForPost(postId, readFromMaster, includeDeleted, allowFromCache)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if len(fileInfos) > 0 {
|
|
s.rootStore.doStandardAddToCache(s.rootStore.fileInfoCache, cacheKey, fileInfos)
|
|
}
|
|
|
|
return fileInfos, nil
|
|
}
|
|
|
|
func (s LocalCacheFileInfoStore) GetByIds(ids []string, includeDeleted, allowFromCache, readFromMaster bool) ([]*model.FileInfo, error) {
|
|
if readFromMaster {
|
|
return s.FileInfoStore.GetByIds(ids, includeDeleted, false, readFromMaster)
|
|
}
|
|
|
|
if !allowFromCache {
|
|
return s.FileInfoStore.GetByIds(ids, includeDeleted, allowFromCache, readFromMaster)
|
|
}
|
|
|
|
var fileIdsToFetch []string
|
|
var fileInfos []*model.FileInfo
|
|
|
|
for _, fileId := range ids {
|
|
cacheKey := fmt.Sprintf("%s_%t", fileId, includeDeleted)
|
|
|
|
var fileInfo *model.FileInfo
|
|
if err := s.rootStore.doStandardReadCache(s.rootStore.fileInfoCache, cacheKey, &fileInfo); err == nil {
|
|
fileInfos = append(fileInfos, fileInfo)
|
|
} else {
|
|
fileIdsToFetch = append(fileIdsToFetch, fileId)
|
|
}
|
|
}
|
|
|
|
if len(fileIdsToFetch) > 0 {
|
|
fetchedFileInfos, err := s.FileInfoStore.GetByIds(fileIdsToFetch, includeDeleted, false, readFromMaster)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, fileInfo := range fetchedFileInfos {
|
|
cacheKey := fmt.Sprintf("%s_%t", fileInfo.Id, includeDeleted)
|
|
s.rootStore.doStandardAddToCache(s.rootStore.fileInfoCache, cacheKey, fileInfo)
|
|
fileInfos = append(fileInfos, fileInfo)
|
|
}
|
|
}
|
|
|
|
return fileInfos, nil
|
|
}
|
|
|
|
func (s LocalCacheFileInfoStore) ClearCaches() {
|
|
s.rootStore.fileInfoCache.Purge()
|
|
if s.rootStore.metrics != nil {
|
|
s.rootStore.metrics.IncrementMemCacheInvalidationCounter(s.rootStore.fileInfoCache.Name())
|
|
}
|
|
}
|
|
|
|
func (s LocalCacheFileInfoStore) InvalidateFileInfosForPostCache(postId string, deleted bool) {
|
|
cacheKey := postId
|
|
if deleted {
|
|
cacheKey += "_deleted"
|
|
}
|
|
s.rootStore.doInvalidateCacheCluster(s.rootStore.fileInfoCache, cacheKey, nil)
|
|
if s.rootStore.metrics != nil {
|
|
s.rootStore.metrics.IncrementMemCacheInvalidationCounter(s.rootStore.fileInfoCache.Name())
|
|
}
|
|
}
|
|
|
|
func (s LocalCacheFileInfoStore) GetStorageUsage(allowFromCache, includeDeleted bool) (int64, error) {
|
|
storageUsageKey := "storage_usage"
|
|
if includeDeleted {
|
|
storageUsageKey += "_deleted"
|
|
}
|
|
|
|
if !allowFromCache {
|
|
usage, err := s.FileInfoStore.GetStorageUsage(allowFromCache, includeDeleted)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
s.rootStore.doStandardAddToCache(s.rootStore.fileInfoCache, storageUsageKey, usage)
|
|
return usage, nil
|
|
}
|
|
|
|
var usage int64
|
|
if err := s.rootStore.doStandardReadCache(s.rootStore.fileInfoCache, storageUsageKey, &usage); err == nil {
|
|
return usage, nil
|
|
}
|
|
|
|
usage, err := s.FileInfoStore.GetStorageUsage(allowFromCache, includeDeleted)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
s.rootStore.doStandardAddToCache(s.rootStore.fileInfoCache, storageUsageKey, usage)
|
|
return usage, nil
|
|
}
|