Урок 116 · Задача 1

DrainWithTimeout: ждать с границей

Ждать вечно при зависшем воркере — тоже баг. Реализуйте drainWithTimeout(wg *sync.WaitGroup, timeout) bool: ждать wg через done-канал или таймаут.

Показать решение

func drainWithTimeout(wg *sync.WaitGroup, timeout time.Duration) bool {
	done := make(chan struct{})
	go func() {
		wg.Wait()
		close(done)
	}()
	select {
	case <-done:
		return true
	case <-time.After(timeout):
		return false
	}
}

package main

import (
	"sync"
	"time"
)

// drainWithTimeout ждёт wg не дольше timeout.
func drainWithTimeout(wg *sync.WaitGroup, timeout time.Duration) bool {
	_ = time.After
	// ваш код
	return false
}