45. sync.Mutex 互斥和互斥锁

作者: 厚土火焱 | 来源:发表于2017-09-02 14:52 被阅读140次

    channel 在 goroutine 中互相通信是非常合适且方便的。如果,我们不需要互相通信,只需要保证同一时刻只能有一个 goroutine 访问共享变量,以免冲突。我们该怎么做呢?
    这就需要用到 互斥和互斥锁。
    go 语言标准库中提供了 sync.Mutex 类型及两个方法: Lock 和 Unlock
    我们可以通过在代码前加 Lock,在代码后加 Unlock 的方法,保证代码执行时的互斥性。

        c.mux.Lock()    
        c.v[key]++      //Lock之后,同一时刻只有一个 goroutine 能访问 c.v
        c.mux.Unlock()
    

    也可以用 defer 语句来保证互斥锁一定会被解锁。

        c.mux.Lock()
        defer c.mux.Unlock()
    

    创建一个代码示例,先建立一个计数器结构体

    //安全计数器,使用 sync.Mutex 保证安全
    type SafeCounter struct {
        v map[string]int
        mux sync.Mutex
    }
    

    v 是计数器map,再加上一个 sync.Mutex 类型。
    给 SafeCounter 增加两个方法,一个 inc 来实现计数器值累加,一个 Value 来实现取得当前计数器值。

    //增加计数器 key 的计数值
    func (c *SafeCounter) Inc(key string)  {
        c.mux.Lock()
        //Lock之后,同一时刻只有一个 goroutine 能访问 c.v
        c.v[key]++
        c.mux.Unlock()
    }
    //取得计数器当前值
    func (c *SafeCounter) Value(key string) int {
        c.mux.Lock()
        defer c.mux.Unlock()
        return c.v[key]
    }
    

    主函数循环 100 次,使用 goroutine 让一个计数器去增加值。然后取得当前计数器的值。
    为了避免运行太快,看不到效果,每次循环要间隔 100 毫秒。

    time.Sleep(100*time.Millisecond)
    

    完整代码示例

    package main
    
    import (
        "sync"
        "fmt"
        "time"
    )
    
    //安全计数器,使用 sync.Mutex 保证安全
    type SafeCounter struct {
        v map[string]int
        mux sync.Mutex
    }
    
    //增加计数器 key 的计数值
    func (c *SafeCounter) Inc(key string)  {
        c.mux.Lock()
        //Lock之后,同一时刻只有一个 goroutine 能访问 c.v
        c.v[key]++
        c.mux.Unlock()
    }
    
    //取得计数器当前值
    func (c *SafeCounter) Value(key string) int {
        c.mux.Lock()
        defer c.mux.Unlock()
        return c.v[key]
    }
    func main() {
        c := SafeCounter{v:make(map[string]int)}
        for i := 0; i < 100; i++{
            go c.Inc("somekey")
            time.Sleep(100*time.Millisecond)
            fmt.Println(c.Value("somekey"))
        }
        fmt.Println(c.Value("somekey"))
    }
    

    运行结果太长,不做展示。请自行实践。

    相关文章

      网友评论

        本文标题:45. sync.Mutex 互斥和互斥锁

        本文链接:https://www.haomeiwen.com/subject/qyfrjxtx.html