diff options
author | 2021-11-13 12:29:08 +0100 | |
---|---|---|
committer | 2021-11-13 12:29:08 +0100 | |
commit | 829a934d23ab221049b4d54926305d8d5d64c9ad (patch) | |
tree | f4e382b289c113d3ba8a3c7a183507a5609c46c0 /vendor/codeberg.org/gruf/go-mutexes/mutex_safe.go | |
parent | smtp + email confirmation (#285) (diff) | |
download | gotosocial-829a934d23ab221049b4d54926305d8d5d64c9ad.tar.xz |
update dependencies (#296)
Diffstat (limited to 'vendor/codeberg.org/gruf/go-mutexes/mutex_safe.go')
-rw-r--r-- | vendor/codeberg.org/gruf/go-mutexes/mutex_safe.go | 39 |
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) } +} |