go语言实现多协程调用并且在超时或者全部协程执行完成后退出
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!")
}
}