清華尹成帶你實戰GO案例（35)Go 互斥

Go 互斥
上面的例子中，我們看過了如何在多個協程之間原子地訪問計數器，對於更復雜的例子，我們可以使用
Mutex 來在多個協程之間安全地訪問資料。

package main
import (
"fmt"
"math/rand"
"runtime"
"sync"
"sync/atomic"
"time"
)
func main() {
// 這個例子的狀態就是一個map
var state = make(map[int]int)
// 這個`mutex`將同步對狀態的訪問
var mutex = &sync.Mutex{}
// ops將對狀態的操作進行計數
var ops int64 = 0
// 這裡我們啟動100個協程來不斷地讀取這個狀態
for r := 0; r < 100; r++ {
go func() {
total := 0
for {
// 對於每次讀取，我們選取一個key來訪問，
// mutex的`Lock`函式用來保證對狀態的
// 唯一性訪問，訪問結束後，使用`Unlock`
// 來解鎖，然後增加ops計數器
key := rand.Intn(5)
mutex.Lock()
total += state[key]
mutex.Unlock()
atomic.AddInt64(&ops, 1)
// 為了保證這個協程不會讓排程器出於飢餓狀態，
// 我們顯式地使用`runtime.Gosched`釋放了資源
// 控制權，這種控制權會在通道操作結束或者
// time.Sleep結束後自動釋放。但是這裡我們需要
// 手動地釋放資源控制權
runtime.Gosched()
}
}()
}
// 同樣我們使用10個協程來模擬寫狀態
for w := 0; w < 10; w++ {
go func() {
for {
key := rand.Intn(5)
val := rand.Intn(100)
mutex.Lock()
state[key] = val
mutex.Unlock()
atomic.AddInt64(&ops, 1)
runtime.Gosched()
}
}()
}
// 主協程Sleep，讓那10個協程能夠執行一段時間
time.Sleep(time.Second)
// 輸出總操作次數
opsFinal := atomic.LoadInt64(&ops)
fmt.Println("ops:", opsFinal)
// 最後鎖定並輸出狀態
mutex.Lock()
fmt.Println("state:", state)
mutex.Unlock()
}

執行結果

ops: 3931611

state: map[0:84 2:20 3:18 1:65 4:31]

網址：http://www.qukuailianxueyuan.io/

欲領取造幣技術與全套虛擬機器資料

區塊鏈技術交流QQ群：756146052  備註：CSDN

尹成學院微信：備註：CSDN