mirror of
https://github.com/diamondburned/arikawa.git
synced 2024-12-03 20:32:55 +00:00
34 lines
436 B
Go
34 lines
436 B
Go
|
package moreatomic
|
||
|
|
|
||
|
|
import "github.com/sasha-s/go-deadlock"
|
||
|
|
|
||
|
|
type BusyMutex struct {
|
||
|
|
busy Bool
|
||
|
|
mut deadlock.Mutex
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *BusyMutex) TryLock() bool {
|
||
|
|
if m.busy.Get() {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
m.mut.Lock()
|
||
|
|
m.busy.Set(true)
|
||
|
|
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *BusyMutex) IsBusy() bool {
|
||
|
|
return m.busy.Get()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *BusyMutex) Lock() {
|
||
|
|
m.mut.Lock()
|
||
|
|
m.busy.Set(true)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *BusyMutex) Unlock() {
|
||
|
|
m.busy.Set(false)
|
||
|
|
m.mut.Unlock()
|
||
|
|
}
|