Go語言中通過Groutine 啟動一個Go協程，不同協程之間是并發執行的，就像C++/Java中線程之間線程安全是一個常見的問題。
如下Go 語言代碼:

func TestConcurrent(t *testing.T) {var counter int = 0for i := 0;i < 5000; i ++{go func() { // 啟動groutine 進行變量的自增counter ++}()}time.Sleep(time.Second * 1)t.Logf("Counter = %d", counter) 
}

最后的結果輸出如下:

=== RUN   TestConcurrentconcurrent_test.go:18: Counter = 4663
--- PASS: TestConcurrent (1.00s)

希望變量Counter的結果是5000，最終因為協程之間并發訪問共享內存的原因導致部分協程執行的結果沒有得到變更，從而數據不符合預期。

Go中提供了Mutex和RWMutex ，來實現鎖功能。前者就是正常的Lock, Unlock；后者提供了讀寫鎖，在協程讀寫分離中提供了讀讀互不影響，讀寫互斥。

如下測試代碼：

func TestMutex(t *testing.T) {var counter int = 0// var mu sync.Mutex // mutex lockvar mu sync.RWMutex // rwmutex lockfor i := 0;i < 5000; i ++{go func() {// defer mutex unlock// 如果協程異常退出，能夠釋放鎖defer func() { mu.Unlock()}()mu.Lock()counter ++}()}time.Sleep(time.Second * 1)t.Logf("Counter = %d", counter)
}

最終輸出如下，通過mutex 實現了變量訪問的安全:

=== RUN   TestMutexconcurrent_test.go:36: Counter = 5000
--- PASS: TestMutex (1.01s)

可以看到如上代碼需要通過Sleep來讓所有的協程都執行完，顯然不合理。、

假如有更多的協程，這個sleep時間還需要程序員來修改，Go語言提供了WaitGroup機制，就像是C++中的條件變量，主線程可以等待在條件變量上，當線程執行完可以通過條件變量喚醒主線程繼續執行。
var wg sync.WaitGroup 聲明一個WaitGroup變量
通過wg.Add(1) 和 wg.Done() 來檢測線程的完成情況，wg.Wait()表示主線程等待在wg變量上

func TestWaitGroup(t *testing.T) {var counter int = 0var mu sync.RWMutex // rw lock, 讀寫互斥var wg sync.WaitGroupfor i := 0;i < 5000; i ++{wg.Add(1) go func() {defer func() {mu.Unlock()}()mu.Lock()counter ++wg.Done() // wg.Add(-1)}()}wg.Wait()t.Logf("Counter = %d", counter)
}

輸出如下，可以看到TestWaitGroup 函數的執行時間很快，不需要Sleep方式來消耗更多的時間使程序被動執行完畢。

=== RUN   TestWaitGroupconcurrent_test.go:57: Counter = 5000
--- PASS: TestWaitGroup (0.00s)

總結

以上是生活随笔為你收集整理的Go 分布式学习利器（18）-- Go并发编程之lock+WaitGroup实现线程安全的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。