02.go from https://github.com/brnstz/routine

02.go

package main

import (
	"flag"
	"fmt"
	"log"
	"sync"

	"github.com/brnstz/routine/wikimg"
)

var (
	// Print a blank line with the given 256 ANSI color
	fmtSpec = "\x1b[30;48;5;%dm%-80s\x1b[0m\n"
)

func main() {
	var max, workers, buffer int

	flag.IntVar(&max, "max", 100, "maximum number of images to retrieve")
	flag.IntVar(&workers, "workers", 25, "number of background workers")
	flag.IntVar(&buffer, "buffer", 10000, "size of buffered channels")
	flag.Parse()

	// Create a new image puller with our max
	p := wikimg.NewPuller(max)

	// Create a buffered channel for communicating between image
	// puller loop and workers
	imgURLs := make(chan string, buffer)

	// Use wg to wait for goroutines to complete
	wg := sync.WaitGroup{}

	for i := 0; i < workers; i++ {
		// Tell wg that we're adding something to wait for. Be sure to do this
		// outside the go call.
		wg.Add(1)

		go func() {
			for imgURL := range imgURLs {

				// Get the first color in this image
				color, _, err := p.FirstColor(imgURL)
				if err != nil {
					log.Println(err)
					continue
				}

				// Print color to the terminal
				fmt.Printf(fmtSpec, color, "")
			}

			// Once there is nothing else in imgURLs, this goroutine
			// is finished and can tell that to wg.
			wg.Done()
		}()
	}

	// Loop to retrieve more images
	for {
		imgURL, err := p.Next()

		if err == wikimg.EndOfResults {
			// Break from loop when end of results is reached
			break

		} else if err != nil {
			// Log error and continue getting URLs
			log.Println(err)
			continue
		}

		// Send this imgURL to the channel
		imgURLs <- imgURL
	}

	// There are no more imgURLs to send, close the channel. This
	// will cause the range in the goroutines to complete, once any
	// buffered entries are exhausted.
	close(imgURLs)

	// Wait until all goroutines call wg.Done()
	wg.Wait()
}