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.
181 lines
5.1 KiB
Go
181 lines
5.1 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package localcachelayer
|
|
|
|
import (
|
|
"bytes"
|
|
"sync"
|
|
|
|
"github.com/hanzoteam/server/server/public/model"
|
|
"github.com/hanzoteam/server/server/public/shared/request"
|
|
"github.com/hanzoteam/server/server/v8/channels/store"
|
|
"github.com/hanzoteam/server/server/v8/channels/store/sqlstore"
|
|
)
|
|
|
|
type LocalCacheEmojiStore struct {
|
|
store.EmojiStore
|
|
rootStore *LocalCacheStore
|
|
emojiByIdMut sync.Mutex
|
|
emojiByIdInvalidations map[string]bool
|
|
emojiByNameMut sync.Mutex
|
|
emojiByNameInvalidations map[string]bool
|
|
}
|
|
|
|
func (es *LocalCacheEmojiStore) handleClusterInvalidateEmojiById(msg *model.ClusterMessage) {
|
|
if bytes.Equal(msg.Data, clearCacheMessageData) {
|
|
es.rootStore.emojiCacheById.Purge()
|
|
} else {
|
|
es.emojiByIdMut.Lock()
|
|
es.emojiByIdInvalidations[string(msg.Data)] = true
|
|
es.emojiByIdMut.Unlock()
|
|
es.rootStore.emojiCacheById.Remove(string(msg.Data))
|
|
}
|
|
}
|
|
|
|
func (es *LocalCacheEmojiStore) handleClusterInvalidateEmojiIdByName(msg *model.ClusterMessage) {
|
|
if bytes.Equal(msg.Data, clearCacheMessageData) {
|
|
es.rootStore.emojiIdCacheByName.Purge()
|
|
} else {
|
|
es.emojiByNameMut.Lock()
|
|
es.emojiByNameInvalidations[string(msg.Data)] = true
|
|
es.emojiByNameMut.Unlock()
|
|
es.rootStore.emojiIdCacheByName.Remove(string(msg.Data))
|
|
}
|
|
}
|
|
|
|
func (es *LocalCacheEmojiStore) Get(rctx request.CTX, id string, allowFromCache bool) (*model.Emoji, error) {
|
|
if allowFromCache {
|
|
if emoji, ok := es.getFromCacheById(id); ok {
|
|
return emoji, nil
|
|
}
|
|
}
|
|
|
|
// If it was invalidated, then we need to query master.
|
|
es.emojiByIdMut.Lock()
|
|
if es.emojiByIdInvalidations[id] {
|
|
// And then remove the key from the map.
|
|
rctx = sqlstore.RequestContextWithMaster(rctx)
|
|
delete(es.emojiByIdInvalidations, id)
|
|
}
|
|
es.emojiByIdMut.Unlock()
|
|
|
|
emoji, err := es.EmojiStore.Get(rctx, id, allowFromCache)
|
|
|
|
if allowFromCache && err == nil {
|
|
es.addToCache(emoji)
|
|
}
|
|
|
|
return emoji, err
|
|
}
|
|
|
|
func (es *LocalCacheEmojiStore) GetByName(rctx request.CTX, name string, allowFromCache bool) (*model.Emoji, error) {
|
|
if id, ok := model.GetSystemEmojiId(name); ok {
|
|
return es.Get(rctx, id, allowFromCache)
|
|
}
|
|
|
|
if allowFromCache {
|
|
if emoji, ok := es.getFromCacheByName(name); ok {
|
|
return emoji, nil
|
|
}
|
|
}
|
|
|
|
// If it was invalidated, then we need to query master.
|
|
es.emojiByNameMut.Lock()
|
|
if es.emojiByNameInvalidations[name] {
|
|
rctx = sqlstore.RequestContextWithMaster(rctx)
|
|
// And then remove the key from the map.
|
|
delete(es.emojiByNameInvalidations, name)
|
|
}
|
|
es.emojiByNameMut.Unlock()
|
|
|
|
emoji, err := es.EmojiStore.GetByName(rctx, name, allowFromCache)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if allowFromCache {
|
|
es.addToCache(emoji)
|
|
}
|
|
|
|
return emoji, nil
|
|
}
|
|
|
|
func (es *LocalCacheEmojiStore) GetMultipleByName(rctx request.CTX, names []string) ([]*model.Emoji, error) {
|
|
emojis := []*model.Emoji{}
|
|
remainingEmojiNames := make([]string, 0)
|
|
|
|
for _, name := range names {
|
|
if emoji, ok := es.getFromCacheByName(name); ok {
|
|
emojis = append(emojis, emoji)
|
|
} else {
|
|
// If it was invalidated, then we need to query master.
|
|
es.emojiByNameMut.Lock()
|
|
if es.emojiByNameInvalidations[name] {
|
|
rctx = sqlstore.RequestContextWithMaster(rctx)
|
|
// And then remove the key from the map.
|
|
delete(es.emojiByNameInvalidations, name)
|
|
}
|
|
es.emojiByNameMut.Unlock()
|
|
|
|
remainingEmojiNames = append(remainingEmojiNames, name)
|
|
}
|
|
}
|
|
|
|
if len(remainingEmojiNames) > 0 {
|
|
remainingEmojis, err := es.EmojiStore.GetMultipleByName(rctx, remainingEmojiNames)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, emoji := range remainingEmojis {
|
|
es.addToCache(emoji)
|
|
emojis = append(emojis, emoji)
|
|
}
|
|
}
|
|
|
|
return emojis, nil
|
|
}
|
|
|
|
func (es *LocalCacheEmojiStore) Delete(emoji *model.Emoji, time int64) error {
|
|
err := es.EmojiStore.Delete(emoji, time)
|
|
|
|
if err == nil {
|
|
es.removeFromCache(emoji)
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
func (es *LocalCacheEmojiStore) addToCache(emoji *model.Emoji) {
|
|
es.rootStore.doStandardAddToCache(es.rootStore.emojiCacheById, emoji.Id, emoji)
|
|
es.rootStore.doStandardAddToCache(es.rootStore.emojiIdCacheByName, emoji.Name, emoji.Id)
|
|
}
|
|
|
|
func (es *LocalCacheEmojiStore) getFromCacheById(id string) (*model.Emoji, bool) {
|
|
var emoji *model.Emoji
|
|
if err := es.rootStore.doStandardReadCache(es.rootStore.emojiCacheById, id, &emoji); err == nil {
|
|
return emoji, true
|
|
}
|
|
return nil, false
|
|
}
|
|
|
|
func (es *LocalCacheEmojiStore) getFromCacheByName(name string) (*model.Emoji, bool) {
|
|
var emojiId string
|
|
if err := es.rootStore.doStandardReadCache(es.rootStore.emojiIdCacheByName, name, &emojiId); err == nil {
|
|
return es.getFromCacheById(emojiId)
|
|
}
|
|
return nil, false
|
|
}
|
|
|
|
func (es *LocalCacheEmojiStore) removeFromCache(emoji *model.Emoji) {
|
|
es.emojiByIdMut.Lock()
|
|
es.emojiByIdInvalidations[emoji.Id] = true
|
|
es.emojiByIdMut.Unlock()
|
|
es.rootStore.doInvalidateCacheCluster(es.rootStore.emojiCacheById, emoji.Id, nil)
|
|
|
|
es.emojiByNameMut.Lock()
|
|
es.emojiByNameInvalidations[emoji.Name] = true
|
|
es.emojiByNameMut.Unlock()
|
|
es.rootStore.doInvalidateCacheCluster(es.rootStore.emojiIdCacheByName, emoji.Name, nil)
|
|
}
|