You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
724 B
Go
52 lines
724 B
Go
package syncx
|
|
|
|
import (
|
|
"errors"
|
|
"time"
|
|
)
|
|
|
|
var ErrTimeout = errors.New("borrow timeout")
|
|
|
|
type TimeoutLimit struct {
|
|
limit Limit
|
|
cond *Cond
|
|
}
|
|
|
|
func NewTimeoutLimit(n int) TimeoutLimit {
|
|
return TimeoutLimit{
|
|
limit: NewLimit(n),
|
|
cond: NewCond(),
|
|
}
|
|
}
|
|
|
|
func (l TimeoutLimit) Borrow(timeout time.Duration) error {
|
|
if l.TryBorrow() {
|
|
return nil
|
|
}
|
|
|
|
var ok bool
|
|
for {
|
|
timeout, ok = l.cond.WaitWithTimeout(timeout)
|
|
if ok && l.TryBorrow() {
|
|
return nil
|
|
}
|
|
|
|
if timeout <= 0 {
|
|
return ErrTimeout
|
|
}
|
|
}
|
|
}
|
|
|
|
func (l TimeoutLimit) Return() error {
|
|
if err := l.limit.Return(); err != nil {
|
|
return err
|
|
}
|
|
|
|
l.cond.Signal()
|
|
return nil
|
|
}
|
|
|
|
func (l TimeoutLimit) TryBorrow() bool {
|
|
return l.limit.TryBorrow()
|
|
}
|