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

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

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

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

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

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

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

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

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

  1. //增加计数器 key 的计数值
  2. func (c *SafeCounter) Inc(key string) {
  3. c.mux.Lock()
  4. //Lock之后,同一时刻只有一个 goroutine 能访问 c.v
  5. c.v[key]++
  6. c.mux.Unlock()
  7. }
  8. //取得计数器当前值
  9. func (c *SafeCounter) Value(key string) int {
  10. c.mux.Lock()
  11. defer c.mux.Unlock()
  12. return c.v[key]
  13. }

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

  1. time.Sleep(100*time.Millisecond)

完整代码示例

  1. package main
  2. import (
  3. "sync"
  4. "fmt"
  5. "time"
  6. )
  7. //安全计数器,使用 sync.Mutex 保证安全
  8. type SafeCounter struct {
  9. v map[string]int
  10. mux sync.Mutex
  11. }
  12. //增加计数器 key 的计数值
  13. func (c *SafeCounter) Inc(key string) {
  14. c.mux.Lock()
  15. //Lock之后,同一时刻只有一个 goroutine 能访问 c.v
  16. c.v[key]++
  17. c.mux.Unlock()
  18. }
  19. //取得计数器当前值
  20. func (c *SafeCounter) Value(key string) int {
  21. c.mux.Lock()
  22. defer c.mux.Unlock()
  23. return c.v[key]
  24. }
  25. func main() {
  26. c := SafeCounter{v:make(map[string]int)}
  27. for i := 0; i < 100; i++{
  28. go c.Inc("somekey")
  29. time.Sleep(100*time.Millisecond)
  30. fmt.Println(c.Value("somekey"))
  31. }
  32. fmt.Println(c.Value("somekey"))
  33. }

Golang 互斥和互斥锁 - 图1