为 sync.WaitGroup 中 Wait 函数⽀持 WaitTimeout 功能

package main

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

func main() {
	wg := sync.WaitGroup{}
	c := make(chan struct{})
	for i := 0; i < 10; i++ {
		wg.Add(1)
		go func(num int, close <-chan struct{}) {
			defer wg.Done()
			<-close
			fmt.Println(num)
		}(i, c)
	}
	if WaitTimeout(&wg, time.Second*5) {
		close(c)
		fmt.Println("timeout exit")
	}
	fmt.Println("WaitTimeout执行完了")
	time.Sleep(time.Second * 10)
}

func WaitTimeout(wg *sync.WaitGroup, timeout time.Duration) bool {
	ch := make(chan bool)
	go time.AfterFunc(timeout, func() {
		fmt.Println(timeout,"过去了")
		ch <- true
	})
	go func() {
		wg.Wait()
		ch <- false
	}()

	return <-ch
}