認識 Go 原子操作

Others
,

簡介

當想要對某個變數並行修改，除了使用 sync 套件的 Mutex 互斥鎖，還可以使用 sync/atomic 套件的原子操作，它能夠保證變數被讀取或修改期間不被其他的執行單位所影響。

範例

以下啟動 50 個 goroutine，每個 goroutine 都幫計數器加一。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
package main

import (
	"fmt"
	"sync"
	"sync/atomic"
)

func main() {
	var couter1 uint64
	var couter2 uint64

	var wg sync.WaitGroup

	for i := 0; i < 50; i++ {
		wg.Add(1)
		go func() {
			for c := 0; c < 1000; c++ {
				couter1 += 1 // 不使用原子操作
				atomic.AddUint64(&couter2, 1) // 使用原子操作
			}
			wg.Done()
		}()
	}
	wg.Wait()

	fmt.Printf("couter1: %d\n", couter1) // 小於 50000
	fmt.Printf("couter2: %d\n", couter2) // 50000
}

參考資料