Golang 互斥和互斥锁

  channel在goroutine中互相通信是非常合适且方便的。如果我们不需要相互通信，只需要保证同一时刻只能有一个goroutine访问共享变量，以免冲突，这就需要用到互斥和互斥锁。

  go语言标准库中提供了sync.Mutex类型及两个方法：Lock和Unlock。

  我们可以通过在代码前加Lock，在代码后加Unlock的方法，保证代码执行时的互斥性。

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

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

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

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

  v是计数器map，再加上一个sync.Mutex类型。

 

  getsafeCounter增加两个方法，一个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"))
}

 

参考：