在前面的例子中,我们看到了如何使用原子操作(atomic-counters)来管理简单的计数器。 对于更加复杂的情况,我们可以使用一个互斥量 来在 Go 协程间安全的访问数据。
package main
import (
"fmt"
"sync"
)
// Container 中定义了 counters 的 map ,由于我们希望从多个 goroutine 同时更新它, 因此我们添加了一个 互斥锁Mutex 来同步访问。 请注意不能复制互斥锁,如果需要传递这个 struct,应使用指针完成。
type Container struct {
mu sync.Mutex
counters map[string]int
}
// 在访问 counters 之前锁定互斥锁; 使用 [defer](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
// 这个函数在循环中递增对 name 的计数
doIncrement := func(name string, n int) {
for i := 0; i < n; i++ {
c.inc(name)
}
wg.Done()
}
wg.Add(3)
// 同时运行多个 goroutines; 请注意,它们都访问相同的 Container,其中两个访问相同的计数器
go doIncrement("a", 10000)
go doIncrement("a", 10000)
go doIncrement("b", 10000)
// 等待上面的 goroutines 都执行结束
wg.Wait()
fmt.Println(c.counters)
}
运行这个程序显示计数器更新和我们期望的一致
$ go run mutexes.go
map[a:20000 b:10000]
原创文章,转载请注明出处:http://www.nwumba.cn/article/77/