dynworker

type Config struct {

	// Custom queue to use.
	InputQueue TaskQueue

	// Prometheus registerer.
	Prometheus prometheus.Registerer

	// Function to use as the worker.
	WorkerFunc TaskFn

	// Function to use to determine scaling.
	ScalerFunc ScalerFn

	// Worker pool name for logger and metrics.
	Name string

	// Minimum number of workers.
	MinWorkers int64

	// Maximum number of workers.
	MaxWorkers int64

	// Idle timeout duration.
	IdleTimeout time.Duration

	// Drain target duration.
	DrainTarget time.Duration
}

type ScalerFn func() int

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

type TaskFn func(*Task) error

type TaskQueue interface {
	Put(context.Context, *Task) error
	Get(context.Context) (*Task, error)
	Len() int
}

type UserData any

type WorkerPool interface {
	// Start the worker pool.
	Start()

	// Stop the worker pool.
	Stop()

	// Submit a task to the worker pool.
	Submit(UserData) error

	// Return the number of current workers in the pool.
	WorkerCount() int64

	// Return the minimum number of workers in the pool.
	MinWorkers() int64

	// Return the maximum number of workers in the pool.
	MaxWorkers() int64

	// Set the minimum number of workers to the given value.
	SetMinWorkers(int64)

	// Set the maximum number of workers to the given value.
	SetMaxWorkers(int64)

	// Set the task callback function.
	SetTaskFunction(TaskFn)

	// Set the task scaler function.
	SetScalerFunction(ScalerFn)

	// Return the name of the pool.
	Name() string
}