taskpool

type Config[TIn, TOut any] struct {
	MinWorkers          int
	MaxWorkers          int           // 0 = runtime.NumCPU() * 2
	IdleTimeout         time.Duration // Workers stop if idle longer than this
	TaskQueueSize       int
	TaskTimeout         time.Duration
	MaxTaskTimeout      time.Duration
	DelegateFactory     DelegateFactory[TIn, TOut]
	DelegateConfig      any
	HealthCheckInterval time.Duration
	Logger              log.Logger
}

type DefaultScheduler[TIn, TOut any] struct {
	// contains filtered or unexported fields
}

type DelegateFactory[TIn, TOut any] func() WorkerDelegate[TIn, TOut]

type Priority int

type Result[TOut any] struct {
	TaskId   string
	Data     TOut
	Error    error
	Duration time.Duration
}

type Scheduler[TIn, TOut any] interface {
	// Submit blocks until task completes or context is canceled.
	Submit(ctx context.Context, payload TIn, opts ...SubmitOption) (Result[TOut], error)

	// SubmitAsync returns immediately with a result channel.
	SubmitAsync(ctx context.Context, payload TIn, opts ...SubmitOption) (<-chan Result[TOut], error)

	Stats() SchedulerStats

	// Shutdown waits for running tasks to complete.
	Shutdown(ctx context.Context) error
}

type SchedulerStats struct {
	TotalSubmitted uint64
	TotalCompleted uint64
	TotalFailed    uint64
	ActiveWorkers  int
	IdleWorkers    int
	TotalWorkers   int
	QueuedTasks    int
}

type SubmitOption func(*submitOptions)

type Task[TIn, TOut any] struct {
	Id          string
	Context     context.Context
	Priority    Priority
	Payload     TIn
	Result      chan<- Result[TOut]
	Done        chan struct{} // Closed by worker after execution
	SubmittedAt time.Time
}

type Worker[TIn, TOut any] struct {
	// contains filtered or unexported fields
}

type WorkerDelegate[TIn, TOut any] interface {
	// Init is called once when worker starts.
	Init(ctx context.Context, config any) error

	// Execute runs for each task. Must respect context cancellation.
	Execute(ctx context.Context, payload TIn) (TOut, error)

	// Destroy is called once when worker stops.
	Destroy() error

	// HealthCheck is called periodically.
	HealthCheck() error
}

type WorkerPool[TIn, TOut any] struct {
	// contains filtered or unexported fields
}