ccede75412
Use the full power of closure by accepting the parameters to anonymous function being run by goroutine.
53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
// 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)
|
|
|
|
// 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(i int) {
|
|
defer wg.Done()
|
|
worker(i)
|
|
// use the incremental value of i in each different goroutine.
|
|
}(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. For more
|
|
// advanced use cases, consider using the
|
|
// [errgroup package](https://pkg.go.dev/golang.org/x/sync/errgroup).
|
|
}
|