WaitGroups

To wait for multiple goroutines to finish, we can use a wait group.

package main

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

// This is the function we’ll run in every goroutine.
func worker(id int) {
    fmt.Printf("Worker %d starting\n", id)

    // Sleep to simulate an expensive task.
    time.Sleep(time.Second)
    fmt.Printf("Worker %d done\n", id)
}

func main() {

    // This WaitGroup is used to wait for all the goroutines launched here to finish.
    // Note: if a WaitGroup is explicitly passed into functions
    // it should be done by pointer.
    var wg sync.WaitGroup

    // Launch several goroutines and increment the WaitGroup counter for each.
    for i := 1; i <= 5; i++ {
        wg.Add(1)

        // Avoid re-use of the same i value in each goroutine closure.
        i := i

        // Wrap the worker call in a closure that makes sure 
        // to tell the WaitGroup that this worker is done. 
        // This way the worker itself does not have to be aware 
        // of the concurrency primitives involved in its execution.
        go func() {
            defer wg.Done()
            worker(i)
        }()
    }

    // Block until the WaitGroup counter goes back to 0; 
    // all the workers notified they’re done.
    wg.Wait()

}
// Note that this approach has no straightforward way to propagate errors from workers. 
$ go run waitgroups.go
Worker 5 starting
Worker 3 starting
Worker 4 starting
Worker 1 starting
Worker 2 starting
Worker 4 done
Worker 1 done
Worker 2 done
Worker 5 done
Worker 3 done

The order of workers starting up and finishing is likely to be different for each invocation.

Source | License