在之前的文章中，我們詳細說了 Go 語言中 goroutine + channel 通過通信的方式來共享內(nèi)存，從而實現(xiàn)并發(fā)編程。

但同時 Go 也提供了傳統(tǒng)通過共享變量，也就是共享內(nèi)存的方式來實現(xiàn)并發(fā)。這篇文章會介紹 Go

提供的相關(guān)機制。

1. 什么是競態(tài)

在一個 Go 程序運行起來之后，會有很多的 goroutine 同時運行，每個 goroutine 中代碼的執(zhí)行是順序的，如果我們無法確定兩個 goroutine 中代碼的執(zhí)行順序。就可以說這兩個 goroutine 是并發(fā)執(zhí)行的。

如果一段代碼無論是順序執(zhí)行還是并發(fā)執(zhí)行，結(jié)果都是正確的，那就可以說這個代碼是并發(fā)安全的。

并發(fā)不安全的代碼出現(xiàn)的問題有多種，比如死鎖、活鎖、經(jīng)態(tài)等等。死鎖和活鎖都表示代碼已經(jīng)無法繼續(xù)執(zhí)行了，而競態(tài)則表示代碼是可以執(zhí)行的，但是有可能會出現(xiàn)錯誤的結(jié)果。

有一個典型的例子就是向銀行賬戶中存款：

var balance int
func Deposit(amount int) {
    balance = balance + amount
}
func Balance() int {
    return balance
}

假如現(xiàn)在有兩個人同時向這個賬戶中存款，各自存了 100 次：

for i := 0; i < 100; i++ {
    go func() {
        Deposit(100)
    }()

    go func() {
        Deposit(100)
    }()
}
// 休眠一秒，讓上面的 goroutine 執(zhí)行完成
time.Sleep(1 * time.Second)
fmt.Println(Balance())

如果程序正確，那么最后的輸出應(yīng)該是 20000，但多次運行，結(jié)果可能是 19800、19900 或者其他的值。這個時候，我們就會說這個程序存在數(shù)據(jù)競態(tài)。

這個問題的根本原因是 balance = balance + amount 這行代碼在 CPU 上的執(zhí)行操作不是原子的，有可能執(zhí)行到一半的時候會被打斷。

2. 如何消除競態(tài)

發(fā)生了競態(tài)，就要想辦法解決。總的來說，解決競態(tài)有三種辦法：

不要修改變量

如果一個變量不需要修改，在任何地方訪問都是安全的，但這個方法卻無法解決上面的問題。

不要多個 goroutine 中去訪問同一個變量

我們前面說聊過的 goroutine + channel 就是這樣的一個思路，通過 channel 阻塞來更新變量，這也符合 Go 代碼的設(shè)計理念：不要通過共享內(nèi)存來通信，而應(yīng)該通過通信來共享內(nèi)存。

同一時間只允許一個 goroutine 訪問變量
如果在同一時間只能有一個 goroutine 訪問變量，其他的 goruotine 需要等到當(dāng)前的訪問結(jié)束之后，才能訪問，這樣也可以消除競態(tài)，下面將要說到的工具就是用來保證同一時間只能有一個 goroutine 來訪問變量。

3. Go 提供的并發(fā)工具

在上面我們已經(jīng)說到了解決競態(tài)的三種辦法，下面的這些工具就是 Go 中用來實現(xiàn)同一時間只能有一個 goroutine 訪問變量。我們分別來看一下：

3.1 互斥鎖

這個是解決競態(tài)最經(jīng)典的工具，它的原理就是如果要訪問一個資源，那么就必須要拿到這個資源的鎖，只有拿到鎖才有資格訪問資源，其他的 goroutine 想要訪問，必須等到當(dāng)前 goroutine 釋放了鎖，搶到鎖之后再訪問

在使用之前，需要先為資源申請一把鎖，使用的就是 sync.Mutex，這是 Go 語言中互斥鎖的實現(xiàn)：

var mu sync.Mutex
var balance int

每個拿到鎖的 goroutine 都需要保證在對變量的訪問結(jié)束之后，把鎖釋放掉，即使發(fā)生在異常情況，也需要釋放，這里可以使用 defer 來保證最終會釋放鎖：

func Deposit(amount int) {
    mu.Lock()
    defer mu.Unlock()
    balance = balance + amount
}

func Balance() int {
    mu.Lock()
    defer mu.Unlock()
    return balance
}

把代碼改完之后，再去運行上面存款的代碼，無論運行多少遍，最終的結(jié)果都是 20000，到這里，我們競態(tài)的問題就算是解決了，但是還有點小問題。

3.2 讀寫互斥鎖

上面的互斥鎖解決了訪問數(shù)據(jù)的競態(tài)問題，但是還有個小問題，就是讀余額的操作有點低效，每次來讀余額的時候，都還需要去搶鎖，實際上，這個變量如果沒有改變，即使同時被多個 goroutine 讀，也不會產(chǎn)生并發(fā)安全的問題。

我們想要的一個理想的場景就是，如果這個變量沒有在寫入，就可以運行多個 goroutine 同時讀，這樣可以大大提高效率。

Go 也提供了這個工具，那就是讀寫鎖。這個鎖讀與讀是不互斥的，簡單來說就是這個鎖可以保證同時只能有一個 goroutine 在寫入，如果有 goroutine 在寫入，其他的 goroutine 既不能讀，也不能寫，但允許多個 goroutine 同時來讀。

我們把上面的代碼再改一下，只需要改一個地方：

var mu sync.RWMutex // 替換 sync.Mutex
var balance int

這樣改完之后，上面存款的代碼還是會一直輸出 20000，但同時可以允許多個 goroutine 同時讀余額。

大多數(shù) Go 語言中的競態(tài)問題都可以使用這兩個工具來解決。

3.3 Once

Go 語言中還提供了這樣的一個工具，可以保證代碼只會執(zhí)行一遍，多用于資源初始化等場景。使用的方式也很簡單：

o := &sync.Once{}
for i := 0; i < 100; i++ {
    o.Do(func(){
        go func() {
            Deposit(100)
        }()

        go func() {
            Deposit(100)
        }()
    })
}
// 休眠一秒，讓上面的 goroutine 執(zhí)行完成
time.Sleep(1 * time.Second)
fmt.Println(Balance())

如果上面的代碼使用 Once 來控制之后，都只會存一次，所以上面的代碼會永遠輸出 200。

3.4 競態(tài)檢測器

很多處在競態(tài)的錯誤很難發(fā)現(xiàn)，Go 語言中提供了一個工具，可以幫忙檢查代碼中是否存在競態(tài)。使用起來很簡單，只需要在以下命令之后加上 -race 參數(shù)就可以：

$ go run -race
$ go build -race
$ go test -race

加上這個參數(shù)之后，編譯器會對代碼在執(zhí)行時對所有共享變量的訪問，如果發(fā)現(xiàn)一個 goroutine 寫入一個變量之后，沒有任何同步的操作，就有另外一個 goroutine 讀寫了這個變量，那就說明這里存在競態(tài)，就會報錯。比如下面的代碼：

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. 小結(jié)

Go 中也提供了傳統(tǒng)語言所提供的并發(fā)編程機制，也可以通過共享內(nèi)存的方法來實現(xiàn)并發(fā)編程。Go 提供的接口相對來說比較簡潔，提供的能力卻足夠強大。

到此這篇關(guān)于Golang共享變量如何解決問題的文章就介紹到這了,更多相關(guān)Golang共享變量內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評論