mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-19 07:26:49 +00:00
242b02c382
* Removes lock unlock timer and instead sets unlocks between getting a nonce and sending a payload. Increases dispatch channel buffer to deal with len(enabledCurrencies) > ~100 * Adds additional comments to help explain the situation * Fixes bug that could unlock mutex too early * Fixes LIES where Gemini gets a nonce and then proceeds to declare it doesn't get a nonce causing an unrecoverable lock * Fun new concept! The creation of a tested timed mutex. Unlocking an unlocked mutex cannot occur and response can be checked to verify whether the mutex was unlocked from timeout or command. * Adds new cmd parameter "dispatchjobbuffer" * Expands comments and renames benchmark. Makes `Timer` property private * Happy little linters * Renames jobBuffer and all related instances to jobs limit * Tiny error message update * Grammatical fix and setting dispatch.Start to use defaults
78 lines
1.5 KiB
Go
78 lines
1.5 KiB
Go
package timedmutex
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
// NewTimedMutex creates a new timed mutex with a
|
|
// specified duration
|
|
func NewTimedMutex(length time.Duration) *TimedMutex {
|
|
return &TimedMutex{
|
|
duration: length,
|
|
}
|
|
}
|
|
|
|
// LockForDuration will start a timer, lock the mutex,
|
|
// then allow the caller to continue
|
|
// After the duration, the mutex will be unlocked
|
|
func (t *TimedMutex) LockForDuration() {
|
|
var wg sync.WaitGroup
|
|
wg.Add(1)
|
|
go t.lockAndSetTimer(&wg)
|
|
wg.Wait()
|
|
}
|
|
|
|
func (t *TimedMutex) lockAndSetTimer(wg *sync.WaitGroup) {
|
|
t.mtx.Lock()
|
|
t.setTimer()
|
|
wg.Done()
|
|
}
|
|
|
|
// UnlockIfLocked will unlock the mutex if its currently locked
|
|
// Will return true if successfully unlocked
|
|
func (t *TimedMutex) UnlockIfLocked() bool {
|
|
if t.isTimerNil() {
|
|
return false
|
|
}
|
|
|
|
if !t.stopTimer() {
|
|
return false
|
|
}
|
|
t.mtx.Unlock()
|
|
return true
|
|
}
|
|
|
|
// stopTimer will return true if timer has been stopped by this command
|
|
// If the timer has expired, clear the channel
|
|
func (t *TimedMutex) stopTimer() bool {
|
|
t.timerLock.Lock()
|
|
defer t.timerLock.Unlock()
|
|
if !t.timer.Stop() {
|
|
select {
|
|
case <-t.timer.C:
|
|
default:
|
|
}
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
// isTimerNil safely read locks to detect nil
|
|
func (t *TimedMutex) isTimerNil() bool {
|
|
t.timerLock.RLock()
|
|
defer t.timerLock.RUnlock()
|
|
return t.timer == nil
|
|
}
|
|
|
|
// setTimer safely locks and sets a timer
|
|
// which will automatically execute a mutex unlock
|
|
// once timer expires
|
|
func (t *TimedMutex) setTimer() {
|
|
t.timerLock.Lock()
|
|
t.timer = time.AfterFunc(t.duration, func() {
|
|
t.mtx.Unlock()
|
|
})
|
|
t.timerLock.Unlock()
|
|
}
|