chore: refresh gramsrv public release
This commit is contained in:
parent
75cebe8dbf
commit
70b6820474
1274 changed files with 378751 additions and 59919 deletions
107
internal/app/channels/read_model_cache.go
Normal file
107
internal/app/channels/read_model_cache.go
Normal file
|
|
@ -0,0 +1,107 @@
|
|||
package channels
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
|
||||
"telesrv/internal/app/readmodel"
|
||||
"telesrv/internal/domain"
|
||||
"telesrv/internal/readmodelcache"
|
||||
"telesrv/internal/store"
|
||||
)
|
||||
|
||||
const (
|
||||
// Full channel views are version-token guarded by channel_base,
|
||||
// channel_member, and dialog_light. Keep the snapshot long-lived; write-side
|
||||
// read-model bumps, not time, drive correctness.
|
||||
defaultChannelViewReadModelTTL = 24 * time.Hour
|
||||
channelViewReadModelMaxEntries = 8192
|
||||
)
|
||||
|
||||
type channelViewCacheKey struct {
|
||||
userID int64
|
||||
channelID int64
|
||||
}
|
||||
|
||||
// channelViewReadModelCache 与 channelResolveReadModelCache 都缓存 domain.ChannelView,
|
||||
// 由统一缓存原语承载(版本闸门 / epoch 守卫 / LRU / clone)。
|
||||
type channelViewReadModelCache struct {
|
||||
cache *readmodelcache.Cache[channelViewCacheKey, domain.ChannelView]
|
||||
}
|
||||
|
||||
func newChannelViewReadModelCache(ttl time.Duration) *channelViewReadModelCache {
|
||||
if ttl <= 0 {
|
||||
ttl = defaultChannelViewReadModelTTL
|
||||
}
|
||||
return &channelViewReadModelCache{
|
||||
cache: readmodelcache.New[channelViewCacheKey, domain.ChannelView](readmodelcache.Config[channelViewCacheKey, domain.ChannelView]{
|
||||
MaxEntries: channelViewReadModelMaxEntries,
|
||||
TTL: ttl,
|
||||
Clone: cloneChannelView,
|
||||
}),
|
||||
}
|
||||
}
|
||||
|
||||
func (c *channelViewReadModelCache) getOrLoad(ctx context.Context, key channelViewCacheKey, hash int64, load func() (domain.ChannelView, error)) (domain.ChannelView, error) {
|
||||
if c == nil {
|
||||
return load()
|
||||
}
|
||||
return c.cache.GetOrLoadVersioned(ctx, key, hash, load)
|
||||
}
|
||||
|
||||
func (s *Service) cachedChannelView(ctx context.Context, userID, channelID int64) (domain.ChannelView, error) {
|
||||
if s.viewCache == nil || s.versions == nil {
|
||||
return s.channels.GetChannel(ctx, userID, channelID)
|
||||
}
|
||||
hash, err := s.channelViewHash(ctx, userID, channelID)
|
||||
if err != nil {
|
||||
return domain.ChannelView{}, err
|
||||
}
|
||||
if hash == 0 {
|
||||
return s.channels.GetChannel(ctx, userID, channelID)
|
||||
}
|
||||
key := channelViewCacheKey{userID: userID, channelID: channelID}
|
||||
return s.viewCache.getOrLoad(ctx, key, hash, func() (domain.ChannelView, error) {
|
||||
return s.channels.GetChannel(ctx, userID, channelID)
|
||||
})
|
||||
}
|
||||
|
||||
func (s *Service) channelViewHash(ctx context.Context, userID, channelID int64) (int64, error) {
|
||||
keys := []store.ReadModelKey{
|
||||
{Model: readmodel.ModelChannelBase, OwnerUserID: 0, PeerType: domain.PeerTypeChannel, PeerID: channelID},
|
||||
{Model: readmodel.ModelChannelMember, OwnerUserID: userID, PeerType: domain.PeerTypeChannel, PeerID: channelID},
|
||||
{Model: readmodel.ModelDialogLight, OwnerUserID: userID, PeerType: domain.PeerTypeChannel, PeerID: channelID},
|
||||
// 快照含 SelfBoostsApplied:必须把 channel_self_boosts 纳入校验 token,否则 apply/revoke
|
||||
// 加成不会失效这份长 TTL 快照。注:boost 自然到期是 time-based、不触发写,故其残余仍受 TTL 约束。
|
||||
{Model: readmodel.ModelChannelSelfBoosts, OwnerUserID: userID, PeerType: domain.PeerTypeChannel, PeerID: channelID},
|
||||
}
|
||||
rows, err := s.versions.ReadModelHashes(ctx, keys)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
base := rows[keys[0]]
|
||||
if base == 0 {
|
||||
return 0, nil
|
||||
}
|
||||
return readmodel.MixHashes(base, rows[keys[1]], rows[keys[2]], rows[keys[3]]), nil
|
||||
}
|
||||
|
||||
func cloneChannelView(in domain.ChannelView) domain.ChannelView {
|
||||
in.Channel = cloneChannel(in.Channel)
|
||||
if in.Dialog.DefaultSendAs != nil {
|
||||
peer := *in.Dialog.DefaultSendAs
|
||||
in.Dialog.DefaultSendAs = &peer
|
||||
}
|
||||
if in.ExportedInvite != nil {
|
||||
invite := *in.ExportedInvite
|
||||
in.ExportedInvite = &invite
|
||||
}
|
||||
return in
|
||||
}
|
||||
|
||||
func cloneChannel(in domain.Channel) domain.Channel {
|
||||
in.PhotoStripped = append([]byte(nil), in.PhotoStripped...)
|
||||
in.ReactionPolicy.Emoticons = append([]string(nil), in.ReactionPolicy.Emoticons...)
|
||||
in.ReactionPolicy.CustomEmojiIDs = append([]int64(nil), in.ReactionPolicy.CustomEmojiIDs...)
|
||||
return in
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue