chore: refresh gramsrv public release
This commit is contained in:
A
parent
75cebe8dbf
commit
70b6820474
1274 changed files with 378751 additions and 59919 deletions
241
internal/store/postgres/channel_topic_read.go
Normal file
241
internal/store/postgres/channel_topic_read.go
Normal file
|
|
@ -0,0 +1,241 @@
|
|||
package postgres
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
|
||||
"github.com/jackc/pgx/v5"
|
||||
"telesrv/internal/domain"
|
||||
"telesrv/internal/store/postgres/sqlcgen"
|
||||
)
|
||||
|
||||
// topicReadFallbackInbox 是 viewer 在某 topic 无已读行时的初始已读水位：话题根之前的消息
|
||||
// 视为已读（root-1），且不低于 available_min（裁剪线之前不可见）。General(topic 1) 的
|
||||
// root-1=0，即“从头未读”。这样迁移上线时不会把存量话题整体翻成未读暴涨。
|
||||
func topicReadFallbackInbox(topicID, availableMinID int) int {
|
||||
base := topicID - 1
|
||||
if availableMinID > base {
|
||||
base = availableMinID
|
||||
}
|
||||
if base < 0 {
|
||||
base = 0
|
||||
}
|
||||
return base
|
||||
}
|
||||
|
||||
// channelTopicMessageCond 返回某 topic 的消息归属条件，占位符承载 topicID。General(=1) 归并
|
||||
// reply_to_top_id IN (placeholder, 0)：General 内新消息(=1)与开 forum 前历史消息(=0)都算 General。
|
||||
func channelTopicMessageCond(topicID int, placeholder string) string {
|
||||
if topicID == domain.ForumGeneralTopicID {
|
||||
// General 归并 reply_to_top_id ∈ {placeholder,0}，但排除其它话题的根服务消息
|
||||
// （它们 reply_to_top_id=0，但 id 是某话题根，不属于 General）。
|
||||
return "(cm.reply_to_top_id = " + placeholder + " OR cm.reply_to_top_id = 0)" +
|
||||
" AND NOT EXISTS (SELECT 1 FROM channel_forum_topics gt WHERE gt.channel_id = cm.channel_id AND gt.topic_id = cm.id AND NOT gt.deleted)"
|
||||
}
|
||||
return "cm.reply_to_top_id = " + placeholder
|
||||
}
|
||||
|
||||
// channelTopicReadInbox 取 viewer 在某 topic 的已读水位，无行时回落 fallback。
|
||||
func (s *ChannelStore) channelTopicReadInbox(ctx context.Context, db sqlcgen.DBTX, channelID, userID int64, topicID, availableMinID int) (int, error) {
|
||||
var v int
|
||||
err := db.QueryRow(ctx, `SELECT read_inbox_max_id FROM channel_topic_read WHERE channel_id = $1 AND user_id = $2 AND topic_id = $3`, channelID, userID, topicID).Scan(&v)
|
||||
if errors.Is(err, pgx.ErrNoRows) {
|
||||
return topicReadFallbackInbox(topicID, availableMinID), nil
|
||||
}
|
||||
if err != nil {
|
||||
return 0, fmt.Errorf("select channel topic read: %w", err)
|
||||
}
|
||||
if fb := topicReadFallbackInbox(topicID, availableMinID); v < fb {
|
||||
v = fb
|
||||
}
|
||||
return v, nil
|
||||
}
|
||||
|
||||
// topicReadWater 是 viewer 在某 topic 的 inbox/outbox 已读水位。
|
||||
type topicReadWater struct {
|
||||
Inbox int
|
||||
Outbox int
|
||||
}
|
||||
|
||||
// channelTopicReadBatch 批量取多个 topic 的 per-viewer 已读水位（getForumTopics 现算与下发用），
|
||||
// 无行的 topic 回落 fallback（inbox=root-1，outbox=0）。返回 topicID -> 水位。
|
||||
func (s *ChannelStore) channelTopicReadBatch(ctx context.Context, channelID, userID int64, topicIDs []int32, availableMinID int) (map[int]topicReadWater, error) {
|
||||
out := make(map[int]topicReadWater, len(topicIDs))
|
||||
for _, t := range topicIDs {
|
||||
out[int(t)] = topicReadWater{Inbox: topicReadFallbackInbox(int(t), availableMinID)}
|
||||
}
|
||||
if len(topicIDs) == 0 {
|
||||
return out, nil
|
||||
}
|
||||
rows, err := s.db.Query(ctx, `SELECT topic_id, read_inbox_max_id, read_outbox_max_id FROM channel_topic_read WHERE channel_id = $1 AND user_id = $2 AND topic_id = ANY($3::int[])`, channelID, userID, topicIDs)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("batch channel topic read: %w", err)
|
||||
}
|
||||
defer rows.Close()
|
||||
for rows.Next() {
|
||||
var topicID, inbox, outbox int
|
||||
if err := rows.Scan(&topicID, &inbox, &outbox); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
w := out[topicID]
|
||||
if inbox > w.Inbox {
|
||||
w.Inbox = inbox
|
||||
}
|
||||
w.Outbox = outbox
|
||||
out[topicID] = w
|
||||
}
|
||||
return out, rows.Err()
|
||||
}
|
||||
|
||||
// channelTopicTopMessageID 取某 topic 当前最新可见消息 id，作为 readDiscussion 推进上界。
|
||||
func (s *ChannelStore) channelTopicTopMessageID(ctx context.Context, channelID int64, topicID, availableMinID int) (int, error) {
|
||||
var v int
|
||||
err := s.db.QueryRow(ctx, `
|
||||
SELECT COALESCE(MAX(cm.id), 0)::int
|
||||
FROM channel_messages cm
|
||||
WHERE cm.channel_id = $1 AND `+channelTopicMessageCond(topicID, "$2")+` AND cm.id > $3 AND NOT cm.deleted`,
|
||||
channelID, topicID, availableMinID).Scan(&v)
|
||||
if err != nil {
|
||||
return 0, fmt.Errorf("channel topic top message: %w", err)
|
||||
}
|
||||
return v, nil
|
||||
}
|
||||
|
||||
// ReadChannelTopicHistory 推进 viewer 在 forum 单话题的 per-topic 已读水位（messages.readDiscussion），
|
||||
// 不碰频道级 channel_members.read_inbox_max_id（消除话题间已读串扰），并返回需推进 outbox 的发送者。
|
||||
func (s *ChannelStore) ReadChannelTopicHistory(ctx context.Context, req domain.ReadChannelTopicHistoryRequest) (domain.ReadChannelTopicHistoryResult, error) {
|
||||
if req.UserID == 0 || req.ChannelID == 0 || req.TopicID <= 0 {
|
||||
return domain.ReadChannelTopicHistoryResult{}, domain.ErrChannelInvalid
|
||||
}
|
||||
channel, member, err := s.getChannelForMember(ctx, s.db, req.UserID, req.ChannelID)
|
||||
if err != nil {
|
||||
return domain.ReadChannelTopicHistoryResult{}, err
|
||||
}
|
||||
if !channel.Forum {
|
||||
return domain.ReadChannelTopicHistoryResult{}, domain.ErrChannelForumMissing
|
||||
}
|
||||
topMax, err := s.channelTopicTopMessageID(ctx, req.ChannelID, req.TopicID, member.AvailableMinID)
|
||||
if err != nil {
|
||||
return domain.ReadChannelTopicHistoryResult{}, err
|
||||
}
|
||||
maxID := req.MaxID
|
||||
if maxID <= 0 || maxID > topMax {
|
||||
maxID = topMax
|
||||
}
|
||||
prev, err := s.channelTopicReadInbox(ctx, s.db, req.ChannelID, req.UserID, req.TopicID, member.AvailableMinID)
|
||||
if err != nil {
|
||||
return domain.ReadChannelTopicHistoryResult{}, err
|
||||
}
|
||||
if maxID <= prev {
|
||||
return domain.ReadChannelTopicHistoryResult{Channel: channel, TopicID: req.TopicID, MaxID: prev, Changed: false, Pts: channel.Pts}, nil
|
||||
}
|
||||
if _, err := s.db.Exec(ctx, `
|
||||
INSERT INTO channel_topic_read (channel_id, user_id, topic_id, read_inbox_max_id, read_inbox_date, updated_at)
|
||||
VALUES ($1, $2, $3, $4, $5, now())
|
||||
ON CONFLICT (channel_id, user_id, topic_id) DO UPDATE SET
|
||||
read_inbox_date = CASE WHEN channel_topic_read.read_inbox_max_id < EXCLUDED.read_inbox_max_id THEN EXCLUDED.read_inbox_date ELSE channel_topic_read.read_inbox_date END,
|
||||
read_inbox_max_id = GREATEST(channel_topic_read.read_inbox_max_id, EXCLUDED.read_inbox_max_id),
|
||||
updated_at = now()`,
|
||||
req.ChannelID, req.UserID, req.TopicID, maxID, req.Date); err != nil {
|
||||
return domain.ReadChannelTopicHistoryResult{}, fmt.Errorf("upsert channel topic read inbox: %w", err)
|
||||
}
|
||||
outbox, err := s.advanceChannelTopicReadOutbox(ctx, req.ChannelID, req.UserID, req.TopicID, prev, maxID)
|
||||
if err != nil {
|
||||
return domain.ReadChannelTopicHistoryResult{}, err
|
||||
}
|
||||
return domain.ReadChannelTopicHistoryResult{Channel: channel, TopicID: req.TopicID, MaxID: maxID, Changed: true, Pts: channel.Pts, OutboxUpdates: outbox}, nil
|
||||
}
|
||||
|
||||
// advanceChannelTopicReadOutbox 推进话题内被本次已读覆盖到的发送者的 per-topic read_outbox 水位，
|
||||
// 返回这些发送者用于在线下发 updateReadChannelDiscussionOutbox（已读回执 ✓✓）。回执 UPSERT 幂等，
|
||||
// 无需与 inbox 强原子。
|
||||
func (s *ChannelStore) advanceChannelTopicReadOutbox(ctx context.Context, channelID, readerUserID int64, topicID, prev, maxID int) ([]domain.ChannelReadOutboxUpdate, error) {
|
||||
rows, err := s.db.Query(ctx, `
|
||||
SELECT DISTINCT cm.sender_user_id
|
||||
FROM channel_messages cm
|
||||
WHERE cm.channel_id = $1 AND `+channelTopicMessageCond(topicID, "$2")+`
|
||||
AND cm.id > $3 AND cm.id <= $4 AND cm.sender_user_id <> $5 AND cm.sender_user_id <> 0 AND NOT cm.deleted`,
|
||||
channelID, topicID, prev, maxID, readerUserID)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("scan channel topic outbox senders: %w", err)
|
||||
}
|
||||
senders := make([]int64, 0, 8)
|
||||
for rows.Next() {
|
||||
var sender int64
|
||||
if err := rows.Scan(&sender); err != nil {
|
||||
rows.Close()
|
||||
return nil, err
|
||||
}
|
||||
senders = append(senders, sender)
|
||||
}
|
||||
rows.Close()
|
||||
if err := rows.Err(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
updates := make([]domain.ChannelReadOutboxUpdate, 0, len(senders))
|
||||
for _, sender := range senders {
|
||||
if _, err := s.db.Exec(ctx, `
|
||||
INSERT INTO channel_topic_read (channel_id, user_id, topic_id, read_outbox_max_id, updated_at)
|
||||
VALUES ($1, $2, $3, $4, now())
|
||||
ON CONFLICT (channel_id, user_id, topic_id) DO UPDATE SET
|
||||
read_outbox_max_id = GREATEST(channel_topic_read.read_outbox_max_id, EXCLUDED.read_outbox_max_id),
|
||||
updated_at = now()`,
|
||||
channelID, sender, topicID, maxID); err != nil {
|
||||
return nil, fmt.Errorf("upsert channel topic read outbox: %w", err)
|
||||
}
|
||||
updates = append(updates, domain.ChannelReadOutboxUpdate{UserID: sender, MaxID: maxID})
|
||||
}
|
||||
return updates, nil
|
||||
}
|
||||
|
||||
// GeneralForumTopic 现算 General 话题（id=1）对 viewer 的状态：归属 reply_to_top_id ∈ {0,1} 且
|
||||
// 排除其它话题根服务消息，未读用 General 自己的 per-topic 水位（不被普通话题已读串扰）。
|
||||
func (s *ChannelStore) GeneralForumTopic(ctx context.Context, viewerUserID, channelID int64) (domain.ChannelForumTopic, error) {
|
||||
if viewerUserID == 0 || channelID == 0 {
|
||||
return domain.ChannelForumTopic{}, domain.ErrChannelInvalid
|
||||
}
|
||||
channel, member, err := s.getChannelForMember(ctx, s.db, viewerUserID, channelID)
|
||||
if err != nil {
|
||||
return domain.ChannelForumTopic{}, err
|
||||
}
|
||||
if !channel.Forum {
|
||||
return domain.ChannelForumTopic{}, domain.ErrChannelForumMissing
|
||||
}
|
||||
gid := domain.ForumGeneralTopicID
|
||||
waters, err := s.channelTopicReadBatch(ctx, channelID, viewerUserID, []int32{int32(gid)}, member.AvailableMinID)
|
||||
if err != nil {
|
||||
return domain.ChannelForumTopic{}, err
|
||||
}
|
||||
water := waters[gid]
|
||||
cond := channelTopicMessageCond(gid, "$2")
|
||||
var topMsg int
|
||||
if err := s.db.QueryRow(ctx, `
|
||||
SELECT COALESCE(MAX(cm.id), 0)::int FROM channel_messages cm
|
||||
WHERE cm.channel_id = $1 AND `+cond+` AND cm.id > $3 AND NOT cm.deleted`,
|
||||
channelID, gid, member.AvailableMinID).Scan(&topMsg); err != nil {
|
||||
return domain.ChannelForumTopic{}, fmt.Errorf("general top message: %w", err)
|
||||
}
|
||||
var unread int
|
||||
if err := s.db.QueryRow(ctx, `
|
||||
SELECT COUNT(*)::int FROM (
|
||||
SELECT 1 FROM channel_messages cm
|
||||
WHERE cm.channel_id = $1 AND `+cond+` AND cm.id > $3 AND cm.sender_user_id <> $4 AND NOT cm.deleted
|
||||
LIMIT $5
|
||||
) g`, channelID, gid, water.Inbox, viewerUserID, domain.MaxDialogUnreadCount).Scan(&unread); err != nil {
|
||||
return domain.ChannelForumTopic{}, fmt.Errorf("general unread: %w", err)
|
||||
}
|
||||
return domain.ChannelForumTopic{
|
||||
ChannelID: channelID,
|
||||
TopicID: gid,
|
||||
Title: "General",
|
||||
CreatorUserID: channel.CreatorUserID,
|
||||
Date: channel.Date,
|
||||
TopMessageID: topMsg,
|
||||
ReadInboxMaxID: water.Inbox,
|
||||
ReadOutboxMaxID: water.Outbox,
|
||||
UnreadCount: unread,
|
||||
UnreadMentionsCount: s.countChannelUnreadMentionsForTop(ctx, viewerUserID, channelID, gid),
|
||||
UnreadReactionsCount: s.countChannelUnreadReactionsForTop(ctx, viewerUserID, channelID, gid, member.AvailableMinID),
|
||||
}, nil
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue