How do I use WaitGroup in Go?

Can you explain how to use WaitGroup in Go?

Sure! A WaitGroup is used to wait for a collection of goroutines to finish executing. You can add to the WaitGroup, and then call Wait() to block until all goroutines have completed.

How do I add goroutines to the WaitGroup?

You can use the Add(n int) method to add n goroutines to the WaitGroup. For example, wg.Add(1) adds one goroutine.

What about signaling when a goroutine is done?

You should call Done() on the WaitGroup when a goroutine finishes its work. This decrements the counter.

Can you show me a code example?

Absolutely! Here's a simple example:

package main

import (
    "fmt"
    "sync"
)

func main() {
    var wg sync.WaitGroup

    for i := 0; i < 5; i++ {
        wg.Add(1)
        go func(i int) {
            defer wg.Done()
            fmt.Println("Goroutine", i)
        }(i)
    }

    wg.Wait()
    fmt.Println("All goroutines finished.")
}