Function NewPool

internal/workshare/workshare_pool.go:37–69

NewPool creates a worker pool that launches a given number of goroutines that can invoke shared work.

(numWorkers int)

// NewPool creates a worker pool that launches a given number of goroutines that can invoke shared work.
func NewPool[T any](numWorkers int) *Pool[T] {
	if numWorkers < 0 {
		numWorkers = 0
	}

	w := &Pool[T]{
		// channel must be unbuffered so that it has exactly as many slots as there are goroutines capable of reading from it
		// this way by pushing to the channel we can be sure that a pre-spun goroutine will pick it up soon.
		work:      make(chan workItem[T]),
		closed:    make(chan struct{}),
		semaphore: make(chan struct{}, numWorkers),
	}

	for range numWorkers {
		w.wg.Go(func() {
			for {
				select {
				case it := <-w.work:
					w.activeWorkers.Add(1)
					it.process(w, it.request)
					w.activeWorkers.Add(-1)
					<-w.semaphore
					it.wg.Done()

				case <-w.closed:
					return
				}
			}
		})
	}

	return w
}

// Close closes the worker pool.
func (w *Pool[T]) Close() {