chore: refresh gramsrv public release

internal/store/postgres/message_store.go

@@ -0,0 +1,55 @@
+package postgres
+
+import (
+	"context"
+
+	"github.com/jackc/pgx/v5"
+	"go.uber.org/zap"
+
+	"telesrv/internal/store"
+	"telesrv/internal/store/postgres/sqlcgen"
+)
+
+// MessageStore 用 PostgreSQL 实现 store.MessageStore。
+type MessageStore struct {
+	db     sqlcgen.DBTX
+	q      *sqlcgen.Queries
+	boxIDs store.BoxIDAllocator
+	log    *zap.Logger
+}
+
+type txBeginner interface {
+	Begin(ctx context.Context) (pgx.Tx, error)
+}
+
+// MessageStoreOption 调整 PostgreSQL MessageStore 依赖。
+type MessageStoreOption func(*MessageStore)
+
+// WithMessageAllocators 注入 Redis-backed box_id allocator。
+func WithMessageAllocators(boxIDs store.BoxIDAllocator) MessageStoreOption {
+	return func(s *MessageStore) {
+		s.boxIDs = boxIDs
+	}
+}
+
+// WithMessageLogger 注入消息 store 日志器，用于追踪消息与 pts 原子写入。
+func WithMessageLogger(log *zap.Logger) MessageStoreOption {
+	return func(s *MessageStore) {
+		s.log = log
+	}
+}
+
+// NewMessageStore 基于 pgx 连接池（或事务）创建 MessageStore。
+func NewMessageStore(db sqlcgen.DBTX, opts ...MessageStoreOption) *MessageStore {
+	s := &MessageStore{db: db, q: sqlcgen.New(db)}
+	for _, opt := range opts {
+		opt(s)
+	}
+	if s.log == nil {
+		s.log = zap.NewNop()
+	}
+	if s.boxIDs == nil {
+		s.boxIDs = pgBoxIDAllocator{s: s}
+	}
+	return s
+}