1333 lines
44 KiB
Go
1333 lines
44 KiB
Go
// Code generated by sqlc. DO NOT EDIT.
|
||
// versions:
|
||
// sqlc v1.31.1
|
||
// source: user_update_event.sql
|
||
|
||
package sqlcgen
|
||
|
||
import (
|
||
"context"
|
||
)
|
||
|
||
const appendUserUpdateEvent = `-- name: AppendUserUpdateEvent :exec
|
||
INSERT INTO user_update_events (
|
||
user_id,
|
||
pts,
|
||
pts_count,
|
||
date,
|
||
event_type,
|
||
event_bool,
|
||
event_phone,
|
||
event_peers,
|
||
peer_settings,
|
||
message_ids,
|
||
dialog_filter,
|
||
filter_order,
|
||
folder_peers,
|
||
story_payload,
|
||
reaction_payload,
|
||
emoji_status_payload,
|
||
message_box_id,
|
||
peer_type,
|
||
peer_id,
|
||
filter_id,
|
||
max_id,
|
||
still_unread_count,
|
||
channel_pts,
|
||
tags_enabled,
|
||
folder_id
|
||
) VALUES (
|
||
$1,
|
||
$2,
|
||
$3,
|
||
$4,
|
||
$5,
|
||
$6::boolean,
|
||
$7::text,
|
||
$8::jsonb,
|
||
$9::jsonb,
|
||
$10::jsonb,
|
||
$11::jsonb,
|
||
$12::jsonb,
|
||
$13::jsonb,
|
||
$14::jsonb,
|
||
$15::jsonb,
|
||
$16::jsonb,
|
||
$17,
|
||
$18::text,
|
||
$19::bigint,
|
||
$20::int,
|
||
$21::int,
|
||
$22::int,
|
||
$23::int,
|
||
$24::boolean,
|
||
$25::int
|
||
)
|
||
`
|
||
|
||
type AppendUserUpdateEventParams struct {
|
||
UserID int64
|
||
Pts int32
|
||
PtsCount int32
|
||
Date int32
|
||
EventType string
|
||
EventBool bool
|
||
EventPhone string
|
||
EventPeers []byte
|
||
PeerSettings []byte
|
||
MessageIds []byte
|
||
DialogFilter []byte
|
||
FilterOrder []byte
|
||
FolderPeers []byte
|
||
StoryPayload []byte
|
||
ReactionPayload []byte
|
||
EmojiStatusPayload []byte
|
||
MessageBoxID *int32
|
||
PeerType *string
|
||
PeerID *int64
|
||
FilterID int32
|
||
MaxID int32
|
||
StillUnreadCount int32
|
||
ChannelPts int32
|
||
TagsEnabled bool
|
||
FolderID int32
|
||
}
|
||
|
||
func (q *Queries) AppendUserUpdateEvent(ctx context.Context, arg AppendUserUpdateEventParams) error {
|
||
_, err := q.db.Exec(ctx, appendUserUpdateEvent,
|
||
arg.UserID,
|
||
arg.Pts,
|
||
arg.PtsCount,
|
||
arg.Date,
|
||
arg.EventType,
|
||
arg.EventBool,
|
||
arg.EventPhone,
|
||
arg.EventPeers,
|
||
arg.PeerSettings,
|
||
arg.MessageIds,
|
||
arg.DialogFilter,
|
||
arg.FilterOrder,
|
||
arg.FolderPeers,
|
||
arg.StoryPayload,
|
||
arg.ReactionPayload,
|
||
arg.EmojiStatusPayload,
|
||
arg.MessageBoxID,
|
||
arg.PeerType,
|
||
arg.PeerID,
|
||
arg.FilterID,
|
||
arg.MaxID,
|
||
arg.StillUnreadCount,
|
||
arg.ChannelPts,
|
||
arg.TagsEnabled,
|
||
arg.FolderID,
|
||
)
|
||
return err
|
||
}
|
||
|
||
const batchListDispatchEvents = `-- name: BatchListDispatchEvents :many
|
||
SELECT
|
||
e.user_id,
|
||
e.pts,
|
||
e.pts_count,
|
||
e.date,
|
||
e.event_type,
|
||
e.event_bool,
|
||
e.event_phone,
|
||
COALESCE(e.event_peers::text, '[]')::text AS event_peers_json,
|
||
COALESCE(e.peer_settings::text, '{}')::text AS peer_settings_json,
|
||
COALESCE(e.message_ids::text, '[]')::text AS message_ids_json,
|
||
COALESCE(e.dialog_filter::text, '{}')::text AS dialog_filter_json,
|
||
COALESCE(e.filter_order::text, '[]')::text AS filter_order_json,
|
||
COALESCE(e.folder_peers::text, '[]')::text AS folder_peers_json,
|
||
COALESCE(e.story_payload::text, '{}')::text AS story_payload_json,
|
||
COALESCE(e.reaction_payload::text, '{}')::text AS reaction_payload_json,
|
||
COALESCE(e.emoji_status_payload::text, '{}')::text AS emoji_status_payload_json,
|
||
COALESCE(e.peer_type, '')::text AS event_peer_type,
|
||
COALESCE(e.peer_id, 0)::bigint AS event_peer_id,
|
||
e.filter_id,
|
||
e.max_id,
|
||
e.still_unread_count,
|
||
e.channel_pts,
|
||
e.tags_enabled,
|
||
e.folder_id,
|
||
COALESCE(m.box_id, 0)::int AS message_id,
|
||
COALESCE(m.private_message_id, 0)::bigint AS private_message_id,
|
||
COALESCE(m.owner_user_id, 0)::bigint AS owner_user_id,
|
||
COALESCE(m.peer_type, '')::text AS peer_type,
|
||
COALESCE(m.peer_id, 0)::bigint AS peer_id,
|
||
COALESCE(m.from_user_id, 0)::bigint AS from_user_id,
|
||
COALESCE(m.message_date, 0)::int AS message_date,
|
||
COALESCE(m.ttl_period, 0)::int AS ttl_period,
|
||
COALESCE(m.expires_at, 0)::int AS expires_at,
|
||
COALESCE(m.edit_date, 0)::int AS edit_date,
|
||
COALESCE(m.hide_edited, false)::boolean AS hide_edited,
|
||
COALESCE(m.outgoing, false)::boolean AS outgoing,
|
||
COALESCE(m.body, '')::text AS body,
|
||
COALESCE(m.entities::text, '[]')::text AS message_entities_json,
|
||
COALESCE(m.silent, false)::boolean AS silent,
|
||
COALESCE(m.noforwards, false)::boolean AS noforwards,
|
||
COALESCE(m.reply_to_msg_id, 0)::int AS reply_to_msg_id,
|
||
COALESCE(m.reply_to_peer_type, '')::text AS reply_to_peer_type,
|
||
COALESCE(m.reply_to_peer_id, 0)::bigint AS reply_to_peer_id,
|
||
COALESCE(m.reply_to_top_id, 0)::int AS reply_to_top_id,
|
||
COALESCE(m.reply_to_story_id, 0)::int AS reply_to_story_id,
|
||
COALESCE(m.quote_text, '')::text AS quote_text,
|
||
COALESCE(m.quote_entities::text, '[]')::text AS quote_entities_json,
|
||
COALESCE(m.quote_offset, 0)::int AS quote_offset,
|
||
COALESCE(m.reply_external, '{}'::jsonb)::text AS reply_external_json,
|
||
COALESCE(m.fwd_from_peer_type, '')::text AS fwd_from_peer_type,
|
||
COALESCE(m.fwd_from_peer_id, 0)::bigint AS fwd_from_peer_id,
|
||
COALESCE(m.fwd_from_name, '')::text AS fwd_from_name,
|
||
COALESCE(m.fwd_date, 0)::int AS fwd_date,
|
||
COALESCE(m.fwd_saved_from_peer_type, '')::text AS fwd_saved_from_peer_type,
|
||
COALESCE(m.fwd_saved_from_peer_id, 0)::bigint AS fwd_saved_from_peer_id,
|
||
COALESCE(m.fwd_saved_from_msg_id, 0)::int AS fwd_saved_from_msg_id,
|
||
COALESCE(m.saved_peer_type, '')::text AS saved_peer_type,
|
||
COALESCE(m.saved_peer_id, 0)::bigint AS saved_peer_id,
|
||
COALESCE(m.media::text, '{}')::text AS media_json,
|
||
COALESCE(m.media_unread, false)::boolean AS media_unread,
|
||
COALESCE(m.reaction_unread, false)::boolean AS reaction_unread,
|
||
COALESCE(m.pinned, false)::boolean AS pinned,
|
||
COALESCE(m.via_bot_id, 0)::bigint AS via_bot_id,
|
||
COALESCE(m.grouped_id, 0)::bigint AS grouped_id,
|
||
COALESCE(m.effect, 0)::bigint AS effect,
|
||
COALESCE(m.reply_markup::text, '{}')::text AS reply_markup_json,
|
||
COALESCE(m.rich_message::text, '{}')::text AS rich_message_json,
|
||
COALESCE(m.deleted, false)::boolean AS message_deleted,
|
||
COALESCE(peer_u.id, 0)::bigint AS peer_user_id,
|
||
COALESCE(peer_u.access_hash, 0)::bigint AS peer_access_hash,
|
||
COALESCE(peer_u.phone, '')::text AS peer_phone,
|
||
COALESCE(peer_u.first_name, '')::text AS peer_first_name,
|
||
COALESCE(peer_u.last_name, '')::text AS peer_last_name,
|
||
COALESCE(peer_u.username, '')::text AS peer_username,
|
||
COALESCE(peer_u.country_code, '')::text AS peer_country_code,
|
||
COALESCE(peer_u.verified, false)::boolean AS peer_verified,
|
||
COALESCE(peer_u.support, false)::boolean AS peer_support,
|
||
COALESCE(peer_u.is_bot, false)::boolean AS peer_is_bot,
|
||
COALESCE(peer_u.bot_info_version, 0)::int AS peer_bot_info_version,
|
||
COALESCE(EXTRACT(EPOCH FROM peer_u.premium_expires_at), 0)::bigint AS peer_premium_until,
|
||
COALESCE(peer_u.emoji_status_document_id, 0)::bigint AS peer_emoji_status_document_id,
|
||
COALESCE(peer_u.emoji_status_until, 0)::bigint AS peer_emoji_status_until,
|
||
COALESCE(from_u.id, 0)::bigint AS from_user_user_id,
|
||
COALESCE(from_u.access_hash, 0)::bigint AS from_user_access_hash,
|
||
COALESCE(from_u.phone, '')::text AS from_user_phone,
|
||
COALESCE(from_u.first_name, '')::text AS from_user_first_name,
|
||
COALESCE(from_u.last_name, '')::text AS from_user_last_name,
|
||
COALESCE(from_u.username, '')::text AS from_user_username,
|
||
COALESCE(from_u.country_code, '')::text AS from_user_country_code,
|
||
COALESCE(from_u.verified, false)::boolean AS from_user_verified,
|
||
COALESCE(from_u.support, false)::boolean AS from_user_support,
|
||
COALESCE(from_u.is_bot, false)::boolean AS from_user_is_bot,
|
||
COALESCE(from_u.bot_info_version, 0)::int AS from_user_bot_info_version,
|
||
COALESCE(EXTRACT(EPOCH FROM from_u.premium_expires_at), 0)::bigint AS from_user_premium_until,
|
||
COALESCE(from_u.emoji_status_document_id, 0)::bigint AS from_user_emoji_status_document_id,
|
||
COALESCE(from_u.emoji_status_until, 0)::bigint AS from_user_emoji_status_until,
|
||
COALESCE(fwd_u.id, 0)::bigint AS fwd_user_id,
|
||
COALESCE(fwd_u.access_hash, 0)::bigint AS fwd_user_access_hash,
|
||
COALESCE(fwd_u.phone, '')::text AS fwd_user_phone,
|
||
COALESCE(fwd_u.first_name, '')::text AS fwd_user_first_name,
|
||
COALESCE(fwd_u.last_name, '')::text AS fwd_user_last_name,
|
||
COALESCE(fwd_u.username, '')::text AS fwd_user_username,
|
||
COALESCE(fwd_u.country_code, '')::text AS fwd_user_country_code,
|
||
COALESCE(fwd_u.verified, false)::boolean AS fwd_user_verified,
|
||
COALESCE(fwd_u.support, false)::boolean AS fwd_user_support,
|
||
COALESCE(fwd_u.is_bot, false)::boolean AS fwd_user_is_bot,
|
||
COALESCE(fwd_u.bot_info_version, 0)::int AS fwd_user_bot_info_version,
|
||
COALESCE(EXTRACT(EPOCH FROM fwd_u.premium_expires_at), 0)::bigint AS fwd_user_premium_until,
|
||
COALESCE(fwd_u.emoji_status_document_id, 0)::bigint AS fwd_user_emoji_status_document_id,
|
||
COALESCE(fwd_u.emoji_status_until, 0)::bigint AS fwd_user_emoji_status_until,
|
||
COALESCE(reply_u.id, 0)::bigint AS reply_user_id,
|
||
COALESCE(reply_u.access_hash, 0)::bigint AS reply_user_access_hash,
|
||
COALESCE(reply_u.phone, '')::text AS reply_user_phone,
|
||
COALESCE(reply_u.first_name, '')::text AS reply_user_first_name,
|
||
COALESCE(reply_u.last_name, '')::text AS reply_user_last_name,
|
||
COALESCE(reply_u.username, '')::text AS reply_user_username,
|
||
COALESCE(reply_u.country_code, '')::text AS reply_user_country_code,
|
||
COALESCE(reply_u.verified, false)::boolean AS reply_user_verified,
|
||
COALESCE(reply_u.support, false)::boolean AS reply_user_support,
|
||
COALESCE(reply_u.is_bot, false)::boolean AS reply_user_is_bot,
|
||
COALESCE(reply_u.bot_info_version, 0)::int AS reply_user_bot_info_version,
|
||
COALESCE(EXTRACT(EPOCH FROM reply_u.premium_expires_at), 0)::bigint AS reply_user_premium_until,
|
||
COALESCE(reply_u.emoji_status_document_id, 0)::bigint AS reply_user_emoji_status_document_id,
|
||
COALESCE(reply_u.emoji_status_until, 0)::bigint AS reply_user_emoji_status_until
|
||
FROM unnest($1::bigint[]) WITH ORDINALITY AS u(user_id, ord)
|
||
JOIN unnest($2::int[]) WITH ORDINALITY AS p(pts, ord) USING (ord)
|
||
JOIN user_update_events e ON e.user_id = u.user_id AND e.pts = p.pts
|
||
LEFT JOIN message_boxes m ON m.owner_user_id = e.user_id AND m.box_id = e.message_box_id
|
||
LEFT JOIN users peer_u ON m.peer_type = 'user' AND peer_u.id = m.peer_id
|
||
LEFT JOIN users from_u ON from_u.id = m.from_user_id
|
||
LEFT JOIN users fwd_u ON m.fwd_from_peer_type = 'user' AND fwd_u.id = m.fwd_from_peer_id
|
||
LEFT JOIN users reply_u ON m.reply_to_peer_type = 'user' AND reply_u.id = m.reply_to_peer_id
|
||
`
|
||
|
||
type BatchListDispatchEventsParams struct {
|
||
UserIds []int64
|
||
PtsList []int32
|
||
}
|
||
|
||
type BatchListDispatchEventsRow struct {
|
||
UserID int64
|
||
Pts int32
|
||
PtsCount int32
|
||
Date int32
|
||
EventType string
|
||
EventBool bool
|
||
EventPhone string
|
||
EventPeersJson string
|
||
PeerSettingsJson string
|
||
MessageIdsJson string
|
||
DialogFilterJson string
|
||
FilterOrderJson string
|
||
FolderPeersJson string
|
||
StoryPayloadJson string
|
||
ReactionPayloadJson string
|
||
EmojiStatusPayloadJson string
|
||
EventPeerType string
|
||
EventPeerID int64
|
||
FilterID int32
|
||
MaxID int32
|
||
StillUnreadCount int32
|
||
ChannelPts int32
|
||
TagsEnabled bool
|
||
FolderID int32
|
||
MessageID int32
|
||
PrivateMessageID int64
|
||
OwnerUserID int64
|
||
PeerType string
|
||
PeerID int64
|
||
FromUserID int64
|
||
MessageDate int32
|
||
TtlPeriod int32
|
||
ExpiresAt int32
|
||
EditDate int32
|
||
HideEdited bool
|
||
Outgoing bool
|
||
Body string
|
||
MessageEntitiesJson string
|
||
Silent bool
|
||
Noforwards bool
|
||
ReplyToMsgID int32
|
||
ReplyToPeerType string
|
||
ReplyToPeerID int64
|
||
ReplyToTopID int32
|
||
ReplyToStoryID int32
|
||
QuoteText string
|
||
QuoteEntitiesJson string
|
||
QuoteOffset int32
|
||
ReplyExternalJson string
|
||
FwdFromPeerType string
|
||
FwdFromPeerID int64
|
||
FwdFromName string
|
||
FwdDate int32
|
||
FwdSavedFromPeerType string
|
||
FwdSavedFromPeerID int64
|
||
FwdSavedFromMsgID int32
|
||
SavedPeerType string
|
||
SavedPeerID int64
|
||
MediaJson string
|
||
MediaUnread bool
|
||
ReactionUnread bool
|
||
Pinned bool
|
||
ViaBotID int64
|
||
GroupedID int64
|
||
Effect int64
|
||
ReplyMarkupJson string
|
||
RichMessageJson string
|
||
MessageDeleted bool
|
||
PeerUserID int64
|
||
PeerAccessHash int64
|
||
PeerPhone string
|
||
PeerFirstName string
|
||
PeerLastName string
|
||
PeerUsername string
|
||
PeerCountryCode string
|
||
PeerVerified bool
|
||
PeerSupport bool
|
||
PeerIsBot bool
|
||
PeerBotInfoVersion int32
|
||
PeerPremiumUntil int64
|
||
PeerEmojiStatusDocumentID int64
|
||
PeerEmojiStatusUntil int64
|
||
FromUserUserID int64
|
||
FromUserAccessHash int64
|
||
FromUserPhone string
|
||
FromUserFirstName string
|
||
FromUserLastName string
|
||
FromUserUsername string
|
||
FromUserCountryCode string
|
||
FromUserVerified bool
|
||
FromUserSupport bool
|
||
FromUserIsBot bool
|
||
FromUserBotInfoVersion int32
|
||
FromUserPremiumUntil int64
|
||
FromUserEmojiStatusDocumentID int64
|
||
FromUserEmojiStatusUntil int64
|
||
FwdUserID int64
|
||
FwdUserAccessHash int64
|
||
FwdUserPhone string
|
||
FwdUserFirstName string
|
||
FwdUserLastName string
|
||
FwdUserUsername string
|
||
FwdUserCountryCode string
|
||
FwdUserVerified bool
|
||
FwdUserSupport bool
|
||
FwdUserIsBot bool
|
||
FwdUserBotInfoVersion int32
|
||
FwdUserPremiumUntil int64
|
||
FwdUserEmojiStatusDocumentID int64
|
||
FwdUserEmojiStatusUntil int64
|
||
ReplyUserID int64
|
||
ReplyUserAccessHash int64
|
||
ReplyUserPhone string
|
||
ReplyUserFirstName string
|
||
ReplyUserLastName string
|
||
ReplyUserUsername string
|
||
ReplyUserCountryCode string
|
||
ReplyUserVerified bool
|
||
ReplyUserSupport bool
|
||
ReplyUserIsBot bool
|
||
ReplyUserBotInfoVersion int32
|
||
ReplyUserPremiumUntil int64
|
||
ReplyUserEmojiStatusDocumentID int64
|
||
ReplyUserEmojiStatusUntil int64
|
||
}
|
||
|
||
// 按 (user_id, pts) 精确批量取账号事件，供 outbox worker 一次性加载一批 claim 的事件详情，
|
||
// 取代逐条 ListUserUpdateEventsAfter。列与 ListUserUpdateEventsAfter 完全一致以复用转换逻辑。
|
||
func (q *Queries) BatchListDispatchEvents(ctx context.Context, arg BatchListDispatchEventsParams) ([]BatchListDispatchEventsRow, error) {
|
||
rows, err := q.db.Query(ctx, batchListDispatchEvents, arg.UserIds, arg.PtsList)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
defer rows.Close()
|
||
var items []BatchListDispatchEventsRow
|
||
for rows.Next() {
|
||
var i BatchListDispatchEventsRow
|
||
if err := rows.Scan(
|
||
&i.UserID,
|
||
&i.Pts,
|
||
&i.PtsCount,
|
||
&i.Date,
|
||
&i.EventType,
|
||
&i.EventBool,
|
||
&i.EventPhone,
|
||
&i.EventPeersJson,
|
||
&i.PeerSettingsJson,
|
||
&i.MessageIdsJson,
|
||
&i.DialogFilterJson,
|
||
&i.FilterOrderJson,
|
||
&i.FolderPeersJson,
|
||
&i.StoryPayloadJson,
|
||
&i.ReactionPayloadJson,
|
||
&i.EmojiStatusPayloadJson,
|
||
&i.EventPeerType,
|
||
&i.EventPeerID,
|
||
&i.FilterID,
|
||
&i.MaxID,
|
||
&i.StillUnreadCount,
|
||
&i.ChannelPts,
|
||
&i.TagsEnabled,
|
||
&i.FolderID,
|
||
&i.MessageID,
|
||
&i.PrivateMessageID,
|
||
&i.OwnerUserID,
|
||
&i.PeerType,
|
||
&i.PeerID,
|
||
&i.FromUserID,
|
||
&i.MessageDate,
|
||
&i.TtlPeriod,
|
||
&i.ExpiresAt,
|
||
&i.EditDate,
|
||
&i.HideEdited,
|
||
&i.Outgoing,
|
||
&i.Body,
|
||
&i.MessageEntitiesJson,
|
||
&i.Silent,
|
||
&i.Noforwards,
|
||
&i.ReplyToMsgID,
|
||
&i.ReplyToPeerType,
|
||
&i.ReplyToPeerID,
|
||
&i.ReplyToTopID,
|
||
&i.ReplyToStoryID,
|
||
&i.QuoteText,
|
||
&i.QuoteEntitiesJson,
|
||
&i.QuoteOffset,
|
||
&i.ReplyExternalJson,
|
||
&i.FwdFromPeerType,
|
||
&i.FwdFromPeerID,
|
||
&i.FwdFromName,
|
||
&i.FwdDate,
|
||
&i.FwdSavedFromPeerType,
|
||
&i.FwdSavedFromPeerID,
|
||
&i.FwdSavedFromMsgID,
|
||
&i.SavedPeerType,
|
||
&i.SavedPeerID,
|
||
&i.MediaJson,
|
||
&i.MediaUnread,
|
||
&i.ReactionUnread,
|
||
&i.Pinned,
|
||
&i.ViaBotID,
|
||
&i.GroupedID,
|
||
&i.Effect,
|
||
&i.ReplyMarkupJson,
|
||
&i.RichMessageJson,
|
||
&i.MessageDeleted,
|
||
&i.PeerUserID,
|
||
&i.PeerAccessHash,
|
||
&i.PeerPhone,
|
||
&i.PeerFirstName,
|
||
&i.PeerLastName,
|
||
&i.PeerUsername,
|
||
&i.PeerCountryCode,
|
||
&i.PeerVerified,
|
||
&i.PeerSupport,
|
||
&i.PeerIsBot,
|
||
&i.PeerBotInfoVersion,
|
||
&i.PeerPremiumUntil,
|
||
&i.PeerEmojiStatusDocumentID,
|
||
&i.PeerEmojiStatusUntil,
|
||
&i.FromUserUserID,
|
||
&i.FromUserAccessHash,
|
||
&i.FromUserPhone,
|
||
&i.FromUserFirstName,
|
||
&i.FromUserLastName,
|
||
&i.FromUserUsername,
|
||
&i.FromUserCountryCode,
|
||
&i.FromUserVerified,
|
||
&i.FromUserSupport,
|
||
&i.FromUserIsBot,
|
||
&i.FromUserBotInfoVersion,
|
||
&i.FromUserPremiumUntil,
|
||
&i.FromUserEmojiStatusDocumentID,
|
||
&i.FromUserEmojiStatusUntil,
|
||
&i.FwdUserID,
|
||
&i.FwdUserAccessHash,
|
||
&i.FwdUserPhone,
|
||
&i.FwdUserFirstName,
|
||
&i.FwdUserLastName,
|
||
&i.FwdUserUsername,
|
||
&i.FwdUserCountryCode,
|
||
&i.FwdUserVerified,
|
||
&i.FwdUserSupport,
|
||
&i.FwdUserIsBot,
|
||
&i.FwdUserBotInfoVersion,
|
||
&i.FwdUserPremiumUntil,
|
||
&i.FwdUserEmojiStatusDocumentID,
|
||
&i.FwdUserEmojiStatusUntil,
|
||
&i.ReplyUserID,
|
||
&i.ReplyUserAccessHash,
|
||
&i.ReplyUserPhone,
|
||
&i.ReplyUserFirstName,
|
||
&i.ReplyUserLastName,
|
||
&i.ReplyUserUsername,
|
||
&i.ReplyUserCountryCode,
|
||
&i.ReplyUserVerified,
|
||
&i.ReplyUserSupport,
|
||
&i.ReplyUserIsBot,
|
||
&i.ReplyUserBotInfoVersion,
|
||
&i.ReplyUserPremiumUntil,
|
||
&i.ReplyUserEmojiStatusDocumentID,
|
||
&i.ReplyUserEmojiStatusUntil,
|
||
); err != nil {
|
||
return nil, err
|
||
}
|
||
items = append(items, i)
|
||
}
|
||
if err := rows.Err(); err != nil {
|
||
return nil, err
|
||
}
|
||
return items, nil
|
||
}
|
||
|
||
const claimDispatchOutbox = `-- name: ClaimDispatchOutbox :many
|
||
WITH picked_heads AS (
|
||
SELECT h.target_user_id, h.head_id
|
||
FROM dispatch_outbox_user_heads h
|
||
WHERE (
|
||
h.status = 'pending'
|
||
AND h.next_attempt_at <= now()
|
||
)
|
||
OR (
|
||
h.status = 'dispatching'
|
||
AND h.updated_at < now() - make_interval(secs => $1::int)
|
||
)
|
||
ORDER BY h.next_attempt_at ASC, h.target_user_id ASC, h.head_pts ASC, h.head_id ASC
|
||
LIMIT $2
|
||
FOR UPDATE OF h SKIP LOCKED
|
||
)
|
||
UPDATE dispatch_outbox d
|
||
SET
|
||
status = 'dispatching',
|
||
attempts = d.attempts + 1,
|
||
updated_at = now()
|
||
FROM picked_heads p
|
||
WHERE d.target_user_id = p.target_user_id
|
||
AND d.id = p.head_id
|
||
RETURNING
|
||
d.id,
|
||
d.target_user_id,
|
||
d.pts,
|
||
d.event_type,
|
||
d.exclude_auth_key_id,
|
||
d.exclude_session_id,
|
||
d.attempts
|
||
`
|
||
|
||
type ClaimDispatchOutboxParams struct {
|
||
LeaseSeconds int32
|
||
LimitCount int32
|
||
}
|
||
|
||
type ClaimDispatchOutboxRow struct {
|
||
ID int64
|
||
TargetUserID int64
|
||
Pts int32
|
||
EventType string
|
||
ExcludeAuthKeyID int64
|
||
ExcludeSessionID int64
|
||
Attempts int32
|
||
}
|
||
|
||
// durable head 表只保留每用户一行，并同步 head 的 readiness。claim 先锁
|
||
// lane head 再更新对应 outbox 行，既不会扫描 backlog，也不会并发领取同一用户。
|
||
func (q *Queries) ClaimDispatchOutbox(ctx context.Context, arg ClaimDispatchOutboxParams) ([]ClaimDispatchOutboxRow, error) {
|
||
rows, err := q.db.Query(ctx, claimDispatchOutbox, arg.LeaseSeconds, arg.LimitCount)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
defer rows.Close()
|
||
var items []ClaimDispatchOutboxRow
|
||
for rows.Next() {
|
||
var i ClaimDispatchOutboxRow
|
||
if err := rows.Scan(
|
||
&i.ID,
|
||
&i.TargetUserID,
|
||
&i.Pts,
|
||
&i.EventType,
|
||
&i.ExcludeAuthKeyID,
|
||
&i.ExcludeSessionID,
|
||
&i.Attempts,
|
||
); err != nil {
|
||
return nil, err
|
||
}
|
||
items = append(items, i)
|
||
}
|
||
if err := rows.Err(); err != nil {
|
||
return nil, err
|
||
}
|
||
return items, nil
|
||
}
|
||
|
||
const claimDispatchOutboxShards = `-- name: ClaimDispatchOutboxShards :many
|
||
WITH picked_heads AS (
|
||
SELECT h.target_user_id, h.head_id
|
||
FROM dispatch_outbox_user_heads h
|
||
-- 256 与 store.DispatchOutboxLogicalShards、0069 generated column 是同一
|
||
-- schema 常量；不得随 worker 数变化。
|
||
WHERE h.logical_shard = ANY($1::smallint[])
|
||
AND (
|
||
(
|
||
h.status = 'pending'
|
||
AND h.next_attempt_at <= now()
|
||
)
|
||
OR (
|
||
h.status = 'dispatching'
|
||
AND h.updated_at < now() - make_interval(secs => $2::int)
|
||
)
|
||
)
|
||
ORDER BY h.next_attempt_at ASC, h.target_user_id ASC, h.head_pts ASC, h.head_id ASC
|
||
LIMIT $3
|
||
FOR UPDATE OF h SKIP LOCKED
|
||
)
|
||
UPDATE dispatch_outbox d
|
||
SET
|
||
status = 'dispatching',
|
||
attempts = d.attempts + 1,
|
||
updated_at = now()
|
||
FROM picked_heads p
|
||
WHERE d.target_user_id = p.target_user_id
|
||
AND d.id = p.head_id
|
||
RETURNING
|
||
d.id,
|
||
d.target_user_id,
|
||
d.pts,
|
||
d.event_type,
|
||
d.exclude_auth_key_id,
|
||
d.exclude_session_id,
|
||
d.attempts
|
||
`
|
||
|
||
type ClaimDispatchOutboxShardsParams struct {
|
||
ShardIds []int16
|
||
LeaseSeconds int32
|
||
LimitCount int32
|
||
}
|
||
|
||
type ClaimDispatchOutboxShardsRow struct {
|
||
ID int64
|
||
TargetUserID int64
|
||
Pts int32
|
||
EventType string
|
||
ExcludeAuthKeyID int64
|
||
ExcludeSessionID int64
|
||
Attempts int32
|
||
}
|
||
|
||
// 固定 logical shard 由 target_user_id 决定；运行时 worker 只领取分配给自己的
|
||
// shard 集合，因此同一用户永远只有一条串行 lane，而不同用户可并行。
|
||
func (q *Queries) ClaimDispatchOutboxShards(ctx context.Context, arg ClaimDispatchOutboxShardsParams) ([]ClaimDispatchOutboxShardsRow, error) {
|
||
rows, err := q.db.Query(ctx, claimDispatchOutboxShards, arg.ShardIds, arg.LeaseSeconds, arg.LimitCount)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
defer rows.Close()
|
||
var items []ClaimDispatchOutboxShardsRow
|
||
for rows.Next() {
|
||
var i ClaimDispatchOutboxShardsRow
|
||
if err := rows.Scan(
|
||
&i.ID,
|
||
&i.TargetUserID,
|
||
&i.Pts,
|
||
&i.EventType,
|
||
&i.ExcludeAuthKeyID,
|
||
&i.ExcludeSessionID,
|
||
&i.Attempts,
|
||
); err != nil {
|
||
return nil, err
|
||
}
|
||
items = append(items, i)
|
||
}
|
||
if err := rows.Err(); err != nil {
|
||
return nil, err
|
||
}
|
||
return items, nil
|
||
}
|
||
|
||
const deleteFailedDispatchOutbox = `-- name: DeleteFailedDispatchOutbox :one
|
||
WITH doomed AS MATERIALIZED (
|
||
SELECT h.target_user_id, h.head_id AS id
|
||
FROM dispatch_outbox_user_heads h
|
||
WHERE h.status = 'failed'
|
||
AND h.updated_at < now() - make_interval(secs => $1::int)
|
||
AND h.target_user_id = ANY($2::bigint[])
|
||
ORDER BY h.updated_at ASC, h.target_user_id ASC, h.head_id ASC
|
||
LIMIT $3
|
||
FOR UPDATE OF h SKIP LOCKED
|
||
),
|
||
deleted AS (
|
||
DELETE FROM dispatch_outbox d
|
||
USING doomed x
|
||
WHERE d.target_user_id = x.target_user_id
|
||
AND d.id = x.id
|
||
RETURNING d.id
|
||
)
|
||
SELECT count(*)::int AS deleted_count
|
||
FROM deleted
|
||
`
|
||
|
||
type DeleteFailedDispatchOutboxParams struct {
|
||
OlderThanSeconds int32
|
||
TargetUserIds []int64
|
||
LimitCount int32
|
||
}
|
||
|
||
// failed 只能成为 lane head；从 head 表开始并先锁 head，既走 0074 的小索引，也与
|
||
// claim/completion 保持同一 user_heads→outbox 锁序。删除的只是在线任务，durable
|
||
// user_update_events 不动，故客户端仍可经 difference 恢复。
|
||
func (q *Queries) DeleteFailedDispatchOutbox(ctx context.Context, arg DeleteFailedDispatchOutboxParams) (int32, error) {
|
||
row := q.db.QueryRow(ctx, deleteFailedDispatchOutbox, arg.OlderThanSeconds, arg.TargetUserIds, arg.LimitCount)
|
||
var deleted_count int32
|
||
err := row.Scan(&deleted_count)
|
||
return deleted_count, err
|
||
}
|
||
|
||
const enqueueDispatch = `-- name: EnqueueDispatch :exec
|
||
INSERT INTO dispatch_outbox (
|
||
target_user_id,
|
||
pts,
|
||
event_type,
|
||
exclude_auth_key_id,
|
||
exclude_session_id
|
||
) VALUES (
|
||
$1, $2, $3, $4, $5
|
||
)
|
||
ON CONFLICT DO NOTHING
|
||
`
|
||
|
||
type EnqueueDispatchParams struct {
|
||
TargetUserID int64
|
||
Pts int32
|
||
EventType string
|
||
ExcludeAuthKeyID int64
|
||
ExcludeSessionID int64
|
||
}
|
||
|
||
func (q *Queries) EnqueueDispatch(ctx context.Context, arg EnqueueDispatchParams) error {
|
||
_, err := q.db.Exec(ctx, enqueueDispatch,
|
||
arg.TargetUserID,
|
||
arg.Pts,
|
||
arg.EventType,
|
||
arg.ExcludeAuthKeyID,
|
||
arg.ExcludeSessionID,
|
||
)
|
||
return err
|
||
}
|
||
|
||
const getUserUpdateWatermark = `-- name: GetUserUpdateWatermark :one
|
||
SELECT contiguous_pts
|
||
FROM user_update_watermarks
|
||
WHERE user_id = $1
|
||
`
|
||
|
||
func (q *Queries) GetUserUpdateWatermark(ctx context.Context, userID int64) (int32, error) {
|
||
row := q.db.QueryRow(ctx, getUserUpdateWatermark, userID)
|
||
var contiguous_pts int32
|
||
err := row.Scan(&contiguous_pts)
|
||
return contiguous_pts, err
|
||
}
|
||
|
||
const listUserUpdateEventsAfter = `-- name: ListUserUpdateEventsAfter :many
|
||
SELECT
|
||
e.user_id,
|
||
e.pts,
|
||
e.pts_count,
|
||
e.date,
|
||
e.event_type,
|
||
e.event_bool,
|
||
e.event_phone,
|
||
COALESCE(e.event_peers::text, '[]')::text AS event_peers_json,
|
||
COALESCE(e.peer_settings::text, '{}')::text AS peer_settings_json,
|
||
COALESCE(e.message_ids::text, '[]')::text AS message_ids_json,
|
||
COALESCE(e.dialog_filter::text, '{}')::text AS dialog_filter_json,
|
||
COALESCE(e.filter_order::text, '[]')::text AS filter_order_json,
|
||
COALESCE(e.folder_peers::text, '[]')::text AS folder_peers_json,
|
||
COALESCE(e.story_payload::text, '{}')::text AS story_payload_json,
|
||
COALESCE(e.reaction_payload::text, '{}')::text AS reaction_payload_json,
|
||
COALESCE(e.emoji_status_payload::text, '{}')::text AS emoji_status_payload_json,
|
||
COALESCE(e.peer_type, '')::text AS event_peer_type,
|
||
COALESCE(e.peer_id, 0)::bigint AS event_peer_id,
|
||
e.filter_id,
|
||
e.max_id,
|
||
e.still_unread_count,
|
||
e.channel_pts,
|
||
e.tags_enabled,
|
||
e.folder_id,
|
||
COALESCE(m.box_id, 0)::int AS message_id,
|
||
COALESCE(m.private_message_id, 0)::bigint AS private_message_id,
|
||
COALESCE(m.owner_user_id, 0)::bigint AS owner_user_id,
|
||
COALESCE(m.peer_type, '')::text AS peer_type,
|
||
COALESCE(m.peer_id, 0)::bigint AS peer_id,
|
||
COALESCE(m.from_user_id, 0)::bigint AS from_user_id,
|
||
COALESCE(m.message_date, 0)::int AS message_date,
|
||
COALESCE(m.ttl_period, 0)::int AS ttl_period,
|
||
COALESCE(m.expires_at, 0)::int AS expires_at,
|
||
COALESCE(m.edit_date, 0)::int AS edit_date,
|
||
COALESCE(m.hide_edited, false)::boolean AS hide_edited,
|
||
COALESCE(m.outgoing, false)::boolean AS outgoing,
|
||
COALESCE(m.body, '')::text AS body,
|
||
COALESCE(m.entities::text, '[]')::text AS message_entities_json,
|
||
COALESCE(m.silent, false)::boolean AS silent,
|
||
COALESCE(m.noforwards, false)::boolean AS noforwards,
|
||
COALESCE(m.reply_to_msg_id, 0)::int AS reply_to_msg_id,
|
||
COALESCE(m.reply_to_peer_type, '')::text AS reply_to_peer_type,
|
||
COALESCE(m.reply_to_peer_id, 0)::bigint AS reply_to_peer_id,
|
||
COALESCE(m.reply_to_top_id, 0)::int AS reply_to_top_id,
|
||
COALESCE(m.reply_to_story_id, 0)::int AS reply_to_story_id,
|
||
COALESCE(m.quote_text, '')::text AS quote_text,
|
||
COALESCE(m.quote_entities::text, '[]')::text AS quote_entities_json,
|
||
COALESCE(m.quote_offset, 0)::int AS quote_offset,
|
||
COALESCE(m.reply_external, '{}'::jsonb)::text AS reply_external_json,
|
||
COALESCE(m.fwd_from_peer_type, '')::text AS fwd_from_peer_type,
|
||
COALESCE(m.fwd_from_peer_id, 0)::bigint AS fwd_from_peer_id,
|
||
COALESCE(m.fwd_from_name, '')::text AS fwd_from_name,
|
||
COALESCE(m.fwd_date, 0)::int AS fwd_date,
|
||
COALESCE(m.fwd_saved_from_peer_type, '')::text AS fwd_saved_from_peer_type,
|
||
COALESCE(m.fwd_saved_from_peer_id, 0)::bigint AS fwd_saved_from_peer_id,
|
||
COALESCE(m.fwd_saved_from_msg_id, 0)::int AS fwd_saved_from_msg_id,
|
||
COALESCE(m.saved_peer_type, '')::text AS saved_peer_type,
|
||
COALESCE(m.saved_peer_id, 0)::bigint AS saved_peer_id,
|
||
COALESCE(m.media::text, '{}')::text AS media_json,
|
||
COALESCE(m.media_unread, false)::boolean AS media_unread,
|
||
COALESCE(m.reaction_unread, false)::boolean AS reaction_unread,
|
||
COALESCE(m.pinned, false)::boolean AS pinned,
|
||
COALESCE(m.via_bot_id, 0)::bigint AS via_bot_id,
|
||
COALESCE(m.grouped_id, 0)::bigint AS grouped_id,
|
||
COALESCE(m.effect, 0)::bigint AS effect,
|
||
COALESCE(m.reply_markup::text, '{}')::text AS reply_markup_json,
|
||
COALESCE(m.rich_message::text, '{}')::text AS rich_message_json,
|
||
COALESCE(m.deleted, false)::boolean AS message_deleted,
|
||
COALESCE(peer_u.id, 0)::bigint AS peer_user_id,
|
||
COALESCE(peer_u.access_hash, 0)::bigint AS peer_access_hash,
|
||
COALESCE(peer_u.phone, '')::text AS peer_phone,
|
||
COALESCE(peer_u.first_name, '')::text AS peer_first_name,
|
||
COALESCE(peer_u.last_name, '')::text AS peer_last_name,
|
||
COALESCE(peer_u.username, '')::text AS peer_username,
|
||
COALESCE(peer_u.country_code, '')::text AS peer_country_code,
|
||
COALESCE(peer_u.verified, false)::boolean AS peer_verified,
|
||
COALESCE(peer_u.support, false)::boolean AS peer_support,
|
||
COALESCE(peer_u.is_bot, false)::boolean AS peer_is_bot,
|
||
COALESCE(peer_u.bot_info_version, 0)::int AS peer_bot_info_version,
|
||
COALESCE(EXTRACT(EPOCH FROM peer_u.premium_expires_at), 0)::bigint AS peer_premium_until,
|
||
COALESCE(peer_u.emoji_status_document_id, 0)::bigint AS peer_emoji_status_document_id,
|
||
COALESCE(peer_u.emoji_status_until, 0)::bigint AS peer_emoji_status_until,
|
||
COALESCE(from_u.id, 0)::bigint AS from_user_user_id,
|
||
COALESCE(from_u.access_hash, 0)::bigint AS from_user_access_hash,
|
||
COALESCE(from_u.phone, '')::text AS from_user_phone,
|
||
COALESCE(from_u.first_name, '')::text AS from_user_first_name,
|
||
COALESCE(from_u.last_name, '')::text AS from_user_last_name,
|
||
COALESCE(from_u.username, '')::text AS from_user_username,
|
||
COALESCE(from_u.country_code, '')::text AS from_user_country_code,
|
||
COALESCE(from_u.verified, false)::boolean AS from_user_verified,
|
||
COALESCE(from_u.support, false)::boolean AS from_user_support,
|
||
COALESCE(from_u.is_bot, false)::boolean AS from_user_is_bot,
|
||
COALESCE(from_u.bot_info_version, 0)::int AS from_user_bot_info_version,
|
||
COALESCE(EXTRACT(EPOCH FROM from_u.premium_expires_at), 0)::bigint AS from_user_premium_until,
|
||
COALESCE(from_u.emoji_status_document_id, 0)::bigint AS from_user_emoji_status_document_id,
|
||
COALESCE(from_u.emoji_status_until, 0)::bigint AS from_user_emoji_status_until,
|
||
COALESCE(fwd_u.id, 0)::bigint AS fwd_user_id,
|
||
COALESCE(fwd_u.access_hash, 0)::bigint AS fwd_user_access_hash,
|
||
COALESCE(fwd_u.phone, '')::text AS fwd_user_phone,
|
||
COALESCE(fwd_u.first_name, '')::text AS fwd_user_first_name,
|
||
COALESCE(fwd_u.last_name, '')::text AS fwd_user_last_name,
|
||
COALESCE(fwd_u.username, '')::text AS fwd_user_username,
|
||
COALESCE(fwd_u.country_code, '')::text AS fwd_user_country_code,
|
||
COALESCE(fwd_u.verified, false)::boolean AS fwd_user_verified,
|
||
COALESCE(fwd_u.support, false)::boolean AS fwd_user_support,
|
||
COALESCE(fwd_u.is_bot, false)::boolean AS fwd_user_is_bot,
|
||
COALESCE(fwd_u.bot_info_version, 0)::int AS fwd_user_bot_info_version,
|
||
COALESCE(EXTRACT(EPOCH FROM fwd_u.premium_expires_at), 0)::bigint AS fwd_user_premium_until,
|
||
COALESCE(fwd_u.emoji_status_document_id, 0)::bigint AS fwd_user_emoji_status_document_id,
|
||
COALESCE(fwd_u.emoji_status_until, 0)::bigint AS fwd_user_emoji_status_until,
|
||
COALESCE(reply_u.id, 0)::bigint AS reply_user_id,
|
||
COALESCE(reply_u.access_hash, 0)::bigint AS reply_user_access_hash,
|
||
COALESCE(reply_u.phone, '')::text AS reply_user_phone,
|
||
COALESCE(reply_u.first_name, '')::text AS reply_user_first_name,
|
||
COALESCE(reply_u.last_name, '')::text AS reply_user_last_name,
|
||
COALESCE(reply_u.username, '')::text AS reply_user_username,
|
||
COALESCE(reply_u.country_code, '')::text AS reply_user_country_code,
|
||
COALESCE(reply_u.verified, false)::boolean AS reply_user_verified,
|
||
COALESCE(reply_u.support, false)::boolean AS reply_user_support,
|
||
COALESCE(reply_u.is_bot, false)::boolean AS reply_user_is_bot,
|
||
COALESCE(reply_u.bot_info_version, 0)::int AS reply_user_bot_info_version,
|
||
COALESCE(EXTRACT(EPOCH FROM reply_u.premium_expires_at), 0)::bigint AS reply_user_premium_until,
|
||
COALESCE(reply_u.emoji_status_document_id, 0)::bigint AS reply_user_emoji_status_document_id,
|
||
COALESCE(reply_u.emoji_status_until, 0)::bigint AS reply_user_emoji_status_until
|
||
FROM user_update_events e
|
||
LEFT JOIN message_boxes m ON m.owner_user_id = e.user_id AND m.box_id = e.message_box_id
|
||
LEFT JOIN users peer_u ON m.peer_type = 'user' AND peer_u.id = m.peer_id
|
||
LEFT JOIN users from_u ON from_u.id = m.from_user_id
|
||
LEFT JOIN users fwd_u ON m.fwd_from_peer_type = 'user' AND fwd_u.id = m.fwd_from_peer_id
|
||
LEFT JOIN users reply_u ON m.reply_to_peer_type = 'user' AND reply_u.id = m.reply_to_peer_id
|
||
WHERE e.user_id = $1
|
||
AND e.pts > $2
|
||
ORDER BY e.pts ASC
|
||
LIMIT $3
|
||
`
|
||
|
||
type ListUserUpdateEventsAfterParams struct {
|
||
UserID int64
|
||
Pts int32
|
||
LimitCount int32
|
||
}
|
||
|
||
type ListUserUpdateEventsAfterRow struct {
|
||
UserID int64
|
||
Pts int32
|
||
PtsCount int32
|
||
Date int32
|
||
EventType string
|
||
EventBool bool
|
||
EventPhone string
|
||
EventPeersJson string
|
||
PeerSettingsJson string
|
||
MessageIdsJson string
|
||
DialogFilterJson string
|
||
FilterOrderJson string
|
||
FolderPeersJson string
|
||
StoryPayloadJson string
|
||
ReactionPayloadJson string
|
||
EmojiStatusPayloadJson string
|
||
EventPeerType string
|
||
EventPeerID int64
|
||
FilterID int32
|
||
MaxID int32
|
||
StillUnreadCount int32
|
||
ChannelPts int32
|
||
TagsEnabled bool
|
||
FolderID int32
|
||
MessageID int32
|
||
PrivateMessageID int64
|
||
OwnerUserID int64
|
||
PeerType string
|
||
PeerID int64
|
||
FromUserID int64
|
||
MessageDate int32
|
||
TtlPeriod int32
|
||
ExpiresAt int32
|
||
EditDate int32
|
||
HideEdited bool
|
||
Outgoing bool
|
||
Body string
|
||
MessageEntitiesJson string
|
||
Silent bool
|
||
Noforwards bool
|
||
ReplyToMsgID int32
|
||
ReplyToPeerType string
|
||
ReplyToPeerID int64
|
||
ReplyToTopID int32
|
||
ReplyToStoryID int32
|
||
QuoteText string
|
||
QuoteEntitiesJson string
|
||
QuoteOffset int32
|
||
ReplyExternalJson string
|
||
FwdFromPeerType string
|
||
FwdFromPeerID int64
|
||
FwdFromName string
|
||
FwdDate int32
|
||
FwdSavedFromPeerType string
|
||
FwdSavedFromPeerID int64
|
||
FwdSavedFromMsgID int32
|
||
SavedPeerType string
|
||
SavedPeerID int64
|
||
MediaJson string
|
||
MediaUnread bool
|
||
ReactionUnread bool
|
||
Pinned bool
|
||
ViaBotID int64
|
||
GroupedID int64
|
||
Effect int64
|
||
ReplyMarkupJson string
|
||
RichMessageJson string
|
||
MessageDeleted bool
|
||
PeerUserID int64
|
||
PeerAccessHash int64
|
||
PeerPhone string
|
||
PeerFirstName string
|
||
PeerLastName string
|
||
PeerUsername string
|
||
PeerCountryCode string
|
||
PeerVerified bool
|
||
PeerSupport bool
|
||
PeerIsBot bool
|
||
PeerBotInfoVersion int32
|
||
PeerPremiumUntil int64
|
||
PeerEmojiStatusDocumentID int64
|
||
PeerEmojiStatusUntil int64
|
||
FromUserUserID int64
|
||
FromUserAccessHash int64
|
||
FromUserPhone string
|
||
FromUserFirstName string
|
||
FromUserLastName string
|
||
FromUserUsername string
|
||
FromUserCountryCode string
|
||
FromUserVerified bool
|
||
FromUserSupport bool
|
||
FromUserIsBot bool
|
||
FromUserBotInfoVersion int32
|
||
FromUserPremiumUntil int64
|
||
FromUserEmojiStatusDocumentID int64
|
||
FromUserEmojiStatusUntil int64
|
||
FwdUserID int64
|
||
FwdUserAccessHash int64
|
||
FwdUserPhone string
|
||
FwdUserFirstName string
|
||
FwdUserLastName string
|
||
FwdUserUsername string
|
||
FwdUserCountryCode string
|
||
FwdUserVerified bool
|
||
FwdUserSupport bool
|
||
FwdUserIsBot bool
|
||
FwdUserBotInfoVersion int32
|
||
FwdUserPremiumUntil int64
|
||
FwdUserEmojiStatusDocumentID int64
|
||
FwdUserEmojiStatusUntil int64
|
||
ReplyUserID int64
|
||
ReplyUserAccessHash int64
|
||
ReplyUserPhone string
|
||
ReplyUserFirstName string
|
||
ReplyUserLastName string
|
||
ReplyUserUsername string
|
||
ReplyUserCountryCode string
|
||
ReplyUserVerified bool
|
||
ReplyUserSupport bool
|
||
ReplyUserIsBot bool
|
||
ReplyUserBotInfoVersion int32
|
||
ReplyUserPremiumUntil int64
|
||
ReplyUserEmojiStatusDocumentID int64
|
||
ReplyUserEmojiStatusUntil int64
|
||
}
|
||
|
||
func (q *Queries) ListUserUpdateEventsAfter(ctx context.Context, arg ListUserUpdateEventsAfterParams) ([]ListUserUpdateEventsAfterRow, error) {
|
||
rows, err := q.db.Query(ctx, listUserUpdateEventsAfter, arg.UserID, arg.Pts, arg.LimitCount)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
defer rows.Close()
|
||
var items []ListUserUpdateEventsAfterRow
|
||
for rows.Next() {
|
||
var i ListUserUpdateEventsAfterRow
|
||
if err := rows.Scan(
|
||
&i.UserID,
|
||
&i.Pts,
|
||
&i.PtsCount,
|
||
&i.Date,
|
||
&i.EventType,
|
||
&i.EventBool,
|
||
&i.EventPhone,
|
||
&i.EventPeersJson,
|
||
&i.PeerSettingsJson,
|
||
&i.MessageIdsJson,
|
||
&i.DialogFilterJson,
|
||
&i.FilterOrderJson,
|
||
&i.FolderPeersJson,
|
||
&i.StoryPayloadJson,
|
||
&i.ReactionPayloadJson,
|
||
&i.EmojiStatusPayloadJson,
|
||
&i.EventPeerType,
|
||
&i.EventPeerID,
|
||
&i.FilterID,
|
||
&i.MaxID,
|
||
&i.StillUnreadCount,
|
||
&i.ChannelPts,
|
||
&i.TagsEnabled,
|
||
&i.FolderID,
|
||
&i.MessageID,
|
||
&i.PrivateMessageID,
|
||
&i.OwnerUserID,
|
||
&i.PeerType,
|
||
&i.PeerID,
|
||
&i.FromUserID,
|
||
&i.MessageDate,
|
||
&i.TtlPeriod,
|
||
&i.ExpiresAt,
|
||
&i.EditDate,
|
||
&i.HideEdited,
|
||
&i.Outgoing,
|
||
&i.Body,
|
||
&i.MessageEntitiesJson,
|
||
&i.Silent,
|
||
&i.Noforwards,
|
||
&i.ReplyToMsgID,
|
||
&i.ReplyToPeerType,
|
||
&i.ReplyToPeerID,
|
||
&i.ReplyToTopID,
|
||
&i.ReplyToStoryID,
|
||
&i.QuoteText,
|
||
&i.QuoteEntitiesJson,
|
||
&i.QuoteOffset,
|
||
&i.ReplyExternalJson,
|
||
&i.FwdFromPeerType,
|
||
&i.FwdFromPeerID,
|
||
&i.FwdFromName,
|
||
&i.FwdDate,
|
||
&i.FwdSavedFromPeerType,
|
||
&i.FwdSavedFromPeerID,
|
||
&i.FwdSavedFromMsgID,
|
||
&i.SavedPeerType,
|
||
&i.SavedPeerID,
|
||
&i.MediaJson,
|
||
&i.MediaUnread,
|
||
&i.ReactionUnread,
|
||
&i.Pinned,
|
||
&i.ViaBotID,
|
||
&i.GroupedID,
|
||
&i.Effect,
|
||
&i.ReplyMarkupJson,
|
||
&i.RichMessageJson,
|
||
&i.MessageDeleted,
|
||
&i.PeerUserID,
|
||
&i.PeerAccessHash,
|
||
&i.PeerPhone,
|
||
&i.PeerFirstName,
|
||
&i.PeerLastName,
|
||
&i.PeerUsername,
|
||
&i.PeerCountryCode,
|
||
&i.PeerVerified,
|
||
&i.PeerSupport,
|
||
&i.PeerIsBot,
|
||
&i.PeerBotInfoVersion,
|
||
&i.PeerPremiumUntil,
|
||
&i.PeerEmojiStatusDocumentID,
|
||
&i.PeerEmojiStatusUntil,
|
||
&i.FromUserUserID,
|
||
&i.FromUserAccessHash,
|
||
&i.FromUserPhone,
|
||
&i.FromUserFirstName,
|
||
&i.FromUserLastName,
|
||
&i.FromUserUsername,
|
||
&i.FromUserCountryCode,
|
||
&i.FromUserVerified,
|
||
&i.FromUserSupport,
|
||
&i.FromUserIsBot,
|
||
&i.FromUserBotInfoVersion,
|
||
&i.FromUserPremiumUntil,
|
||
&i.FromUserEmojiStatusDocumentID,
|
||
&i.FromUserEmojiStatusUntil,
|
||
&i.FwdUserID,
|
||
&i.FwdUserAccessHash,
|
||
&i.FwdUserPhone,
|
||
&i.FwdUserFirstName,
|
||
&i.FwdUserLastName,
|
||
&i.FwdUserUsername,
|
||
&i.FwdUserCountryCode,
|
||
&i.FwdUserVerified,
|
||
&i.FwdUserSupport,
|
||
&i.FwdUserIsBot,
|
||
&i.FwdUserBotInfoVersion,
|
||
&i.FwdUserPremiumUntil,
|
||
&i.FwdUserEmojiStatusDocumentID,
|
||
&i.FwdUserEmojiStatusUntil,
|
||
&i.ReplyUserID,
|
||
&i.ReplyUserAccessHash,
|
||
&i.ReplyUserPhone,
|
||
&i.ReplyUserFirstName,
|
||
&i.ReplyUserLastName,
|
||
&i.ReplyUserUsername,
|
||
&i.ReplyUserCountryCode,
|
||
&i.ReplyUserVerified,
|
||
&i.ReplyUserSupport,
|
||
&i.ReplyUserIsBot,
|
||
&i.ReplyUserBotInfoVersion,
|
||
&i.ReplyUserPremiumUntil,
|
||
&i.ReplyUserEmojiStatusDocumentID,
|
||
&i.ReplyUserEmojiStatusUntil,
|
||
); err != nil {
|
||
return nil, err
|
||
}
|
||
items = append(items, i)
|
||
}
|
||
if err := rows.Err(); err != nil {
|
||
return nil, err
|
||
}
|
||
return items, nil
|
||
}
|
||
|
||
const markDispatchDelivered = `-- name: MarkDispatchDelivered :execrows
|
||
WITH locked_head AS MATERIALIZED (
|
||
SELECT h.target_user_id
|
||
FROM dispatch_outbox_user_heads h
|
||
WHERE h.target_user_id = $3::bigint
|
||
FOR UPDATE
|
||
)
|
||
DELETE FROM dispatch_outbox d
|
||
USING locked_head h
|
||
WHERE d.target_user_id = h.target_user_id
|
||
AND d.id = $1::bigint
|
||
AND d.status = 'dispatching'
|
||
AND d.attempts = $2::int
|
||
`
|
||
|
||
type MarkDispatchDeliveredParams struct {
|
||
ID int64
|
||
ExpectedAttempts int32
|
||
TargetUserID int64
|
||
}
|
||
|
||
// 方案 A：投递成功即删除。outbox 是任务队列，delivered 行无保留价值
|
||
// （消息在 message_boxes、离线补偿在 user_update_events），删除让表维持「未完成任务」小稳态。
|
||
// claim 的锁序是 user_heads→outbox；completion 必须先显式锁同一 head 再删 outbox，
|
||
// 否则租约过期 claim 与完成恰好竞争时会形成 outbox→head / head→outbox 环路。
|
||
func (q *Queries) MarkDispatchDelivered(ctx context.Context, arg MarkDispatchDeliveredParams) (int64, error) {
|
||
result, err := q.db.Exec(ctx, markDispatchDelivered, arg.ID, arg.ExpectedAttempts, arg.TargetUserID)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
return result.RowsAffected(), nil
|
||
}
|
||
|
||
const markDispatchDeliveredBatch = `-- name: MarkDispatchDeliveredBatch :execrows
|
||
WITH input AS MATERIALIZED (
|
||
SELECT tu.target_user_id, di.id, ea.attempts
|
||
FROM unnest($1::bigint[]) WITH ORDINALITY AS tu(target_user_id, ord)
|
||
JOIN unnest($2::bigint[]) WITH ORDINALITY AS di(id, ord) USING (ord)
|
||
JOIN unnest($3::int[]) WITH ORDINALITY AS ea(attempts, ord) USING (ord)
|
||
),
|
||
locked_heads AS MATERIALIZED (
|
||
SELECT h.target_user_id
|
||
FROM dispatch_outbox_user_heads h
|
||
JOIN (SELECT DISTINCT target_user_id FROM input) i USING (target_user_id)
|
||
-- Match ClaimDispatchOutbox[Shards] exactly. A stale-lease claim may lock several
|
||
-- dispatching heads while this completion batch locks the same set; a different
|
||
-- multi-row order would merely move the deadlock one level up.
|
||
ORDER BY h.next_attempt_at, h.target_user_id, h.head_pts, h.head_id
|
||
FOR UPDATE OF h
|
||
)
|
||
DELETE FROM dispatch_outbox d
|
||
USING input i, locked_heads h
|
||
WHERE d.target_user_id = h.target_user_id
|
||
AND d.target_user_id = i.target_user_id
|
||
AND d.id = i.id
|
||
AND d.status = 'dispatching'
|
||
AND d.attempts = i.attempts
|
||
`
|
||
|
||
type MarkDispatchDeliveredBatchParams struct {
|
||
TargetUserIds []int64
|
||
Ids []int64
|
||
ExpectedAttempts []int32
|
||
}
|
||
|
||
// 批量删除一批已投递的 (target_user_id, id)；target_user_id 入 WHERE 命中唯一索引并避免串删。
|
||
func (q *Queries) MarkDispatchDeliveredBatch(ctx context.Context, arg MarkDispatchDeliveredBatchParams) (int64, error) {
|
||
result, err := q.db.Exec(ctx, markDispatchDeliveredBatch, arg.TargetUserIds, arg.Ids, arg.ExpectedAttempts)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
return result.RowsAffected(), nil
|
||
}
|
||
|
||
const markDispatchFailed = `-- name: MarkDispatchFailed :execrows
|
||
WITH locked_head AS MATERIALIZED (
|
||
SELECT h.target_user_id
|
||
FROM dispatch_outbox_user_heads h
|
||
WHERE h.target_user_id = $4::bigint
|
||
FOR UPDATE
|
||
)
|
||
UPDATE dispatch_outbox d
|
||
SET
|
||
status = CASE WHEN d.attempts >= 5 THEN 'failed' ELSE 'pending' END,
|
||
next_attempt_at = CASE
|
||
WHEN d.attempts >= 5 THEN d.next_attempt_at
|
||
ELSE now() + make_interval(secs => LEAST(60, d.attempts * d.attempts))
|
||
END,
|
||
last_error = $1::text,
|
||
updated_at = now()
|
||
FROM locked_head h
|
||
WHERE d.target_user_id = h.target_user_id
|
||
AND d.id = $2::bigint
|
||
AND d.status = 'dispatching'
|
||
AND d.attempts = $3::int
|
||
`
|
||
|
||
type MarkDispatchFailedParams struct {
|
||
LastError string
|
||
ID int64
|
||
ExpectedAttempts int32
|
||
TargetUserID int64
|
||
}
|
||
|
||
func (q *Queries) MarkDispatchFailed(ctx context.Context, arg MarkDispatchFailedParams) (int64, error) {
|
||
result, err := q.db.Exec(ctx, markDispatchFailed,
|
||
arg.LastError,
|
||
arg.ID,
|
||
arg.ExpectedAttempts,
|
||
arg.TargetUserID,
|
||
)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
return result.RowsAffected(), nil
|
||
}
|
||
|
||
const maxUserPts = `-- name: MaxUserPts :one
|
||
SELECT COALESCE(MAX(pts), 0)::int AS max_pts
|
||
FROM user_update_events
|
||
WHERE user_id = $1
|
||
`
|
||
|
||
func (q *Queries) MaxUserPts(ctx context.Context, userID int64) (int32, error) {
|
||
row := q.db.QueryRow(ctx, maxUserPts, userID)
|
||
var max_pts int32
|
||
err := row.Scan(&max_pts)
|
||
return max_pts, err
|
||
}
|