Alejandro Durante
36 lines
700 B
Go
36 lines
700 B
Go
package pond
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
)
|
|
|
|
// worker represents a worker goroutine
|
|
func worker(context context.Context, waitGroup *sync.WaitGroup, firstTask func(), tasks <-chan func(), taskExecutor func(func(), bool)) {
|
|
|
|
// If provided, execute the first task immediately, before listening to the tasks channel
|
|
if firstTask != nil {
|
|
taskExecutor(firstTask, true)
|
|
}
|
|
|
|
defer func() {
|
|
waitGroup.Done()
|
|
}()
|
|
|
|
for {
|
|
select {
|
|
case <-context.Done():
|
|
// Pool context was cancelled, exit
|
|
return
|
|
case task, ok := <-tasks:
|
|
if task == nil || !ok {
|
|
// We have received a signal to exit
|
|
return
|
|
}
|
|
|
|
// We have received a task, execute it
|
|
taskExecutor(task, false)
|
|
}
|
|
}
|
|
}
|