[dependencies] update go-store, go-mutexes (#422)

* update go-store, go-mutexes

Signed-off-by: kim <grufwub@gmail.com>

* update vendored code

Signed-off-by: kim <grufwub@gmail.com>
This commit is contained in:
kim
2022-03-08 11:56:53 +00:00
committed by GitHub
parent 8de928b5e9
commit b8879ac68a
13 changed files with 639 additions and 268 deletions

View File

@ -1,6 +1,8 @@
package mutexes
import "sync"
import (
"sync/atomic"
)
// WithSafety wrapps the supplied Mutex to protect unlock fns
// from being called multiple times
@ -19,8 +21,7 @@ type safeMutex struct{ mu Mutex }
func (mu *safeMutex) Lock() func() {
unlock := mu.mu.Lock()
once := sync.Once{}
return func() { once.Do(unlock) }
return once(unlock)
}
// safeRWMutex simply wraps a RWMutex to add multi-unlock safety
@ -28,12 +29,22 @@ type safeRWMutex struct{ mu RWMutex }
func (mu *safeRWMutex) Lock() func() {
unlock := mu.mu.Lock()
once := sync.Once{}
return func() { once.Do(unlock) }
return once(unlock)
}
func (mu *safeRWMutex) RLock() func() {
unlock := mu.mu.RLock()
once := sync.Once{}
return func() { once.Do(unlock) }
return once(unlock)
}
// once will perform 'do' only once, this is safe for unlocks
// as 2 functions calling 'unlock()' don't need absolute guarantees
// that by the time it is completed the unlock was finished.
func once(do func()) func() {
var done uint32
return func() {
if atomic.CompareAndSwapUint32(&done, 0, 1) {
do()
}
}
}