Dynamic pool that allows concurrency for a set of tasks to be throttled.

dynamic_pool.go

package main

import "sync"

func main() {
	jobChan := make(chan (chan int))
	resChan := make(chan int)
	squareGroup := new(sync.WaitGroup)
	
	// The workers
	for i := 0; i < 5; i++ {
		squareGroup.Add(1)
		go func() {
			for taskChan := range jobChan {
				for n := range taskChan {
					resChan <- n * n
				}
			}
			squareGroup.Done()
		}()
	}
	
	// The "submitter"
	go func() {
		taskChan := make(chan int)
		for i := 0; i < 1; i++ {
			jobChan <- taskChan
		}
		for i := 0; i < 10; i++ {
			taskChan <- i
		}
		close(taskChan)
		close(jobChan)
		squareGroup.Wait()
		close(resChan)
	}()
	
	// Collect the results back in the main goroutine
	for s := range resChan {
		println(s)
	}
}