原子计数器

使用WaitGroup+sync/atomic可以实现一个原子计数器

package main

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

func main() {
    var ops uint64

    var wg sync.WaitGroup

    for i := 0; i < 50; i++ {
        wg.Add(1)

        go func() {
            for c := 0; c < 100; c++ {
                atomic.AddUint64(&ops, 1)
            }
            wg.Done()
        }()
    }

    wg.Wait()
    fmt.Println("ops:", ops)
}

通过多个 goroutine 并发地对一个共享变量进行操作，同时确保操作的正确性和程序的同步。

sync.WaitGroup 用于等待一组 goroutine 完成。它提供了一种简单的方式来确保主程序在所有 goroutine 执行完毕后再继续。

• 循环启动了 50 个 goroutine，每个 goroutine 会执行 100 次计数操作。
• 在启动每个 goroutine 之前，调用 wg.Add(1) 将 WaitGroup 的计数器加 1，表示有一个新的 goroutine 需要等待。
• 每个 goroutine 使用 atomic.AddUint64(&ops, 1) 对 ops 变量进行原子递增操作。atomic.AddUint64 确保了即使多个 goroutine 同时对 ops 进行操作，也不会出现数据竞争。
• 在完成所有计数操作后，调用 wg.Done() 将 WaitGroup 的计数器减 1，表示当前 goroutine 已完成。