发布于2026-07-18 阅读(0)
扫一扫,手机访问
在并发编程里,计数器是个很常见的需求。但实现起来,坑也不少。先说核心结论:Go 1.19 及以上版本,优先上 sync.AtomicInt64。它底层走的 CPU 原子指令,无锁、高性能、语义清晰,死锁风险基本为零。Go 1.18 及更早版本,就得手动封装 atomic.AddInt64 和 LoadInt64 了,或者干脆升级 Go 版本。如果直接用 int64 自增,那非原子操作在并发下必然丢数据,这是个硬伤。

sync.AtomicInt64 还是 sync.Mutex?高并发场景下,直接用 int64 自增,数据会丢得你心疼。用 sync.Mutex 也能行,但锁的开销摆在那里。Go 1.19 以上,推荐优先用 sync.AtomicInt64,无锁的 CPU 原子指令,性能好、语义清晰,而且不会死锁。至于 Go 1.18 及之前版本,没有现成的 sync.AtomicInt64,得手动封装 atomic.AddInt64 等函数,或者考虑升级 Go 版本。
常见错误现象:counter++ 在 goroutine 中并发执行后,结果远小于预期。比如启了 100 个 goroutine 各加 100 次,最终只到 2000 多——这就是典型的竞态未防护。
atomic.LoadInt64 和 atomic.AddInt64,千万别混着普通变量赋值,否则还是会有竞态问题。Load 再判断再 Add,这已经不是原子“比较并交换”了。atomic.CompareAndSwapInt64。当单个计数器不够用,需要按不同维度统计时,比如按 HTTP 方法和状态码计数,就别硬套全局变量了。推荐用 map 配合读写锁。但要注意,map 本身不是线程安全的,即使加了锁,也必须保护所有操作入口,不能有遗漏。
更稳妥的做法是用 sync.Map,它适合读多写少的场景,或者自己封装一个带锁的结构体。如果维度固定,比如只有 method 和 status,那用嵌套 map 配合 sync.RWMutex 会更易调试,内存也更紧凑。
示例场景:记录 GET 200、POST 500 的调用次数:
type StatusCounter struct {
mu sync.RWMutex
m map[string]map[int64]int64 // method → status → count
}
func (c *StatusCounter) Inc(method string, status int64) {
c.mu.Lock()
if c.m == nil {
c.m = make(map[string]map[int64]int64)
}
if c.m[method] == nil {
c.m[method] = make(map[int64]int64)
}
c.m[method][status]++
c.mu.Unlock()
}
c.m 必须在 Lock 内,否则并发写 nil map 会 panic。sync.Map 减少锁粒度,但它的 Range 不保证一致性,遍历时可能漏项。直接暴露 sync.AtomicInt64 的值给 Prometheus 是不行的,需要适配成 prometheus.Counter 或 Gauge 类型。最省事的办法是用 prometheus.NewGaugeFunc 包一层:
var reqTotal = &atomic.Int64{}
reqCounter := prometheus.NewGaugeFunc(prometheus.GaugeOpts{
Name: "http_requests_total",
Help: "Total number of HTTP requests.",
}, func() float64 {
return float64(reqTotal.Load())
})
prometheus.MustRegister(reqCounter)
注意,Counter 类型要求单调递增,而 GaugeFunc 可以任意变化,所以适合带重置的计数器。如果业务需要 reset,就别用 prometheus.Counter 了,因为它不支持减操作。
reqTotal 已初始化,虽然 atomic.Int64 默认零值安全,但习惯上还是显式 Store(0) 一下。GaugeFunc 回调里做耗时操作,比如查 DB 或 HTTP 请求,否则会阻塞 Prometheus 抓取。GaugeFunc,别挤在一个回调里计算。典型原因往往是误用了粗粒度锁,比如整个计数器结构共用一把 sync.Mutex,导致高并发下 goroutine 大量阻塞排队。另一个隐蔽问题是缓存行伪共享:多个 atomic.Int64 实例紧挨着定义,被 CPU 放进同一个 cache line,频繁修改会引发 core 间的缓存同步风暴。
怎么验证?用 go tool trace 看看 goroutine 阻塞在 mutex 上的时间,或者用 perf stat -e cache-misses 观察缓存失效率是否异常高。
count atomic.Int64; _ [64]byte。[]byte 混排。真正难调的,从来不是怎么写出来,而是怎么让它在 10k QPS 下还准、还快、还不悄悄吃掉一半 CPU。这些细节,不打日志、不跑压测,根本看不到。