Function NewWorkerPool

pkg/workerpool/workerpool.go:238–277

NewWorkerPool creates a new WorkerPool with the provided configuration.

(cfg Config[T])

// NewWorkerPool creates a new WorkerPool with the provided configuration.
func NewWorkerPool[T any](cfg Config[T]) WorkerPool[T] {
	if cfg.WorkerIdleTimeout == 0 {
		cfg.WorkerIdleTimeout = defaultWorkerIdleTimeout
	}
	// We treat 0 as being default initialized, and use the defaults.
	if cfg.MinWorkers == 0 {
		cfg.MinWorkers = defaultMinWorkers
	}
	// We treat negative values as explicitly disabling the minimum number of workers.
	if cfg.MinWorkers < 0 {
		cfg.MinWorkers = 0
	}
	if cfg.MaxWorkers <= 0 {
		cfg.MaxWorkers = defaultMaxWorkers
	}
	// We treat 0 as being default initialized, and use the defaults.
	if cfg.QueueSize == 0 {
		cfg.QueueSize = defaultQueueSize
	}
	// We treat negative values as explicitly disabling the queue.
	if cfg.QueueSize < 0 {
		cfg.QueueSize = 0
	}
	if cfg.MinWorkers > cfg.MaxWorkers {
		cfg.MaxWorkers = cfg.MinWorkers
	}

	wp := &workerPool[T]{
		Config: cfg,

		mainQueue: make(chan *contextualItem[T], cfg.QueueSize),
		fastQueue: make(chan *contextualItem[T]),
	}

	for i := 0; i < wp.MinWorkers; i++ {
		wp.spawnWorker(nil)
	}

	return wp
}

func incrementIfSmallerThan(i *int32, max int32) bool {
	for v := atomic.LoadInt32(i); v < max; v = atomic.LoadInt32(i) {