user_subscription_port.go 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. package service
  2. import (
  3. "context"
  4. "time"
  5. "github.com/Wei-Shaw/sub2api/internal/pkg/pagination"
  6. )
  7. type UserSubscriptionRepository interface {
  8. Create(ctx context.Context, sub *UserSubscription) error
  9. GetByID(ctx context.Context, id int64) (*UserSubscription, error)
  10. GetByIDForUpdate(ctx context.Context, id int64) (*UserSubscription, error)
  11. GetByIDIncludeDeleted(ctx context.Context, id int64) (*UserSubscription, error)
  12. GetByUserIDAndGroupID(ctx context.Context, userID, groupID int64) (*UserSubscription, error)
  13. GetActiveByUserIDAndGroupID(ctx context.Context, userID, groupID int64) (*UserSubscription, error)
  14. Update(ctx context.Context, sub *UserSubscription) error
  15. Delete(ctx context.Context, id int64) error
  16. Restore(ctx context.Context, subscriptionID int64, restoredStatus string) (*UserSubscription, error)
  17. ListByUserID(ctx context.Context, userID int64) ([]UserSubscription, error)
  18. ListActiveByUserID(ctx context.Context, userID int64) ([]UserSubscription, error)
  19. ListByGroupID(ctx context.Context, groupID int64, params pagination.PaginationParams) ([]UserSubscription, *pagination.PaginationResult, error)
  20. List(ctx context.Context, params pagination.PaginationParams, userID, groupID *int64, status, platform, sortBy, sortOrder string) ([]UserSubscription, *pagination.PaginationResult, error)
  21. ExistsByUserIDAndGroupID(ctx context.Context, userID, groupID int64) (bool, error)
  22. ExistsActiveByUserIDAndGroupID(ctx context.Context, userID, groupID int64) (bool, error)
  23. ExtendExpiry(ctx context.Context, subscriptionID int64, newExpiresAt time.Time) error
  24. UpdateStatus(ctx context.Context, subscriptionID int64, status string) error
  25. UpdateNotes(ctx context.Context, subscriptionID int64, notes string) error
  26. // ActivateWindows 首次使用时激活用量窗口。日窗口按日历日对齐,锚点为当天 0 点
  27. // (dailyStart);周/月窗口为期限对齐滚动窗口,锚点为激活时刻(periodicStart)。
  28. // 仅当三个窗口均未激活时生效。
  29. ActivateWindows(ctx context.Context, id int64, dailyStart, periodicStart time.Time) error
  30. // ResetUsageWindows 手动重置所选窗口的用量。日窗口锚点写入 dailyStart(当天 0 点,
  31. // 保持 0 点刷新节奏不漂移);周/月窗口锚点写入 periodicStart(重置时刻)。
  32. ResetUsageWindows(ctx context.Context, id int64, resetDaily, resetWeekly, resetMonthly bool, dailyStart, periodicStart time.Time) error
  33. ResetDailyUsage(ctx context.Context, id int64, expectedWindowStart *time.Time, newWindowStart time.Time) error
  34. ResetWeeklyUsage(ctx context.Context, id int64, expectedWindowStart *time.Time, newWindowStart time.Time) error
  35. ResetMonthlyUsage(ctx context.Context, id int64, expectedWindowStart *time.Time, newWindowStart time.Time) error
  36. IncrementUsage(ctx context.Context, id int64, costUSD float64) error
  37. BatchUpdateExpiredStatus(ctx context.Context) (int64, error)
  38. }