## 為 sync.WaitGroup 中 Wait 函數支持 WaitTimeout 功能. ~~~go 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") } time.Sleep(time.Second * 10) } func WaitTimeout(wg *sync.WaitGroup, timeout time.Duration) bool { // 要求手寫代碼 // 要求sync.WaitGroup支持timeout功能 // 如果timeout到了超時時間返回true // 如果WaitGroup自然結束返回false } ~~~ **解析** 首先`sync.WaitGroup`對象的`Wait`函數本身是阻塞的，同時，超時用到的`time.Timer`對象也需要阻塞的讀。 同時阻塞的兩個對象肯定要每個啟動一個協程,每個協程去處理一個阻塞，難點在于怎么知道哪個阻塞先完成。 目前我用的方式是聲明一個沒有緩沖的`chan`，誰先完成誰優先向管道中寫入數據。 ~~~go 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") } time.Sleep(time.Second * 10) } func WaitTimeout(wg *sync.WaitGroup, timeout time.Duration) bool { // 要求手寫代碼 // 要求sync.WaitGroup支持timeout功能 // 如果timeout到了超時時間返回true // 如果WaitGroup自然結束返回false ch := make(chan bool, 1) go time.AfterFunc(timeout, func() { ch <- true }) go func() { wg.Wait() ch <- false }() return <- ch } ~~~