scheduling

type AssignedItemWithTask struct {
	AssignedItem *v1.AssignedItem
	Task         *v1.V1TaskWithPayload
}

type ConcurrencyResults struct {
	*v1.RunConcurrencyResult

	TenantId uuid.UUID
}

type Extensions struct {
	// contains filtered or unexported fields
}

type Pool interface {
	GetResultsCh() chan *QueueResults
	GetConcurrencyResultsCh() chan *ConcurrencyResults

	// AddExtension registers a scheduler extension (metrics, autoscaling, ...)
	// against the pool. Extensions are written once against the shared contract
	// in this package and work with either implementation.
	AddExtension(ext SchedulerExtension)

	SetTenants(tenants []*sqlcv1.Tenant)
	Replenish(ctx context.Context, tenantId uuid.UUID)

	NotifyQueues(ctx context.Context, tenantId uuid.UUID, queueNames []string)
	NotifyConcurrency(ctx context.Context, tenantId uuid.UUID, strategyIds []int64)
	NotifyNewWorker(ctx context.Context, tenantId uuid.UUID, workerId uuid.UUID)
	NotifyNewQueue(ctx context.Context, tenantId uuid.UUID, queueName string)
	NotifyNewConcurrencyStrategy(ctx context.Context, tenantId uuid.UUID, strategyId int64)

	RunOptimisticScheduling(ctx context.Context, tenantId uuid.UUID, opts []*v1.WorkflowNameTriggerOpts, localWorkerIds map[uuid.UUID]struct{}) (map[uuid.UUID][]*AssignedItemWithTask, []*v1.V1TaskWithPayload, []*v1.DAGWithData, []v1.IdempotencyCollision, error)
	RunOptimisticSchedulingFromEvents(ctx context.Context, tenantId uuid.UUID, opts []v1.EventTriggerOpts, localWorkerIds map[uuid.UUID]struct{}) (map[uuid.UUID][]*AssignedItemWithTask, *v1.TriggerFromEventsResult, error)
}

type PostAssignInput struct {
	HasUnassignedStepRuns bool
}

type QueueResults struct {
	TenantId uuid.UUID
	Assigned []*v1.AssignedItem
	Buffered []*v1.AssignedItem

	Unassigned         []*sqlcv1.V1QueueItem
	SchedulingTimedOut []*sqlcv1.V1QueueItem
	RateLimited        []*v1.RateLimitResult
}

type SchedulerExtension interface {
	SetTenants(tenants []*sqlcv1.Tenant)
	ReportSnapshot(ctx context.Context, tenantId uuid.UUID, input *SnapshotInput)
	PostAssign(tenantId uuid.UUID, input *PostAssignInput)
	CleanupTenant(tenantId uuid.UUID) error
	Cleanup() error
}

type SlotCp struct {
	WorkerId uuid.UUID
	Used     bool
}

type SlotUtilization struct {
	UtilizedSlots    int
	NonUtilizedSlots int
}

type SnapshotInput struct {
	Workers map[uuid.UUID]*WorkerCp

	// WorkerSlotUtilization is the per-worker slot utilization summed across slot types.
	WorkerSlotUtilization map[uuid.UUID]*SlotUtilization

	// WorkerSlotUtilizationByType breaks WorkerSlotUtilization down by slot type.
	WorkerSlotUtilizationByType map[uuid.UUID]map[string]*SlotUtilization
}

type WorkerCp struct {
	WorkerId uuid.UUID
	MaxRuns  int
	Labels   []*sqlcv1.ListManyWorkerLabelsRow
	Name     string
}