golang simple workers pool

workers_pool.go

package utils

type WorkerPool[T any] struct {
	workFn func(jobID int, result chan<- T)
	result chan T
}

func NewWorkerPool[T any]() WorkerPool[T] {
	pool := WorkerPool[T]{}
	pool.result = make(chan T)

	return pool
}

func (pool WorkerPool[T]) Run(goroutinWorkers int, jobLimit int, workFn func(jobID int, result chan<- T)) <-chan T {
	jobs := make(chan int)
	pool.workFn = workFn
	for i := 0; i < goroutinWorkers; i++ {
		go pool.work(jobs)
	}

	go func() {
		for i := 0; i < jobLimit; i++ {
			jobs <- i
		}
		close(jobs)
	}()

	return pool.result
}

func (pool WorkerPool[T]) work(jobs <-chan int) {
	for jobID := range jobs {
		pool.workFn(jobID, pool.result)
	}
}