83 lines
3.2 KiB
Go
83 lines
3.2 KiB
Go
package repository
|
||
|
||
import (
|
||
"backend/pkg/notification/domain/entity"
|
||
"context"
|
||
|
||
"github.com/gocql/gocql"
|
||
)
|
||
|
||
type NotificationRepository interface {
|
||
NotificationEventRepository
|
||
UserNotificationRepository
|
||
NotificationCursorRepository
|
||
}
|
||
|
||
// ---- 1. Event ----
|
||
// 專心管「事件本體」，fan-out 前先寫這張。
|
||
// 通常由上游 domain event consumer 呼叫 Create。
|
||
|
||
type QueryNotificationEventParam struct {
|
||
ObjectID *string
|
||
ObjectType *string
|
||
Limit *int
|
||
}
|
||
|
||
type NotificationEventRepository interface {
|
||
// Create 建立一筆新的 NotificationEvent。
|
||
Create(ctx context.Context, e *entity.NotificationEvent) error
|
||
// GetByID 依 EventID 取得事件。
|
||
GetByID(ctx context.Context, id string) (*entity.NotificationEvent, error)
|
||
// ListByObject 依 object_type + object_id 查詢相關事件（選用，debug / 後台用）。
|
||
ListByObject(ctx context.Context, param QueryNotificationEventParam) ([]*entity.NotificationEvent, error)
|
||
}
|
||
|
||
// ---- 2. 使用者通知：user_notification ----
|
||
// 管使用者的小鈴鐺 row，fan-out 之後用這個寫入。
|
||
|
||
// ListLatestOptions 查列表用的參數
|
||
type ListLatestOptions struct {
|
||
UserID string
|
||
Buckets []string // e.g. []string{"202511", "202510"}
|
||
Limit int // 建議在 service 層限制最大值，例如 <= 100
|
||
}
|
||
|
||
type UserNotificationRepository interface {
|
||
// CreateUserNotification 建立單一通知（針對某一個 user）。
|
||
// 由呼叫端決定 bucket 與 TTL 秒數。
|
||
CreateUserNotification(ctx context.Context, n *entity.UserNotification, ttlSeconds int) error
|
||
|
||
// BulkCreate 批次建立多筆通知（fan-out worker 使用）。
|
||
// 一般期望要嘛全部成功要嘛全部失敗。
|
||
BulkCreate(ctx context.Context, list []*entity.UserNotification, ttlSeconds int) error
|
||
|
||
// ListLatest 取得某 user 最新的通知列表（小鈴鐺拉下來用）。
|
||
ListLatest(ctx context.Context, opt ListLatestOptions) ([]*entity.UserNotification, error)
|
||
|
||
// MarkRead 將單一通知設為已讀。
|
||
// 用 (user_id, bucket, ts) 精準定位那一筆資料。
|
||
MarkRead(ctx context.Context, userID, bucket string, ts gocql.UUID) error
|
||
|
||
// MarkAllRead 將指定 buckets 範圍內的通知設為已讀。
|
||
// 常見用法：最近幾個 bucket（例如最近 30 天）全部標為已讀。
|
||
// Cassandra 不適合全表掃描，實作時可分批 select 再 update。
|
||
MarkAllRead(ctx context.Context, userID string, buckets []string) error
|
||
|
||
// CountUnreadApprox 回傳未讀數（允許是近似值）。
|
||
// 實作方式可以是：
|
||
// - 掃少量 buckets 中 status='UNREAD' 的 row，然後在應用端計算
|
||
// - 或讀取外部 counter（Redis / 另一張 counter table）
|
||
CountUnreadApprox(ctx context.Context, userID string, buckets []string) (int64, error)
|
||
}
|
||
|
||
// ---- 3. NotificationCursorRepository ----
|
||
// 管 last_seen 光標，用來減少大量「每一筆更新已讀」的成本。
|
||
|
||
type NotificationCursorRepository interface {
|
||
// GetCursor 取得某 user 的光標，如果不存在可以回傳 (nil, nil)。
|
||
GetCursor(ctx context.Context, userID string) (*entity.NotificationCursor, error)
|
||
// UpsertCursor 新增或更新光標。
|
||
// 一般在使用者打開通知列表、或捲到最上面時更新。
|
||
UpsertCursor(ctx context.Context, cursor *entity.NotificationCursor) error
|
||
}
|