mirror of
https://github.com/muety/wakapi.git
synced 2023-08-10 21:12:56 +03:00
24 lines
466 B
Go
24 lines
466 B
Go
|
package utils
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
// WaitTimeout waits for the waitgroup for the specified max timeout.
|
||
|
|
// Returns true if waiting timed out.
|
||
|
|
// See // https://stackoverflow.com/a/32843750/3112139.
|
||
|
|
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
|
||
|
|
}
|
||
|
|
}
|