golang chan实现互斥锁

最新推荐文章于 2024-04-07 11:24:27 发布

gitxuzan_

最新推荐文章于 2024-04-07 11:24:27 发布

阅读量293

点赞数

分类专栏： golang 文章标签： golang 开发语言后端

本文链接：https://blog.csdn.net/qq_36517296/article/details/120973347

版权

golang 专栏收录该内容

45 篇文章 1 订阅

订阅专栏

该代码示例展示了如何利用Go语言的通道（Channel）实现一个简单的互斥锁Mutex。Mutex提供了Lock、Unlock、TryLock和LockTimeout方法，确保了并发安全。TryLock尝试获取锁，如果无法立即获取则返回false。LockTimeout方法允许设置超时时间来尝试获取锁。

摘要由CSDN通过智能技术生成


// 使用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)
}