chore: refresh gramsrv public release
This commit is contained in:
A
parent
75cebe8dbf
commit
70b6820474
1274 changed files with 378751 additions and 59919 deletions
264
internal/rpc/messages_polls_updates.go
Normal file
264
internal/rpc/messages_polls_updates.go
Normal file
|
|
@ -0,0 +1,264 @@
|
|||
package rpc
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
|
||||
"github.com/gotd/td/tg"
|
||||
"github.com/gotd/td/tgerr"
|
||||
"go.uber.org/zap"
|
||||
|
||||
"telesrv/internal/domain"
|
||||
)
|
||||
|
||||
// 本文件是 poll 链路的响应/推送/事件辅助，与 reaction（messages_reactions_send.go）同构：
|
||||
// - 私聊：双方各记一条 message_poll durable event（无 pts 的 updateMessagePoll + aux pts 簿记），
|
||||
// 离线端经 getDifference 拿到消息快照与最新 poll 状态；
|
||||
// - 频道：实时 fan-out 给在线 viewer（作者进 explicit 收件人），无 durable event——
|
||||
// 与 channel reaction 现状一致（离线缺口见 compatibility-matrix.md，客户端靠 getPollResults 刷新）。
|
||||
|
||||
// pollMutationErr 把 poll/消息域错误映射为 RPC error。
|
||||
func pollMutationErr(err error) error {
|
||||
switch {
|
||||
case err == nil:
|
||||
return nil
|
||||
case errors.Is(err, domain.ErrPollClosed):
|
||||
return pollClosedErr()
|
||||
case errors.Is(err, domain.ErrPollRevoteNotAllowed):
|
||||
return revoteNotAllowedErr()
|
||||
case errors.Is(err, domain.ErrPollOptionInvalid):
|
||||
return optionInvalidErr()
|
||||
case errors.Is(err, domain.ErrPollNotCreator):
|
||||
return tgerr.New(403, "MESSAGE_AUTHOR_REQUIRED")
|
||||
case errors.Is(err, domain.ErrPollNotFound), errors.Is(err, domain.ErrPollInvalid), errors.Is(err, domain.ErrMessageIDInvalid):
|
||||
return messageIDInvalidErr()
|
||||
default:
|
||||
return channelInvalidErr(err)
|
||||
}
|
||||
}
|
||||
|
||||
// loadMessagePoll 按 viewer 视角加载消息上的 poll（已 enrich）。
|
||||
func (r *Router) loadMessagePoll(ctx context.Context, userID int64, peer domain.Peer, msgID int) (*domain.MessagePoll, domain.Peer, bool) {
|
||||
switch peer.Type {
|
||||
case domain.PeerTypeChannel:
|
||||
if r.deps.Channels == nil {
|
||||
return nil, peer, false
|
||||
}
|
||||
history, err := r.deps.Channels.GetMessages(ctx, userID, peer.ID, []int{msgID})
|
||||
if err != nil {
|
||||
return nil, peer, false
|
||||
}
|
||||
for _, msg := range history.Messages {
|
||||
if msg.ID == msgID && msg.Media != nil && msg.Media.Kind == domain.MessageMediaKindPoll && msg.Media.Poll != nil {
|
||||
return msg.Media.Poll, peer, true
|
||||
}
|
||||
}
|
||||
case domain.PeerTypeUser:
|
||||
if r.deps.Messages == nil {
|
||||
return nil, peer, false
|
||||
}
|
||||
list, err := r.deps.Messages.GetMessages(ctx, userID, []int{msgID})
|
||||
if err != nil {
|
||||
return nil, peer, false
|
||||
}
|
||||
for _, msg := range list.Messages {
|
||||
if msg.ID == msgID && msg.Peer == peer && msg.Media != nil && msg.Media.Kind == domain.MessageMediaKindPoll && msg.Media.Poll != nil {
|
||||
return msg.Media.Poll, peer, true
|
||||
}
|
||||
}
|
||||
}
|
||||
return nil, peer, false
|
||||
}
|
||||
|
||||
// pollUpdateRefs 收集 updateMessagePoll 响应需要的 users/chats（recent voters 头像 + channel 实体）。
|
||||
// channel 由调用方传入（频道 poll 用结果里的 res.Channel；私聊传零值）：与 reaction fan-out 同款用
|
||||
// 单个 channel 一次性投影 tgChannels，避免频道 poll fan-out 每 viewer 一次 GetChannel 的 DB N+1
|
||||
// （poll 聚合 N+1 已由 ChannelPollFanoutViews 消除，这是 poll fan-out 残留的 per-viewer DB 调用）。
|
||||
func (r *Router) pollUpdateRefs(ctx context.Context, viewerUserID int64, peer domain.Peer, poll *domain.MessagePoll, channel domain.Channel) ([]tg.UserClass, []tg.ChatClass) {
|
||||
userIDs := make([]int64, 0, domain.MaxPollRecentVoters)
|
||||
if poll != nil && poll.Results != nil {
|
||||
userIDs = append(userIDs, poll.Results.RecentVoters...)
|
||||
}
|
||||
users := r.tgUsersForIDs(ctx, viewerUserID, userIDs)
|
||||
chats := []tg.ChatClass{}
|
||||
if peer.Type == domain.PeerTypeChannel {
|
||||
if channel.ID != 0 {
|
||||
// fan-out 路径：调用方已带 channel，一次性投影（同 reaction），免 per-viewer GetChannel。
|
||||
chats = tgChannels(viewerUserID, []domain.Channel{channel})
|
||||
} else if r.deps.Channels != nil {
|
||||
// 非 fan-out 单 viewer 路径（getPollResults）：channel 未带，保持原 GetChannel 行为不变。
|
||||
if view, err := r.deps.Channels.GetChannel(ctx, viewerUserID, peer.ID); err == nil && view.Channel.ID != 0 {
|
||||
chats = []tg.ChatClass{tgChannelChatForView(viewerUserID, view)}
|
||||
}
|
||||
}
|
||||
}
|
||||
return users, chats
|
||||
}
|
||||
|
||||
// privatePollUpdates 记录双方 durable event、推送双方在线 session，并返回投票者视角响应。
|
||||
func (r *Router) privatePollUpdates(ctx context.Context, requestUserID int64, res domain.PrivateMessagePollResult) tg.UpdatesClass {
|
||||
recordedEvents := r.recordPrivateMessagePollEvents(ctx, requestUserID, res)
|
||||
var requesterUpdates *tg.Updates
|
||||
for _, msg := range res.Messages {
|
||||
if msg.ID <= 0 || msg.Media == nil || msg.Media.Poll == nil {
|
||||
continue
|
||||
}
|
||||
update := tgUpdateMessagePoll(msg.Peer, msg.ID, msg.Media.Poll)
|
||||
if update == nil {
|
||||
continue
|
||||
}
|
||||
users, chats := r.pollUpdateRefs(ctx, msg.OwnerUserID, msg.Peer, msg.Media.Poll, domain.Channel{})
|
||||
updates := &tg.Updates{
|
||||
Updates: []tg.UpdateClass{update},
|
||||
Users: users,
|
||||
Chats: chats,
|
||||
Date: int(r.clock.Now().Unix()),
|
||||
}
|
||||
// message_poll 事件占账号 pts 但 updateMessagePoll 无 pts 字段，附 aux 簿记推水位。
|
||||
updates.Updates = appendAuxPtsBookkeeping(updates.Updates, recordedEvents[msg.OwnerUserID])
|
||||
if msg.OwnerUserID == requestUserID {
|
||||
requesterUpdates = updates
|
||||
}
|
||||
r.pushUserUpdates(ctx, msg.OwnerUserID, updates)
|
||||
}
|
||||
if requesterUpdates == nil {
|
||||
return tgEmptyUpdates(int(r.clock.Now().Unix()))
|
||||
}
|
||||
return requesterUpdates
|
||||
}
|
||||
|
||||
func (r *Router) recordPrivateMessagePollEvents(ctx context.Context, requestUserID int64, res domain.PrivateMessagePollResult) map[int64]domain.UpdateEvent {
|
||||
if r.deps.Updates == nil {
|
||||
return nil
|
||||
}
|
||||
recorder, ok := r.deps.Updates.(messagePollUpdateRecorder)
|
||||
if !ok {
|
||||
return nil
|
||||
}
|
||||
authKeyID, _ := AuthKeyIDFrom(ctx)
|
||||
events := make(map[int64]domain.UpdateEvent, len(res.Messages))
|
||||
for _, msg := range res.Messages {
|
||||
if msg.OwnerUserID == 0 || msg.ID == 0 {
|
||||
continue
|
||||
}
|
||||
eventAuthKeyID := [8]byte{}
|
||||
if msg.OwnerUserID == requestUserID {
|
||||
eventAuthKeyID = authKeyID
|
||||
}
|
||||
event, _, err := recorder.RecordMessagePoll(ctx, eventAuthKeyID, msg.OwnerUserID, msg)
|
||||
if err != nil {
|
||||
r.log.Warn("record message poll event failed")
|
||||
continue
|
||||
}
|
||||
events[msg.OwnerUserID] = event
|
||||
}
|
||||
return events
|
||||
}
|
||||
|
||||
// onEditMessageClosePoll 处理 editMessage + InputMediaPoll：当前唯一支持的 poll 编辑是
|
||||
// 关闭（closed=true，仅 poll 创建者）；改题/改选项无官方客户端路径，显式拒绝。
|
||||
func (r *Router) onEditMessageClosePoll(ctx context.Context, req *tg.MessagesEditMessageRequest, media *tg.InputMediaPoll) (tg.UpdatesClass, error) {
|
||||
if req.ID <= 0 || req.ID > domain.MaxMessageBoxID {
|
||||
return nil, messageIDInvalidErr()
|
||||
}
|
||||
if !media.Poll.Closed {
|
||||
return nil, mediaInvalidErr()
|
||||
}
|
||||
userID, peer, err := r.reactionPeer(ctx, req.Peer, nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
now := int(r.clock.Now().Unix())
|
||||
switch peer.Type {
|
||||
case domain.PeerTypeChannel:
|
||||
if r.deps.Channels == nil {
|
||||
return nil, messageIDInvalidErr()
|
||||
}
|
||||
res, err := r.deps.Channels.CloseMessagePoll(ctx, userID, domain.CloseChannelMessagePollRequest{
|
||||
UserID: userID,
|
||||
ChannelID: peer.ID,
|
||||
MessageID: req.ID,
|
||||
Date: now,
|
||||
})
|
||||
if err != nil {
|
||||
return nil, pollMutationErr(err)
|
||||
}
|
||||
return r.channelPollUpdates(ctx, userID, peer, req.ID, res, true), nil
|
||||
case domain.PeerTypeUser:
|
||||
if r.deps.Messages == nil {
|
||||
return nil, messageIDInvalidErr()
|
||||
}
|
||||
res, err := r.deps.Messages.CloseMessagePoll(ctx, userID, domain.ClosePrivateMessagePollRequest{
|
||||
UserID: userID,
|
||||
Peer: peer,
|
||||
MessageID: req.ID,
|
||||
Date: now,
|
||||
})
|
||||
if err != nil {
|
||||
return nil, pollMutationErr(err)
|
||||
}
|
||||
return r.privatePollUpdates(ctx, userID, res), nil
|
||||
default:
|
||||
return nil, peerIDInvalidErr()
|
||||
}
|
||||
}
|
||||
|
||||
// channelPollUpdates 组装投票者视角响应；push 为 true 时按 viewer 重建并 fan-out 给在线成员。
|
||||
//
|
||||
// Phase 4 模板化：fan-out 前一次性批量加载所有收件人的 per-viewer poll 投影（ChannelPollFanoutViews
|
||||
// 把 viewer-invariant 聚合只算一次 + 批量 viewerOptions/可见性），消除原先每 viewer 一次 GetMessages
|
||||
// 的 N+1。dispatch 仍同步（poll 是 viewer-only 无 durable event，改异步丢队列即永久漏）。预取未覆盖
|
||||
// 的 viewer 回退逐 viewer loadMessagePoll 保正确（与旧路径字节同源）。actor echo 仍用 res poll（旧行为）。
|
||||
func (r *Router) channelPollUpdates(ctx context.Context, userID int64, peer domain.Peer, msgID int, res domain.ChannelMessagePollResult, push bool) tg.UpdatesClass {
|
||||
var batched map[int64]*domain.MessagePoll
|
||||
if push && peer.Type == domain.PeerTypeChannel && r.deps.Channels != nil {
|
||||
recipients := r.channelFanoutRecipients(ctx, channelFanoutViewers, res.Channel.ID, res.Recipients)
|
||||
if len(recipients) > 0 {
|
||||
views, err := r.deps.Channels.ChannelPollFanoutViews(ctx, res.Channel.ID, msgID, recipients, int(r.clock.Now().Unix()))
|
||||
if err != nil {
|
||||
r.log.Warn("channel poll fanout prefetch failed; falling back to per-viewer reload",
|
||||
zap.Int64("channel_id", res.Channel.ID), zap.Int("msg_id", msgID), zap.Error(err))
|
||||
} else {
|
||||
batched = views
|
||||
}
|
||||
}
|
||||
}
|
||||
build := func(viewerUserID int64) *tg.Updates {
|
||||
poll := res.Message.Media.Poll
|
||||
if viewerUserID != userID {
|
||||
// 其它 viewer 的 chosen/correct/solution 门控不同，按其视角取投影。
|
||||
if p, evaluated := batched[viewerUserID]; evaluated {
|
||||
if p == nil {
|
||||
return nil // 预取已判定该 viewer 不可见
|
||||
}
|
||||
poll = p
|
||||
} else {
|
||||
// 预取未覆盖（极少：并发/收件人集漂移）→ 回退逐 viewer 重载。
|
||||
reloaded, _, found := r.loadMessagePoll(ctx, viewerUserID, peer, msgID)
|
||||
if !found {
|
||||
return nil
|
||||
}
|
||||
poll = reloaded
|
||||
}
|
||||
}
|
||||
update := tgUpdateMessagePoll(peer, msgID, poll)
|
||||
if update == nil {
|
||||
return nil
|
||||
}
|
||||
users, chats := r.pollUpdateRefs(ctx, viewerUserID, peer, poll, res.Channel)
|
||||
return &tg.Updates{
|
||||
Updates: []tg.UpdateClass{update},
|
||||
Users: users,
|
||||
Chats: chats,
|
||||
Date: int(r.clock.Now().Unix()),
|
||||
}
|
||||
}
|
||||
updates := build(userID)
|
||||
if push {
|
||||
r.pushChannelViewerUpdates(ctx, userID, res.Channel.ID, res.Recipients, build)
|
||||
}
|
||||
if updates == nil {
|
||||
return tgEmptyUpdates(int(r.clock.Now().Unix()))
|
||||
}
|
||||
return updates
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue