package syncx import ( "time" "git.i2edu.net/i2/go-zero/core/lang" "git.i2edu.net/i2/go-zero/core/timex" ) // A Cond is used to wait for conditions. type Cond struct { signal chan lang.PlaceholderType } // NewCond returns a Cond. func NewCond() *Cond { return &Cond{ signal: make(chan lang.PlaceholderType), } } // WaitWithTimeout wait for signal return remain wait time or timed out. func (cond *Cond) WaitWithTimeout(timeout time.Duration) (time.Duration, bool) { timer := time.NewTimer(timeout) defer timer.Stop() begin := timex.Now() select { case <-cond.signal: elapsed := timex.Since(begin) remainTimeout := timeout - elapsed return remainTimeout, true case <-timer.C: return 0, false } } // Wait waits for signals. func (cond *Cond) Wait() { <-cond.signal } // Signal wakes one goroutine waiting on c, if there is any. func (cond *Cond) Signal() { select { case cond.signal <- lang.Placeholder: default: } }