代码语言:javascript复制
//channel 实现互斥锁
type Mutex struct {
ch chan struct{}
}
func newMutex() *Mutex {
mu := Mutex{ch: make(chan struct{}, 1)}
mu.ch <- struct{}{}
return &mu
}
// 加锁,也就是可以直接取出来数据
func (m *Mutex) lock() {
<-m.ch
}
func (m *Mutex) tryLock() {
select {
case <-m.ch:
default:
}
}
//解锁
func (m *Mutex) unLock() {
select {
case m.ch <- struct{}{}:
return
default:
panic("unlock of unlocked mutex")
}
}
func Test_chanMutex(t *testing.T) {
m := newMutex()
m.lock()
m.tryLock()
t.Log("hahaha")
m.unLock()
}