Go by Example: Mutexes 互斥锁

英文源地址
在之前的示例中, 我们看到了如何使用原子操作来管理简单的计数器状态. 对于复杂的状态, 我们可以使用互斥锁跨多个协程安全地访问数据.

package main

import (
	"fmt"
	"sync"
)

// Container 存放计数器地映射: 因为我们想从多个协程并发地更新它
// 所以我们添加了一个互斥锁来同步访问
// 注意, 互斥锁不能被复制.所以如果传递这个结构体, 应该通过指针来完成.
type Container struct {
    
    
	mu       sync.Mutex
	counters map[string]int
}

// 在访问计数器前锁定互斥锁, 在函数末尾使用defer语句解锁它
func (c *Container) inc(name string) {
    
    
	c.mu.Lock()
	defer c.mu.Unlock()
	c.counters[name]++
}

// 注意互斥锁的零值是可用的, 所以这里不需要初始化
func main() {
    
    
	c := Container{
    
    
		counters: map[string]int{
    
    "a": 0, "b": 0},
	}
	var wg sync.WaitGroup

	// 这个函数在循环中增加一个命名的计数器
	doIncrement := func(name string, n int) {
    
    
		for i := 0; i < n; i++ {
    
    
			c.inc(name)
		}
		wg.Done()
	}

	// 并发运行多个协程时注意, 它们都访问同一个container, 其中两个访问同一个计时器
	wg.Add(3)
	go doIncrement("a", 10000)
	go doIncrement("a", 10000)
	go doIncrement("b", 10000)

	// 等待协程运行完成
	wg.Wait()
	fmt.Println(c.counters)
}

运行程序显示计数器按预期进行了更新.

$ go run mutexes.go
map[a:20000 b:10000]

接下来, 我们将研究如何仅仅使用协程和通道来实现相同的状态管理任务.