diff options
author | 2022-03-08 11:56:53 +0000 | |
---|---|---|
committer | 2022-03-08 12:56:53 +0100 | |
commit | b8879ac68a30e8bccd1c96cc4630da791d8996c4 (patch) | |
tree | 77adeeaf2456610b771d9df8dc38207014215aea /vendor/codeberg.org/gruf/go-mutexes/mutex_safe.go | |
parent | [performance] Database optimizations (#419) (diff) | |
download | gotosocial-b8879ac68a30e8bccd1c96cc4630da791d8996c4.tar.xz |
[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>
Diffstat (limited to 'vendor/codeberg.org/gruf/go-mutexes/mutex_safe.go')
-rw-r--r-- | vendor/codeberg.org/gruf/go-mutexes/mutex_safe.go | 25 |
1 files changed, 18 insertions, 7 deletions
diff --git a/vendor/codeberg.org/gruf/go-mutexes/mutex_safe.go b/vendor/codeberg.org/gruf/go-mutexes/mutex_safe.go index 7a9747521..5a0383dce 100644 --- a/vendor/codeberg.org/gruf/go-mutexes/mutex_safe.go +++ b/vendor/codeberg.org/gruf/go-mutexes/mutex_safe.go @@ -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() + } + } } |