Как корректно дождаться завершения всех запущенных горутин?

Ожидание завершения горутин с WaitGroup

В Go горутины выполняются конкурентно, и основной поток не ждёт их завершения автоматически. Для синхронизации используется пакет sync, а именно тип WaitGroup. Он предоставляет счётчик, который увеличивается при запуске горутины и уменьшается при её завершении. Метод Wait блокирует выполнение, пока счётчик не станет равным нулю.

Пример использования

package main

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

func worker(id int, wg *sync.WaitGroup) {
    defer wg.Done() // Уменьшаем счётчик при выходе
    fmt.Printf("Worker %d starting\n", id)
    time.Sleep(time.Second)
    fmt.Printf("Worker %d done\n", id)
}

func main() {
    var wg sync.WaitGroup

    for i := 1; i <= 3; i++ {
        wg.Add(1) // Увеличиваем счётчик перед запуском
        go worker(i, &wg)
    }

    wg.Wait() // Ожидаем завершения всех горутин
    fmt.Println("All workers completed")
}

Важные моменты

• Вызов Add должен происходить до запуска горутины, чтобы избежать состояния гонки.
• Done вызывается внутри горутины, обычно через defer для гарантии выполнения.
• WaitGroup не копируется, передаётся по указателю.

Вывод

WaitGroup — простой и эффективный способ дождаться завершения группы горутин. Используйте его, когда нужно выполнить несколько параллельных задач и дождаться их окончания перед продолжением работы.