mirror of
https://github.com/diamondburned/arikawa.git
synced 2024-11-01 12:34:28 +00:00
44 lines
616 B
Go
44 lines
616 B
Go
package moreatomic
|
|
|
|
import (
|
|
"context"
|
|
)
|
|
|
|
type CtxMutex struct {
|
|
mut chan struct{}
|
|
}
|
|
|
|
func NewCtxMutex() *CtxMutex {
|
|
return &CtxMutex{
|
|
mut: make(chan struct{}, 1),
|
|
}
|
|
}
|
|
|
|
func (m *CtxMutex) Lock(ctx context.Context) error {
|
|
select {
|
|
case m.mut <- struct{}{}:
|
|
return nil
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
}
|
|
}
|
|
|
|
// TryUnlock returns true if the mutex has been unlocked.
|
|
func (m *CtxMutex) TryUnlock() bool {
|
|
select {
|
|
case <-m.mut:
|
|
return true
|
|
default:
|
|
return false
|
|
}
|
|
}
|
|
|
|
func (m *CtxMutex) Unlock() {
|
|
select {
|
|
case <-m.mut:
|
|
// return
|
|
default:
|
|
panic("Unlock of already unlocked mutex.")
|
|
}
|
|
}
|