mirror of
https://github.com/HDT3213/godis.git
synced 2025-10-05 16:57:06 +08:00
39 lines
498 B
Go
39 lines
498 B
Go
package wait
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
type Wait struct {
|
|
wg sync.WaitGroup
|
|
}
|
|
|
|
func (w *Wait) Add(delta int) {
|
|
w.wg.Add(delta)
|
|
}
|
|
|
|
func (w *Wait) Done() {
|
|
w.wg.Done()
|
|
}
|
|
|
|
func (w *Wait) Wait() {
|
|
w.wg.Wait()
|
|
}
|
|
|
|
// return isTimeout
|
|
func (w *Wait) WaitWithTimeout(timeout time.Duration) bool {
|
|
c := make(chan bool)
|
|
go func() {
|
|
defer close(c)
|
|
w.wg.Wait()
|
|
c <- true
|
|
}()
|
|
select {
|
|
case <-c:
|
|
return false // completed normally
|
|
case <-time.After(timeout):
|
|
return true // timed out
|
|
}
|
|
}
|