142 lines
4.6 KiB
Go
142 lines
4.6 KiB
Go
package usecase
|
|
|
|
import (
|
|
"context"
|
|
|
|
"haixun-backend/internal/model/job/domain/entity"
|
|
"haixun-backend/internal/model/job/domain/repository"
|
|
)
|
|
|
|
type CreateRunRequest struct {
|
|
TemplateType string
|
|
Scope string
|
|
ScopeID string
|
|
TenantID string
|
|
OwnerUID string
|
|
Payload map[string]any
|
|
ScheduledAt *int64
|
|
}
|
|
|
|
type CancelRunRequest struct {
|
|
JobID string
|
|
Reason string
|
|
}
|
|
|
|
type ClaimNextRequest struct {
|
|
WorkerType string
|
|
WorkerID string
|
|
}
|
|
|
|
type UpdateProgressRequest struct {
|
|
JobID string
|
|
WorkerID string
|
|
Phase string
|
|
Summary string
|
|
Percentage int
|
|
Steps []entity.StepProgress
|
|
}
|
|
|
|
type CompleteRunRequest struct {
|
|
JobID string
|
|
WorkerID string
|
|
Result map[string]any
|
|
}
|
|
|
|
type FailRunRequest struct {
|
|
JobID string
|
|
WorkerID string
|
|
Error string
|
|
Phase string
|
|
}
|
|
|
|
type AcknowledgeCancelRequest struct {
|
|
JobID string
|
|
WorkerID string
|
|
}
|
|
|
|
type UpsertTemplateRequest struct {
|
|
Type string
|
|
Version int
|
|
Name string
|
|
Description string
|
|
Enabled bool
|
|
Repeatable bool
|
|
ConcurrencyPolicy string
|
|
DedupeKeys []string
|
|
TimeoutSeconds int
|
|
CancelPolicy entity.CancelPolicy
|
|
RetryPolicy entity.RetryPolicy
|
|
Steps []entity.TemplateStep
|
|
}
|
|
|
|
type CreateScheduleRequest struct {
|
|
TemplateType string
|
|
Scope string
|
|
ScopeID string
|
|
Cron string
|
|
Timezone string
|
|
PayloadTemplate map[string]any
|
|
Enabled bool
|
|
}
|
|
|
|
type UpdateScheduleRequest struct {
|
|
ID string
|
|
Cron string
|
|
Timezone string
|
|
PayloadTemplate map[string]any
|
|
Enabled *bool
|
|
}
|
|
|
|
type UseCase interface {
|
|
ListTemplates(ctx context.Context) ([]*entity.Template, error)
|
|
GetTemplate(ctx context.Context, templateType string) (*entity.Template, error)
|
|
UpsertTemplate(ctx context.Context, req UpsertTemplateRequest) (*entity.Template, error)
|
|
EnsureDemoTemplate(ctx context.Context) error
|
|
EnsureStyle8DTemplate(ctx context.Context) error
|
|
EnsureExpandGraphTemplate(ctx context.Context) error
|
|
EnsurePlacementScanTemplate(ctx context.Context) error
|
|
EnsureScanViralTemplate(ctx context.Context) error
|
|
EnsureAnalyzeCopyMissionTemplate(ctx context.Context) error
|
|
EnsureExpandCopyMissionGraphTemplate(ctx context.Context) error
|
|
EnsureGenerateCopyMatrixTemplate(ctx context.Context) error
|
|
EnsureGenerateCopyDraftTemplate(ctx context.Context) error
|
|
EnsureGenerateOutreachDraftTemplate(ctx context.Context) error
|
|
EnsureRefreshThreadsTokenTemplate(ctx context.Context) error
|
|
EnsurePublishAnalyticsTemplate(ctx context.Context) error
|
|
EnsureRefillPublishInventoryTemplate(ctx context.Context) error
|
|
|
|
CreateRun(ctx context.Context, req CreateRunRequest) (*entity.Run, error)
|
|
GetRun(ctx context.Context, jobID string) (*entity.Run, error)
|
|
ListRuns(ctx context.Context, filter repository.RunListFilter, page, pageSize int64) ([]*entity.Run, int64, int64, int64, int64, error)
|
|
RequestCancel(ctx context.Context, req CancelRunRequest) (*entity.Run, error)
|
|
RetryRun(ctx context.Context, jobID string) (*entity.Run, error)
|
|
ListJobEvents(ctx context.Context, jobID string, limit int64) ([]*entity.Event, error)
|
|
|
|
ClaimNext(ctx context.Context, req ClaimNextRequest) (*entity.Run, error)
|
|
RefreshRunLock(ctx context.Context, jobID, workerID string, ttlSeconds int) error
|
|
IsCancelRequested(ctx context.Context, jobID string) (bool, error)
|
|
AcknowledgeCancel(ctx context.Context, req AcknowledgeCancelRequest) (*entity.Run, error)
|
|
UpdateProgress(ctx context.Context, req UpdateProgressRequest) (*entity.Run, error)
|
|
CompleteRun(ctx context.Context, req CompleteRunRequest) (*entity.Run, error)
|
|
FailRun(ctx context.Context, req FailRunRequest) (*entity.Run, error)
|
|
|
|
ListSchedules(ctx context.Context, scope, scopeID string, page, pageSize int64) ([]*entity.Schedule, int64, int64, int64, int64, error)
|
|
GetSchedule(ctx context.Context, scheduleID string) (*entity.Schedule, error)
|
|
CreateSchedule(ctx context.Context, req CreateScheduleRequest) (*entity.Schedule, error)
|
|
UpdateSchedule(ctx context.Context, req UpdateScheduleRequest) (*entity.Schedule, error)
|
|
EnableSchedule(ctx context.Context, scheduleID string) (*entity.Schedule, error)
|
|
DisableSchedule(ctx context.Context, scheduleID string) (*entity.Schedule, error)
|
|
DeleteSchedule(ctx context.Context, scheduleID string) error
|
|
|
|
RunSchedulerTick(ctx context.Context, holder string) (int, error)
|
|
RunMaintenance(ctx context.Context) (MaintenanceResult, error)
|
|
}
|
|
|
|
type MaintenanceResult struct {
|
|
EnqueuedPending int
|
|
ReapedCancelGrace int
|
|
ReapedExpiredLocks int
|
|
RepairedStuckClaims int
|
|
ReapedOrphanedLocks int
|
|
}
|