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.go39
1 files changed, 39 insertions, 0 deletions
diff --git a/vendor/codeberg.org/gruf/go-mutexes/mutex_safe.go b/vendor/codeberg.org/gruf/go-mutexes/mutex_safe.go
new file mode 100644
index 000000000..7a9747521
--- /dev/null
+++ b/vendor/codeberg.org/gruf/go-mutexes/mutex_safe.go
@@ -0,0 +1,39 @@
+package mutexes
+
+import "sync"
+
+// WithSafety wrapps the supplied Mutex to protect unlock fns
+// from being called multiple times
+func WithSafety(mu Mutex) Mutex {
+ return &safeMutex{mu: mu}
+}
+
+// WithSafetyRW wrapps the supplied RWMutex to protect unlock
+// fns from being called multiple times
+func WithSafetyRW(mu RWMutex) RWMutex {
+ return &safeRWMutex{mu: mu}
+}
+
+// safeMutex simply wraps a Mutex to add multi-unlock safety
+type safeMutex struct{ mu Mutex }
+
+func (mu *safeMutex) Lock() func() {
+ unlock := mu.mu.Lock()
+ once := sync.Once{}
+ return func() { once.Do(unlock) }
+}
+
+// safeRWMutex simply wraps a RWMutex to add multi-unlock safety
+type safeRWMutex struct{ mu RWMutex }
+
+func (mu *safeRWMutex) Lock() func() {
+ unlock := mu.mu.Lock()
+ once := sync.Once{}
+ return func() { once.Do(unlock) }
+}
+
+func (mu *safeRWMutex) RLock() func() {
+ unlock := mu.mu.RLock()
+ once := sync.Once{}
+ return func() { once.Do(unlock) }
+}