ed557

思路是把 wg.wait() 放到一個(gè)協(xié)程里，通過chan向外發(fā)送完成信號(hào)。 外層通過一個(gè)select超時(shí)結(jié)構(gòu)來控制最大超時(shí)時(shí)間。

func waitTimeout(wg *sync.WaitGroup, timeout time.Duration) bool {
    c := make(chan struct{})
    go func() {
        defer close(c)
        wg.Wait()
    }()
    select {
    case <-c:
        return false // completed normally
    case <-time.After(timeout):
        return true // timed out
    }
}