go学习笔记——原子计数器

原子计数器

使用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 已完成。
暂无评论

发送评论 编辑评论


				
上一篇
下一篇