chore: refresh gramsrv public release

internal/store/postgres/channel_store.go

@@ -0,0 +1,140 @@
+package postgres
+
+import (
+	"go.uber.org/zap"
+
+	"telesrv/internal/store"
+	"telesrv/internal/store/postgres/sqlcgen"
+)
+
+const channelDialogQueryLimit = 500
+
+const channelDialogCandidateLimit = 10000
+
+const channelMemberFilterBatch = 1000
+
+const retryableChannelTxAttempts = 3
+
+// ChannelStore 用 PostgreSQL 实现 store.ChannelStore。
+type ChannelStore struct {
+	db          sqlcgen.DBTX
+	ids         store.ChannelIDAllocator
+	msgIDs      store.ChannelMessageIDAllocator
+	log         *zap.Logger
+	rowCache    *ChannelRowCache
+	memberCache *ChannelMemberCache
+	dialogCache *ChannelDialogCache
+	boostCache  *ChannelBoostCache
+}
+
+// ChannelStoreOption 调整 PostgreSQL ChannelStore 依赖。
+type ChannelStoreOption func(*ChannelStore)
+
+// WithChannelAllocators 注入 Redis-backed channel id / message id allocator。
+func WithChannelAllocators(ids store.ChannelIDAllocator, msgIDs store.ChannelMessageIDAllocator) ChannelStoreOption {
+	return func(s *ChannelStore) {
+		s.ids = ids
+		s.msgIDs = msgIDs
+	}
+}
+
+// WithChannelLogger 注入频道 store 日志器，用于追踪 channel pts 分配。
+func WithChannelLogger(log *zap.Logger) ChannelStoreOption {
+	return func(s *ChannelStore) {
+		s.log = log
+	}
+}
+
+// WithChannelRowCache 注入「共享频道行」进程内缓存(由 ReadModelChangeListener 实时失效)。
+// 传 nil 等于禁用，不影响任何读路径行为。
+func WithChannelRowCache(cache *ChannelRowCache) ChannelStoreOption {
+	return func(s *ChannelStore) {
+		s.rowCache = cache
+	}
+}
+
+// WithChannelMemberCache 注入「频道成员/访问态」进程内缓存。
+// 传 nil 等于禁用；事务内仍绕过，提交后由 read model listener 失效。
+func WithChannelMemberCache(cache *ChannelMemberCache) ChannelStoreOption {
+	return func(s *ChannelStore) {
+		s.memberCache = cache
+	}
+}
+
+// WithChannelDialogCache 注入 viewer-scoped 频道 dialog 投影缓存。
+// 传 nil 等于禁用；事务内仍绕过，避免写后读旧 dialog/read boundary。
+func WithChannelDialogCache(cache *ChannelDialogCache) ChannelStoreOption {
+	return func(s *ChannelStore) {
+		s.dialogCache = cache
+	}
+}
+
+// WithChannelBoostCache 注入频道 boost 读投影缓存(SelfBoostsApplied + peer total)。
+// 传 nil 等于禁用；事务内仍绕过，避免 send 权限判断读到旧值。
+func WithChannelBoostCache(cache *ChannelBoostCache) ChannelStoreOption {
+	return func(s *ChannelStore) {
+		s.boostCache = cache
+	}
+}
+
+// cacheActive 报告当前句柄是否可用频道行缓存：仅启用缓存且走连接池(非事务)时。
+// 事务内(db != s.db)一律绕过缓存实时读，保证事务读己写。
+func (s *ChannelStore) cacheActive(db sqlcgen.DBTX) bool {
+	return s.rowCache != nil && db == s.db
+}
+
+func (s *ChannelStore) memberCacheActive(db sqlcgen.DBTX) bool {
+	return s.memberCache != nil && db == s.db
+}
+
+func (s *ChannelStore) dialogCacheActive(db sqlcgen.DBTX) bool {
+	return s.dialogCache != nil && db == s.db
+}
+
+func (s *ChannelStore) boostCacheActive(db sqlcgen.DBTX) bool {
+	return s.boostCache != nil && db == s.db
+}
+
+// NewChannelStore 基于 pgx 连接池（或事务）创建 ChannelStore。
+func NewChannelStore(db sqlcgen.DBTX, opts ...ChannelStoreOption) *ChannelStore {
+	s := &ChannelStore{db: db}
+	for _, opt := range opts {
+		opt(s)
+	}
+	if s.log == nil {
+		s.log = zap.NewNop()
+	}
+	if s.ids == nil {
+		s.ids = pgChannelIDAllocator{db: db}
+	}
+	if s.msgIDs == nil {
+		s.msgIDs = pgChannelMessageIDAllocator{db: db}
+	}
+	return s
+}
+
+const channelColumns = `c.id, c.access_hash, c.creator_user_id, c.title, c.about, COALESCE(c.username, ''), c.verified,
+c.broadcast, c.megagroup, c.forum, c.forum_tabs, c.autotranslation, c.restricted_sponsored, c.broadcast_messages_allowed, c.send_paid_messages_stars, c.noforwards, c.join_to_send, c.join_request, c.signatures, c.pre_history_hidden, c.participants_hidden, c.antispam,
+EXISTS (SELECT 1 FROM channel_invites ci WHERE ci.channel_id = c.id AND NOT ci.revoked) AS has_link,
+c.linked_chat_id, c.monoforum, c.linked_monoforum_id, c.slowmode_seconds, c.boosts_unrestrict, c.default_banned_rights::text,
+c.available_reactions::text, c.color_set, c.color, c.color_background_emoji_id, c.profile_color_set, c.profile_color, c.profile_color_background_emoji_id, c.emoji_status_document_id, c.emoji_status_until,
+c.wallpaper::text, c.participants_count, c.admins_count, c.kicked_count, c.banned_count, c.top_message_id, c.pinned_message_id, c.pts,
+c.ttl_period, c.date, c.deleted, c.photo_id, c.photo_dc_id, c.photo_stripped,
+c.active_call_id, c.active_call_access_hash, c.active_call_not_empty`
+
+const channelMessageColumns = `channel_id, id, random_id, sender_user_id, from_peer_type, from_peer_id,
+send_as_peer_type, send_as_peer_id, message_date, edit_date, post, silent, noforwards, body,
+entities::text, reply_to::text, reply_to_msg_id, reply_to_peer_type, reply_to_peer_id, reply_to_top_id,
+fwd_from::text, discussion_channel_id, discussion_message_id, action::text, pts, deleted, media::text,
+reply_markup::text, ttl_period, expires_at, views_count, post_author, pinned, via_bot_id, grouped_id, from_boosts_applied, saved_peer_type, saved_peer_id`
+
+const channelForumTopicColumns = `channel_id, topic_id, creator_user_id, title, icon_color, icon_emoji_id,
+title_missing, closed, hidden, pinned, pinned_order, date, top_message_id, read_inbox_max_id,
+read_outbox_max_id, unread_count, unread_mentions_count, unread_reactions_count,
+unread_poll_votes_count`
+
+const (
+	messageHistoryLoadBackward messageHistoryLoad = iota
+	messageHistoryLoadForward
+	messageHistoryLoadAround
+)