Goroutine management by using workerpool pattern

workerpool.go

package workerpool

type Work[T any] func(workerID int, result chan<- T) error

type Pool[T any] interface {
	Add(w Work[T])
	Shutdown()
	Result() <-chan T
}

type pool[T any] struct {
	maxWorkers int
	resultChan chan T
	workerChan chan Work[T]
}

func (p *pool[T]) Add(w Work[T]) {
	p.workerChan <- w
}

func (p *pool[T]) Result() <-chan T {
	return p.resultChan
}

func (p *pool[T]) Shutdown() {
	close(p.workerChan)
	close(p.resultChan)
}

func (p *pool[T]) run() {
	for i := 0; i < p.maxWorkers; i++ {
		go func(workerID int) {
			for w := range p.workerChan {
				w(workerID, p.resultChan)
			}
		}(i)
	}
}

func New[T any](maxWorkers int) Pool[T] {
	p := &pool[T]{
		maxWorkers: maxWorkers,
		workerChan: make(chan Work[T], maxWorkers),
		resultChan: make(chan T, maxWorkers),
	}

	p.run()

	return p
}