Как сказать основной горутине дождаться завершения всех рабочих горутин?

Канонический способ дождаться завершения всех рабочих горутин в Go — использовать sync.WaitGroup. Он обеспечивает простой и надежный шаблон: увеличьте счетчик перед запуском задания, уменьшите его после его завершения и вызовите Wait() в основной горутине.

Основная схема:

1. Создайте var wg sync.WaitGroup.

2. Перед каждым вызовом горутины wg.Add(1).

3. Внутри горутины выполните defer wg.Done().

4. В основной горутине вызовите wg.Wait().

Почему это работает:

1. WaitGroup подсчитывает количество незавершенных задач.

2. Wait() блокирует выполнение до тех пор, пока счетчик не достигнет нуля.

3. Это гарантирует, что main не завершится раньше рабочих горутин.

Типичные ошибки, которых следует избегать:

1. Вызов Add(1) после запуска горутины (риск гонки и неправильное завершение).

2. Забудьте Done() в ветке ошибки или ранней return.

3. Повторное использование одного и того же WaitGroup на разных этапах без четкого synchronization.

Когда лучше errgroup:

Если помимо ожидания вам также необходимо:

1. собираем первую ошибку,

2. отменить другие задачи с помощью context,

тогда практичнее использовать errgroup.Group.

Заключение:

Для задачи «дождаться завершения всех горутин» стандартным инструментом является sync.WaitGroup: простой контракт, предсказуемое поведение и производственная надежность.

Пример:

Итог

Канонический способ дождаться завершения всех рабочих горутин в Go — использовать sync.WaitGroup.