provider

type CancelBatch struct {
	Changed  int
	Terminal []CancelResult
}

type CancelResult struct {
	Changed      bool
	Terminal     bool
	Type         string
	AttemptCount int64
}

type ClaimOptions struct {
	Types         []string
	Limit         int
	LeaseDuration time.Duration
	Resource      *ClaimResource
}

type ClaimResource struct {
	Name        string
	Concurrency int64
	Costs       map[string]int64
}

type CountQuery struct {
	Types []string
}

type EnqueueRequest struct {
	ID           string
	Type         string
	Payload      []byte
	MaxAttempts  int
	Delay        time.Duration
	DedupeKey    string
	ExclusiveKey string
}

type EnqueueResult struct {
	ID       string
	State    State
	Inserted bool
}

type Executor interface {
	ExecContext(ctx context.Context, query string, arguments ...any) (sql.Result, error)
	QueryRowContext(ctx context.Context, query string, arguments ...any) *sql.Row
}

type FailedCursor struct {
	FinishedAt time.Time
	ID         string
}

type FailedPage struct {
	Jobs []Summary
	More bool
}

type FailedQuery struct {
	Types  []string
	Limit  int
	Before *FailedCursor
}

type JobCursor struct {
	EnqueuedAt time.Time
	ID         string
}

type JobPage struct {
	Jobs []Summary
	More bool
}

type JobQuery struct {
	Types  []string
	States []State
	Limit  int
	Before *JobCursor
}

type Record struct {
	ID              string
	Type            string
	Payload         []byte
	State           State
	AttemptCount    int64
	MaxAttempts     int64
	AvailableAt     time.Time
	LeaseToken      string
	LeaseUntil      *time.Time
	DedupeKey       string
	ExclusiveKey    string
	CancelRequested bool
	LastCode        string
	EnqueuedAt      time.Time
	FinishedAt      *time.Time
	UpdatedAt       time.Time
}

type Renewal struct {
	Renewed         bool
	CancelRequested bool
}

type RetentionPolicy struct {
	OlderThan time.Time
	MaxRows   int
	States    []State
}

type State string

type StateCounts struct {
	Pending   int64
	Leased    int64
	Succeeded int64
	Failed    int64
	Canceled  int64
}

type Store interface {
	EnsureSchema(ctx context.Context) error
	Enqueue(ctx context.Context, executor Executor, request EnqueueRequest) (EnqueueResult, error)
	Claim(ctx context.Context, options ClaimOptions) ([]Record, error)
	Renew(ctx context.Context, id, token string, duration time.Duration) (Renewal, error)
	Succeed(ctx context.Context, id, token, code string) (bool, error)
	Fail(ctx context.Context, id, token, code string) (bool, error)
	RetryAt(ctx context.Context, id, token string, delay time.Duration, code string, uncounted bool) (bool, error)
	MarkCanceled(ctx context.Context, id, token, code string) (bool, error)
	Release(ctx context.Context, id, token string) (bool, error)
	Inspect(ctx context.Context, id string) (Record, error)
	List(ctx context.Context, query JobQuery) (JobPage, error)
	ListFailed(ctx context.Context, query FailedQuery) (FailedPage, error)
	CountByState(ctx context.Context, query CountQuery) (StateCounts, error)
	Cancel(ctx context.Context, id string) (CancelResult, error)
	CancelMany(ctx context.Context, ids []string) (CancelBatch, error)
	Requeue(ctx context.Context, id string) (bool, error)
	RequeueFailed(ctx context.Context, ids []string) (int, error)
	RunRetention(ctx context.Context, policy RetentionPolicy) (int, error)
}

type Summary struct {
	ID              string
	Type            string
	State           State
	AttemptCount    int64
	MaxAttempts     int64
	AvailableAt     time.Time
	CancelRequested bool
	LastCode        string
	EnqueuedAt      time.Time
	FinishedAt      *time.Time
}