diff options
Diffstat (limited to 'vendor/codeberg.org/gruf/go-mutexes/mutex_timeout.go')
-rw-r--r-- | vendor/codeberg.org/gruf/go-mutexes/mutex_timeout.go | 85 |
1 files changed, 0 insertions, 85 deletions
diff --git a/vendor/codeberg.org/gruf/go-mutexes/mutex_timeout.go b/vendor/codeberg.org/gruf/go-mutexes/mutex_timeout.go deleted file mode 100644 index e00444b64..000000000 --- a/vendor/codeberg.org/gruf/go-mutexes/mutex_timeout.go +++ /dev/null @@ -1,85 +0,0 @@ -package mutexes - -import ( - "time" -) - -// TimeoutMutex defines a Mutex with timeouts on locks -type TimeoutMutex interface { - Mutex - - // LockFunc is functionally the same as Lock(), but allows setting a custom hook called on timeout - LockFunc(func()) func() -} - -// TimeoutRWMutex defines a RWMutex with timeouts on locks -type TimeoutRWMutex interface { - RWMutex - - // LockFunc is functionally the same as Lock(), but allows setting a custom hook called on timeout - LockFunc(func()) func() - - // RLockFunc is functionally the same as RLock(), but allows setting a custom hook called on timeout - RLockFunc(func()) func() -} - -// WithTimeout wraps the supplied Mutex to add a timeout -func WithTimeout(mu Mutex, d time.Duration) TimeoutMutex { - return &timeoutMutex{mu: mu, d: d} -} - -// WithTimeoutRW wraps the supplied RWMutex to add read/write timeouts -func WithTimeoutRW(mu RWMutex, rd, wd time.Duration) TimeoutRWMutex { - return &timeoutRWMutex{mu: mu, rd: rd, wd: wd} -} - -// timeoutMutex wraps a Mutex with timeout -type timeoutMutex struct { - mu Mutex // mu is the wrapped mutex - d time.Duration // d is the timeout duration -} - -func (mu *timeoutMutex) Lock() func() { - return mu.LockFunc(func() { panic("lock timed out") }) -} - -func (mu *timeoutMutex) LockFunc(fn func()) func() { - return mutexTimeout(mu.d, mu.mu.Lock(), fn) -} - -// TimeoutRWMutex wraps a RWMutex with timeouts -type timeoutRWMutex struct { - mu RWMutex // mu is the wrapped rwmutex - rd time.Duration // rd is the rlock timeout duration - wd time.Duration // wd is the lock timeout duration -} - -func (mu *timeoutRWMutex) Lock() func() { - return mu.LockFunc(func() { panic("lock timed out") }) -} - -func (mu *timeoutRWMutex) LockFunc(fn func()) func() { - return mutexTimeout(mu.wd, mu.mu.Lock(), fn) -} - -func (mu *timeoutRWMutex) RLock() func() { - return mu.RLockFunc(func() { panic("rlock timed out") }) -} - -func (mu *timeoutRWMutex) RLockFunc(fn func()) func() { - return mutexTimeout(mu.rd, mu.mu.RLock(), fn) -} - -// mutexTimeout performs a timed unlock, calling supplied fn if timeout is reached -func mutexTimeout(d time.Duration, unlock func(), fn func()) func() { - if d < 1 { - // No timeout, just unlock - return unlock - } - - // Start timer to call fn. - t := time.AfterFunc(d, fn) - - // Wrap unlock to stop mutex timer. - return func() { t.Stop(); unlock() } -} |