79 lines
2.5 KiB
Go
79 lines
2.5 KiB
Go
package postgres
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
|
||
"telesrv/internal/store/postgres/sqlcgen"
|
||
)
|
||
|
||
// MessageBoxCounterSource 从 message_boxes durable log 恢复某 owner 的当前最大 box_id。
|
||
type MessageBoxCounterSource struct {
|
||
q *sqlcgen.Queries
|
||
}
|
||
|
||
// NewMessageBoxCounterSource 创建 Redis BoxIDAllocator 的 PG 恢复源。
|
||
func NewMessageBoxCounterSource(db sqlcgen.DBTX) *MessageBoxCounterSource {
|
||
return &MessageBoxCounterSource{q: sqlcgen.New(db)}
|
||
}
|
||
|
||
func (s *MessageBoxCounterSource) Current(ctx context.Context, userID int64) (int, error) {
|
||
v, err := s.q.MaxMessageBoxID(ctx, userID)
|
||
if err != nil {
|
||
return 0, fmt.Errorf("max message box id: %w", err)
|
||
}
|
||
return int(v), nil
|
||
}
|
||
|
||
// ChannelIDCounterSource 从 channels durable 表恢复全局 channel id。
|
||
type ChannelIDCounterSource struct {
|
||
db sqlcgen.DBTX
|
||
}
|
||
|
||
// NewChannelIDCounterSource 创建 Redis ChannelIDAllocator 的 PG 恢复源。
|
||
func NewChannelIDCounterSource(db sqlcgen.DBTX) *ChannelIDCounterSource {
|
||
return &ChannelIDCounterSource{db: db}
|
||
}
|
||
|
||
func (s *ChannelIDCounterSource) Current(ctx context.Context, _ int64) (int, error) {
|
||
var id int
|
||
if err := s.db.QueryRow(ctx, `SELECT COALESCE(MAX(id), 0) FROM channels`).Scan(&id); err != nil {
|
||
return 0, fmt.Errorf("max channel id: %w", err)
|
||
}
|
||
return id, nil
|
||
}
|
||
|
||
// SecretChatIDCounterSource 从 secret_chats 表恢复全局 secret chat id（迁移 0137）。
|
||
type SecretChatIDCounterSource struct {
|
||
db sqlcgen.DBTX
|
||
}
|
||
|
||
// NewSecretChatIDCounterSource 创建 Redis SecretChatIDAllocator 的 PG 恢复源。
|
||
func NewSecretChatIDCounterSource(db sqlcgen.DBTX) *SecretChatIDCounterSource {
|
||
return &SecretChatIDCounterSource{db: db}
|
||
}
|
||
|
||
func (s *SecretChatIDCounterSource) Current(ctx context.Context, _ int64) (int, error) {
|
||
var id int
|
||
if err := s.db.QueryRow(ctx, `SELECT COALESCE(MAX(chat_id), 0) FROM secret_chats`).Scan(&id); err != nil {
|
||
return 0, fmt.Errorf("max secret chat id: %w", err)
|
||
}
|
||
return id, nil
|
||
}
|
||
|
||
// ChannelMessageIDCounterSource 从 channel_messages 恢复某 channel 的当前最大 message id。
|
||
type ChannelMessageIDCounterSource struct {
|
||
db sqlcgen.DBTX
|
||
}
|
||
|
||
func NewChannelMessageIDCounterSource(db sqlcgen.DBTX) *ChannelMessageIDCounterSource {
|
||
return &ChannelMessageIDCounterSource{db: db}
|
||
}
|
||
|
||
func (s *ChannelMessageIDCounterSource) Current(ctx context.Context, channelID int64) (int, error) {
|
||
var id int
|
||
if err := s.db.QueryRow(ctx, `SELECT COALESCE(MAX(id), 0) FROM channel_messages WHERE channel_id = $1`, channelID).Scan(&id); err != nil {
|
||
return 0, fmt.Errorf("max channel message id: %w", err)
|
||
}
|
||
return id, nil
|
||
}
|