work dispatcher

workdisp.go

package main

import (
	"fmt"
	"sync"
	"time"
)

func worker(id int, jobs <-chan int, results chan<- int) {
	for j := range jobs {
		fmt.Println("worker", id, "processing job", j)
		time.Sleep(time.Duration(id) * time.Second)
		results <- j
	}
	fmt.Println("worker", id, "done")
}

func resultCollector(results <-chan int, wg *sync.WaitGroup) {
	for r := range results {
		fmt.Println("got result", r)
		wg.Done()
	}
}

func main() {
	wg := new(sync.WaitGroup)
	jobs := make(chan int)
	results := make(chan int)

	for w := 1; w <= 3; w++ {
		go worker(w, jobs, results)
	}

	go resultCollector(results, wg)

	for j := 1; j <= 9; j++ {
		wg.Add(1)
		jobs <- j
	}
	close(jobs)

	wg.Wait()
}