package main

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

func worker(id int, wg *sync.WaitGroup) {
    defer wg.Done()

    // 模拟每个worker执行的时间不同
    time.Sleep(time.Duration(id) * time.Second * 2)
    fmt.Printf("Worker %d done\n", id)
}

func main() {
    //设置等待组
    var wg sync.WaitGroup
    //标记等待完成
    var wt = make(chan int)
    // 启动5个worker
    for i := 1; i <= 5; i++ {
        wg.Add(1)
        go worker(i, &wg)
    }

    // 等待所有worker完成或者超时
    timeout := time.After(10 * time.Second)
    go func() {
        wg.Wait()
        wt <- 0 // 所有worker都完成
    }()

    // 等待超时或者所有worker完成
    select {
    case <-timeout:
        fmt.Println("Timeout!")
    case <-wt:
        fmt.Println("All workers done!")
    }
}