Golang 使用管道实现互斥锁 – 兰陵美酒郁金香的个人博客


// 使用chan实现互斥锁
type Mutex struct {
    ch chan struct{}
}

// 使用锁需要初始化
func NewMutex() *Mutex {
    mu := &Mutex{make(chan struct{}, 1)}
    mu.ch <- struct{}{}
    return mu
}

// 请求锁，直到获取到
func (m *Mutex) Lock() {
    <-m.ch
}

// 解锁
func (m *Mutex) Unlock() {
    select {
    case m.ch <- struct{}{}:
    default:
        panic("unlock of unlocked mutex")
    }
}

// 尝试获取锁
func (m *Mutex) TryLock() bool {
    select {
    case <-m.ch:
        return true
    default:
    }
    return false
}

// 加入一个超时的设置
func (m *Mutex) LockTimeout(timeout time.Duration) bool {
    timer := time.NewTimer(timeout)
    select {
    case <-m.ch:
        timer.Stop()
        return true
    case <-timer.C:
    }
    return false
}

// 锁是否已被持有
func (m *Mutex) IsLocked() bool {
    return len(m.ch) == 0
}


func main() {
    m := NewMutex()
    ok := m.TryLock()
    fmt.Printf("locked v %v\n", ok)
    ok = m.TryLock()
    fmt.Printf("locked %v\n", ok)
}

// 使用chan实现互斥锁

type Mutex struct {

ch chan struct{}

}

// 使用锁需要初始化

func NewMutex() *Mutex {

mu := &Mutex{make(chan struct{}, 1)}

mu.ch <- struct{}{}

return mu

}

// 请求锁，直到获取到

func (m *Mutex) Lock() {

<-m.ch

}

// 解锁

func (m *Mutex) Unlock() {

select {

case m.ch <- struct{}{}:

default:

panic("unlock of unlocked mutex")

}

// 尝试获取锁

func (m *Mutex) TryLock() bool {

select {

case <-m.ch:

return true

default:

}

return false

}

// 加入一个超时的设置

func (m *Mutex) LockTimeout(timeout time.Duration) bool {

timer := time.NewTimer(timeout)

select {

case <-m.ch:

timer.Stop()

return true

case <-timer.C:

}

return false

}

// 锁是否已被持有

func (m *Mutex) IsLocked() bool {

return len(m.ch) == 0

}

func main() {

m := NewMutex()

ok := m.TryLock()

fmt.Printf("locked v %v\n", ok)

ok = m.TryLock()

fmt.Printf("locked %v\n", ok)

}

暂无

兰陵美酒郁金香

大道至简 Simplicity is the ultimate form of sophistication.

文章评论（0）

你必须登录才能发表评论