channel 实现互斥锁

钚该钚想

于 2023-01-09 16:48:56 发布

阅读量218

点赞数

分类专栏： golang 文章标签： go

本文链接：https://blog.csdn.net/qq_37632591/article/details/128617083

版权

golang 专栏收录该内容

3 篇文章 0 订阅

订阅专栏


//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()
}