channel 实现互斥锁

2023-01-10 13:43:59 浏览数 (2)

代码语言: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()
}

0 人点赞