选择atomic而非Mutex因atomic由CPU指令支持,性能更高,适用于基本类型原子操作。示例中用atomic.AddInt64实现并发安全计数器,最终计数正确并可原子重置,避免了锁开销与数据竞争。

在Go语言中,atomic 包提供了底层的原子操作,适用于轻量级、高性能的并发场景。当我们需要实现一个并发安全的计数器时,使用 sync/atomic 比加锁(如 sync.Mutex)更高效,尤其是在读写频繁但逻辑简单的场景下。
为什么选择 atomic 而不是 Mutex?
在高并发环境下,多个goroutine同时修改同一个变量会导致数据竞争。传统做法是使用互斥锁保护共享变量:
- Mutex 加锁解锁有开销,尤其在争用激烈时性能下降明显。- atomic 操作直接由CPU指令支持,执行更快,适合简单操作如增减、赋值、比较交换等。
- atomic 适用于基本类型(int32, int64, uint32, uint64, pointer等)的原子操作。
atomic 常见操作函数说明
Go 的 sync/atomic 提供了多种原子操作函数,常用的包括:
- atomic.AddInt64(&counter, 1):对 int64 类型变量原子增加。
- atomic.LoadInt64(&counter):原子读取当前值。
- atomic.StoreInt64(&counter, newVal):原子写入新值。
- atomic.CompareAndSwapInt64(&counter, old, new):CAS操作,仅当当前值等于old时才更新为new。
实战:并发安全的计数器实现
下面是一个使用 atomic 实现的高并发计数器示例:
立即学习“go语言免费学习笔记(深入)”;
package mainimport ( "fmt" "sync" "sync/atomic" "time" )
func main() { var counter int64 // 使用 int64 存储计数
var wg sync.WaitGroup numGoroutines := 10 incrementsPerRoutine := 1000 // 启动多个goroutine并发增加计数 for i := 0; i < numGoroutines; i++ { wg.Add(1) go func() { defer wg.Done() for j := 0; j < incrementsPerRoutine; j++ { atomic.AddInt64(&counter, 1) // 原子自增 } }() } // 等待所有goroutine完成 wg.Wait() // 安全读取最终值 finalCount := atomic.LoadInt64(&counter) fmt.Printf("最终计数值: %d\n", finalCount) // 验证是否正确 expected := int64(numGoroutines * incrementsPerRoutine) if finalCount == expected { fmt.Println("✅ 计数正确") } else { fmt.Printf("❌ 计数错误,期望 %d,实际 %d\n", expected, finalCount) } // 演示原子写入 time.Sleep(time.Millisecond * 10) atomic.StoreInt64(&counter, 0) fmt.Printf("重置后计数值: %d\n", atomic.LoadInt64(&counter))}
输出结果:
最终计数值: 10000 ✅ 计数正确 重置后计数值: 0注意事项与最佳实践
虽然
atomic高效,但在使用时仍需注意以下几点:
- 必须确保参与原子操作的变量地址不变,且不能跨包随意暴露。
- 只适用于基本类型的原子操作,复杂结构仍需使用
sync.Mutex。 - 变量应声明为
int64并对其取地址操作,避免内存对齐问题(尤其是32位系统上 int64 原子操作要求对齐)。 - 不要混合使用非原子操作和 atomic 操作访问同一变量,否则会引发 data race。
可以使用 go run -race 来检测数据竞争问题:
go run -race main.go
基本上就这些。atomic 是构建高性能并发程序的重要工具之一,合理使用能让代码既安全又高效。对于简单的计数、状态标记、标志位等场景,优先考虑 atomic 而不是锁。不复杂但容易忽略细节,掌握它对提升Go编程能力很有帮助。










