What are some examples of using WaitGroup in Go?

Can you explain how to use WaitGroup in Go?

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

Can you provide a code example?

Absolutely! Here's a simple example:

package main

import (
    "fmt"
    "sync"
)

func main() {
    var wg sync.WaitGroup

    for i := 1; i <= 3; i++ {
        wg.Add(1)
        go func(i int) {
            defer wg.Done()
            fmt.Printf("Goroutine %d is running\n", i)
        }(i)
    }

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

What does the defer wg.Done() line do?

The defer wg.Done() line ensures that the Done() method is called when the goroutine completes, which decrements the WaitGroup counter.

Are there any common mistakes to avoid when using WaitGroup?

Yes, a common mistake is forgetting to call Done() for each Add() call, which can lead to deadlocks.