summaryrefslogtreecommitdiff
path: root/vendor/codeberg.org/gruf/go-mutexes/mutex_safe.go
diff options
context:
space:
mode:
authorLibravatar kim <89579420+NyaaaWhatsUpDoc@users.noreply.github.com>2022-03-08 11:56:53 +0000
committerLibravatar GitHub <noreply@github.com>2022-03-08 12:56:53 +0100
commitb8879ac68a30e8bccd1c96cc4630da791d8996c4 (patch)
tree77adeeaf2456610b771d9df8dc38207014215aea /vendor/codeberg.org/gruf/go-mutexes/mutex_safe.go
parent[performance] Database optimizations (#419) (diff)
downloadgotosocial-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.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()
+ }
+ }
}