summaryrefslogtreecommitdiff
path: root/vendor/codeberg.org/gruf/go-mutexes/mutex_safe.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/codeberg.org/gruf/go-mutexes/mutex_safe.go')
-rw-r--r--vendor/codeberg.org/gruf/go-mutexes/mutex_safe.go25
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()
+ }
+ }
}