cd43f09263
Some checks failed
Default / Build (push) Failing after 1m46s
Default / Build UI (push) Failing after 2m12s
Release / call-build-workflow (push) Has been cancelled
Release / Publish Docker :release (push) Has been cancelled
Release / Publish Github & APT release (push) Has been cancelled
67 lines
1.5 KiB
Go
67 lines
1.5 KiB
Go
package util
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
var waitInitialTimeout = 10 * time.Second
|
|
|
|
// Waiter provides monitoring of receive timeouts and reception of initial value
|
|
type Waiter struct {
|
|
sync.Mutex
|
|
log func()
|
|
cond *sync.Cond
|
|
updated time.Time
|
|
timeout time.Duration
|
|
}
|
|
|
|
// NewWaiter creates new waiter
|
|
func NewWaiter(timeout time.Duration, logInitialWait func()) *Waiter {
|
|
p := &Waiter{
|
|
log: logInitialWait,
|
|
timeout: timeout,
|
|
}
|
|
p.cond = sync.NewCond(p)
|
|
return p
|
|
}
|
|
|
|
// Update is called when client has received data. Update resets the timeout counter.
|
|
// It is client responsibility to ensure that the waiter is not locked when Update is called.
|
|
func (p *Waiter) Update() {
|
|
p.updated = time.Now()
|
|
p.cond.Broadcast()
|
|
}
|
|
|
|
// Overdue waits for initial update and returns the duration since the last update
|
|
// in excess of timeout. Waiter MUST be locked when calling Overdue.
|
|
func (p *Waiter) Overdue() time.Duration {
|
|
if p.updated.IsZero() {
|
|
p.log()
|
|
|
|
c := make(chan struct{})
|
|
|
|
go func() {
|
|
defer close(c)
|
|
for p.updated.IsZero() {
|
|
p.cond.Wait()
|
|
}
|
|
}()
|
|
|
|
select {
|
|
case <-c:
|
|
// initial value received, lock established
|
|
case <-time.After(waitInitialTimeout):
|
|
p.Update() // unblock the sync.Cond
|
|
<-c // wait for goroutine, re-establish lock
|
|
p.updated = time.Time{} // reset updated to initial value missing
|
|
return waitInitialTimeout
|
|
}
|
|
}
|
|
|
|
if elapsed := time.Since(p.updated); p.timeout != 0 && elapsed > p.timeout {
|
|
return elapsed
|
|
}
|
|
|
|
return 0
|
|
}
|