mirror of
https://gitee.com/wanwujie/sub2api
synced 2026-04-02 22:42:14 +08:00
0c29468f90
- 为定时测试计划增加 auto_recover 配置，补齐前后端类型、接口、仓储与数据库迁移 - 在定时测试成功后自动恢复账号 error、rate-limit 等可恢复运行时状态 - 新增 /admin/accounts/:id/recover-state 接口，合并原有重置状态与清限流操作 - 更新账号管理菜单与定时测试面板，补充自动恢复开关、说明提示和状态展示 - 补充账号恢复、限流清理与仓储同步相关测试
53 lines
2.2 KiB
Go
53 lines
2.2 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
)
|
|
|
|
// ScheduledTestPlan represents a scheduled test plan domain model.
|
|
type ScheduledTestPlan struct {
|
|
ID int64 `json:"id"`
|
|
AccountID int64 `json:"account_id"`
|
|
ModelID string `json:"model_id"`
|
|
CronExpression string `json:"cron_expression"`
|
|
Enabled bool `json:"enabled"`
|
|
MaxResults int `json:"max_results"`
|
|
AutoRecover bool `json:"auto_recover"`
|
|
LastRunAt *time.Time `json:"last_run_at"`
|
|
NextRunAt *time.Time `json:"next_run_at"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
}
|
|
|
|
// ScheduledTestResult represents a single test execution result.
|
|
type ScheduledTestResult struct {
|
|
ID int64 `json:"id"`
|
|
PlanID int64 `json:"plan_id"`
|
|
Status string `json:"status"`
|
|
ResponseText string `json:"response_text"`
|
|
ErrorMessage string `json:"error_message"`
|
|
LatencyMs int64 `json:"latency_ms"`
|
|
StartedAt time.Time `json:"started_at"`
|
|
FinishedAt time.Time `json:"finished_at"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
}
|
|
|
|
// ScheduledTestPlanRepository defines the data access interface for test plans.
|
|
type ScheduledTestPlanRepository interface {
|
|
Create(ctx context.Context, plan *ScheduledTestPlan) (*ScheduledTestPlan, error)
|
|
GetByID(ctx context.Context, id int64) (*ScheduledTestPlan, error)
|
|
ListByAccountID(ctx context.Context, accountID int64) ([]*ScheduledTestPlan, error)
|
|
ListDue(ctx context.Context, now time.Time) ([]*ScheduledTestPlan, error)
|
|
Update(ctx context.Context, plan *ScheduledTestPlan) (*ScheduledTestPlan, error)
|
|
Delete(ctx context.Context, id int64) error
|
|
UpdateAfterRun(ctx context.Context, id int64, lastRunAt time.Time, nextRunAt time.Time) error
|
|
}
|
|
|
|
// ScheduledTestResultRepository defines the data access interface for test results.
|
|
type ScheduledTestResultRepository interface {
|
|
Create(ctx context.Context, result *ScheduledTestResult) (*ScheduledTestResult, error)
|
|
ListByPlanID(ctx context.Context, planID int64, limit int) ([]*ScheduledTestResult, error)
|
|
PruneOldResults(ctx context.Context, planID int64, keepCount int) error
|
|
}
|