2026-01-18 10:52:18 +08:00
|
|
|
|
package service
|
|
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
|
"context"
|
|
|
|
|
|
"time"
|
|
|
|
|
|
|
|
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/pkg/pagination"
|
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
const (
|
|
|
|
|
|
UsageCleanupStatusPending = "pending"
|
|
|
|
|
|
UsageCleanupStatusRunning = "running"
|
|
|
|
|
|
UsageCleanupStatusSucceeded = "succeeded"
|
|
|
|
|
|
UsageCleanupStatusFailed = "failed"
|
|
|
|
|
|
UsageCleanupStatusCanceled = "canceled"
|
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
// UsageCleanupFilters 定义清理任务过滤条件
|
|
|
|
|
|
// 时间范围为必填，其他字段可选
|
|
|
|
|
|
// JSON 序列化用于存储任务参数
|
|
|
|
|
|
//
|
|
|
|
|
|
// start_time/end_time 使用 RFC3339 时间格式
|
|
|
|
|
|
// 以 UTC 或用户时区解析后的时间为准
|
|
|
|
|
|
//
|
|
|
|
|
|
// 说明：
|
|
|
|
|
|
// - nil 表示未设置该过滤条件
|
|
|
|
|
|
// - 过滤条件均为精确匹配
|
|
|
|
|
|
type UsageCleanupFilters struct {
|
|
|
|
|
|
StartTime time.Time `json:"start_time"`
|
|
|
|
|
|
EndTime time.Time `json:"end_time"`
|
|
|
|
|
|
UserID *int64 `json:"user_id,omitempty"`
|
|
|
|
|
|
APIKeyID *int64 `json:"api_key_id,omitempty"`
|
|
|
|
|
|
AccountID *int64 `json:"account_id,omitempty"`
|
|
|
|
|
|
GroupID *int64 `json:"group_id,omitempty"`
|
|
|
|
|
|
Model *string `json:"model,omitempty"`
|
2026-02-28 15:01:20 +08:00
|
|
|
|
RequestType *int16 `json:"request_type,omitempty"`
|
2026-01-18 10:52:18 +08:00
|
|
|
|
Stream *bool `json:"stream,omitempty"`
|
|
|
|
|
|
BillingType *int8 `json:"billing_type,omitempty"`
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// UsageCleanupTask 表示使用记录清理任务
|
|
|
|
|
|
// 状态包含 pending/running/succeeded/failed/canceled
|
|
|
|
|
|
type UsageCleanupTask struct {
|
|
|
|
|
|
ID int64
|
|
|
|
|
|
Status string
|
|
|
|
|
|
Filters UsageCleanupFilters
|
|
|
|
|
|
CreatedBy int64
|
|
|
|
|
|
DeletedRows int64
|
|
|
|
|
|
ErrorMsg *string
|
|
|
|
|
|
CanceledBy *int64
|
|
|
|
|
|
CanceledAt *time.Time
|
|
|
|
|
|
StartedAt *time.Time
|
|
|
|
|
|
FinishedAt *time.Time
|
|
|
|
|
|
CreatedAt time.Time
|
|
|
|
|
|
UpdatedAt time.Time
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// UsageCleanupRepository 定义清理任务持久层接口
|
|
|
|
|
|
type UsageCleanupRepository interface {
|
|
|
|
|
|
CreateTask(ctx context.Context, task *UsageCleanupTask) error
|
|
|
|
|
|
ListTasks(ctx context.Context, params pagination.PaginationParams) ([]UsageCleanupTask, *pagination.PaginationResult, error)
|
|
|
|
|
|
// ClaimNextPendingTask 抢占下一条可执行任务：
|
|
|
|
|
|
// - 优先 pending
|
|
|
|
|
|
// - 若 running 超过 staleRunningAfterSeconds（可能由于进程退出/崩溃/超时），允许重新抢占继续执行
|
|
|
|
|
|
ClaimNextPendingTask(ctx context.Context, staleRunningAfterSeconds int64) (*UsageCleanupTask, error)
|
|
|
|
|
|
// GetTaskStatus 查询任务状态；若不存在返回 sql.ErrNoRows
|
|
|
|
|
|
GetTaskStatus(ctx context.Context, taskID int64) (string, error)
|
|
|
|
|
|
// UpdateTaskProgress 更新任务进度（deleted_rows）用于断点续跑/展示
|
|
|
|
|
|
UpdateTaskProgress(ctx context.Context, taskID int64, deletedRows int64) error
|
|
|
|
|
|
// CancelTask 将任务标记为 canceled（仅允许 pending/running）
|
|
|
|
|
|
CancelTask(ctx context.Context, taskID int64, canceledBy int64) (bool, error)
|
|
|
|
|
|
MarkTaskSucceeded(ctx context.Context, taskID int64, deletedRows int64) error
|
|
|
|
|
|
MarkTaskFailed(ctx context.Context, taskID int64, deletedRows int64, errorMsg string) error
|
|
|
|
|
|
DeleteUsageLogsBatch(ctx context.Context, filters UsageCleanupFilters, limit int) (int64, error)
|
|
|
|
|
|
}
|
