使用atomic包可高效实现并发安全计数器。通过atomic.AddInt64对int64变量进行原子增减,相比互斥锁更轻量,适用于简单数值操作场景。

在go语言中,atomic包提供了底层的原子操作,适用于实现高效的并发安全计数器。相比互斥锁(mutex),atomic操作更轻量,特别适合简单的数值增减场景。
使用atomic.AddInt64实现并发计数器
通过atomic.AddInt64可以安全地对int64类型的变量进行原子自增或自减。这是实现计数器最常用的方式。
示例代码:
package main
立即学习“go语言免费学习笔记(深入)”;
import (
“fmt”
“sync”
“sync/atomic”
)
func main() {
var counter int64
var wg sync.WaitGroup
numGoroutines := 10
incrementTimes := 100
for i := 0; i
wg.Add(1)
go func() {
defer wg.Done()
for j := 0; j
atomic.AddInt64(&counter, 1)
}
}()
}
wg.Wait()
fmt.printf(“最终计数: %dn”, counter) // 输出 1000
}
读取当前计数值:atomic.LoadInt64
如果需要在不修改的情况下读取计数器当前值,应使用atomic.LoadInt64,保证读操作也是原子的。
例如,在循环中安全读取:
value := atomic.LoadInt64(&counter)
fmt.Printf(“当前计数: %dn”, value)
避免直接读取变量(如fmt.Println(counter)),这可能导致数据竞争。
初始化与重置计数器
可以通过atomic.StoreInt64将计数器重置为指定值:
atomic.StoreInt64(&counter, 0) // 重置为0
初始化时直接声明即可:
var counter int64 // 零值即为0,可直接使用
基本上就这些。只要涉及并发环境下的简单计数,优先考虑atomic比mutex更高效。注意始终使用配套的Load、Store、Add等函数访问变量,不要混合普通读写。这样就能安全又高效地实现计数器功能。


