Go原子計數器實例

Go語言中管理狀態的主要機制是通過通道進行通信。在過去的文章中我們已經看到了這一點，例如工人池。 還有一些其他選項用於管理狀態。 這裏我們將使用sync/atomic包來實現由多個goroutine訪問的原子計數器。

使用一個無符號整數表示計數器(正數)。

爲了模擬併發更新，將啓動50個goroutine，每個增量計數器大約是1毫秒。

爲了原子地遞增計數器，這裏使用AddUint64()函數，在ops計數器的內存地址上使用＆語法。

在增量之間等待一秒，允許一些操作累積。

爲了安全地使用計數器，同時它仍然被其他goroutine更新，通過LoadUint64提取一個當前值的副本到opsFinal。 如上所述，需要將獲取值的內存地址&ops給這個函數。

運行程序顯示執行了大約40,000次操作。

所有的示例代碼，都放在 F:\worksp\golang 目錄下。安裝Go編程環境請參考：http://www.yiibai.com/go/go\_environment.html

atomic-counters.go的完整代碼如下所示 -

package main

import "fmt"
import "time"
import "sync/atomic"

func main() {

    // We'll use an unsigned integer to represent our
    // (always-positive) counter.
    var ops uint64 = 0

    // To simulate concurrent updates, we'll start 50
    // goroutines that each increment the counter about
    // once a millisecond.
    for i := 0; i < 50; i++ {
        go func() {
            for {
                // To atomically increment the counter we
                // use `AddUint64`, giving it the memory
                // address of our `ops` counter with the
                // `&` syntax.
                atomic.AddUint64(&ops, 1)

                // Wait a bit between increments.
                time.Sleep(time.Millisecond)
            }
        }()
    }

    // Wait a second to allow some ops to accumulate.
    time.Sleep(time.Second)

    // In order to safely use the counter while it's still
    // being updated by other goroutines, we extract a
    // copy of the current value into `opsFinal` via
    // `LoadUint64`. As above we need to give this
    // function the memory address `&ops` from which to
    // fetch the value.
    opsFinal := atomic.LoadUint64(&ops)
    fmt.Println("ops:", opsFinal)
}

執行上面代碼，將得到以下輸出結果 -

F:\worksp\golang>go run atomic-counters.go
ops: 41360