写点什么

Go 并发编程 - 共享变量

用户头像
Rayjun
关注
发布于: 16 小时前
Go 并发编程-共享变量

在之前的文章中,我们详细说了 Go 语言中 goroutine + channel 通过通信的方式来共享内存,从而实现并发编程。


但同时 Go 也提供了传统通过共享变量,也就是共享内存的方式来实现并发。这篇文章会介绍 Go


提供的相关机制。

1. 什么是竞态

在一个 Go 程序运行起来之后,会有很多的 goroutine 同时运行,每个 goroutine 中代码的执行是顺序的,如果我们无法确定两个 goroutine 中代码的执行顺序。就可以说这两个 goroutine 是并发执行的。


如果一段代码无论是顺序执行还是并发执行,结果都是正确的,那就可以说这个代码是并发安全的。


并发不安全的代码出现的问题有多种,比如死锁、活锁、经态等等。死锁和活锁都表示代码已经无法继续执行了,而竞态则表示代码是可以执行的,但是有可能会出现错误的结果。


有一个典型的例子就是向银行账户中存款:


var balance intfunc Deposit(amount int) {    balance = balance + amount}func Balance() int {    return balance}
复制代码


假如现在有两个人同时向这个账户中存款,各自存了 100 次:


for i := 0; i < 100; i++ {  go func() {    Deposit(100)  }()
go func() { Deposit(100) }()}// 休眠一秒,让上面的 goroutine 执行完成time.Sleep(1 * time.Second)fmt.Println(Balance())
复制代码


如果程序正确,那么最后的输出应该是 20000,但多次运行,结果可能是 19800、19900 或者其他的值。这个时候,我们就会说这个程序存在数据竞态。


这个问题的根本原因是 balance = balance + amount 这行代码在 CPU 上的执行操作不是原子的,有可能执行到一半的时候会被打断。

2. 如何消除竞态

发生了竞态,就要想办法解决。总的来说,解决竞态有三种办法:


  • 不要修改变量

  • 如果一个变量不需要修改,在任何地方访问都是安全的,但这个方法却无法解决上面的问题。

  • 不要多个 goroutine 中去访问同一个变量


我们前面说聊过的 goroutine + channel 就是这样的一个思路,通过 channel 阻塞来更新变量,这也符合 Go 代码的设计理念:不要通过共享内存来通信,而应该通过通信来共享内存


  • 同一时间只允许一个 goroutine 访问变量


如果在同一时间只能有一个 goroutine 访问变量,其他的 goruotine 需要等到当前的访问结束之后,才能访问,这样也可以消除竞态,下面将要说到的工具就是用来保证同一时间只能有一个 goroutine 来访问变量。

3. Go 提供的并发工具

在上面我们已经说到了解决竞态的三种办法,下面的这些工具就是 Go 中用来实现同一时间只能有一个 goroutine 访问变量。我们分别来看一下:

3.1 互斥锁

这个是解决竞态最经典的工具,它的原理就是如果要访问一个资源,那么就必须要拿到这个资源的锁,只有拿到锁才有资格访问资源,其他的 goroutine 想要访问,必须等到当前 goroutine 释放了锁,抢到锁之后再访问


在使用之前,需要先为资源申请一把锁,使用的就是 sync.Mutex,这是 Go 语言中互斥锁的实现:


var mu sync.Mutexvar balance int
复制代码


每个拿到锁的 goroutine 都需要保证在对变量的访问结束之后,把锁释放掉,即使发生在异常情况,也需要释放,这里可以使用 defer 来保证最终会释放锁:


func Deposit(amount int) {  mu.Lock()  defer mu.Unlock()  balance = balance + amount}
func Balance() int { mu.Lock() defer mu.Unlock() return balance}
复制代码


把代码改完之后,再去运行上面存款的代码,无论运行多少遍,最终的结果都是 20000,到这里,我们竞态的问题就算是解决了,但是还有点小问题。

3.2 读写互斥锁

上面的互斥锁解决了访问数据的竞态问题,但是还有个小问题,就是读余额的操作有点低效,每次来读余额的时候,都还需要去抢锁,实际上,这个变量如果没有改变,即使同时被多个 goroutine 读,也不会产生并发安全的问题。


我们想要的一个理想的场景就是,如果这个变量没有在写入,就可以运行多个 goroutine 同时读,这样可以大大提高效率。


Go 也提供了这个工具,那就是读写锁。这个锁读与读是不互斥的,简单来说就是这个锁可以保证同时只能有一个 goroutine 在写入,如果有 goroutine 在写入,其他的 goroutine 既不能读,也不能写,但允许多个 goroutine 同时来读。


我们把上面的代码再改一下,只需要改一个地方:


var mu sync.RWMutex // 替换 sync.Mutexvar balance int
复制代码


这样改完之后,上面存款的代码还是会一直输出 20000,但同时可以允许多个 goroutine 同时读余额。


大多数 Go 语言中的竞态问题都可以使用这两个工具来解决。

3.3 Once

Go 语言中还提供了这样的一个工具,可以保证代码只会执行一遍,多用于资源初始化等场景。使用的方式也很简单:


o := &sync.Once{}for i := 0; i < 100; i++ {  o.Do(func(){    go func() {      Deposit(100)    }()
go func() { Deposit(100) }() })}// 休眠一秒,让上面的 goroutine 执行完成time.Sleep(1 * time.Second)fmt.Println(Balance())
复制代码


如果上面的代码使用 Once 来控制之后,都只会存一次,所以上面的代码会永远输出 200。

3.4 竞态检测器

很多处在竞态的错误很难发现,Go 语言中提供了一个工具,可以帮忙检查代码中是否存在竞态。使用起来很简单,只需要在以下命令之后加上 -race 参数就可以:


$ go run -race
$ go build -race
$ go test -race
复制代码


加上这个参数之后,编译器会对代码在执行时对所有共享变量的访问,如果发现一个 goroutine 写入一个变量之后,没有任何同步的操作,就有另外一个 goroutine 读写了这个变量,那就说明这里存在竞态,就会报错。比如下面的代码:


data := 1
go func() { data = 2}()
go func() { data = 3}()
time.Sleep(2 * time.Second)
复制代码


运行 go run -race main.go 之后,会报下面的错误:


Found 1 data race(s)exit status 66
复制代码

4. 小结

Go 中也提供了传统语言所提供的并发编程机制,也可以通过共享内存的方法来实现并发编程。Go 提供的接口相对来说比较简洁,提供的能力却足够强大。


文 / Rayjun

本文首发于微信公众号【Rayjun】

发布于: 16 小时前阅读数: 6
用户头像

Rayjun

关注

程序员,王小波死忠粉 2017.10.17 加入

非著名程序员,还在学习如何写代码,公众号同名

评论

发布
暂无评论
Go 并发编程-共享变量