使用 go.uber.org/atomic

Go 的 sync/atomic 包仅仅提供针对原始类型（int32, int64, …）的原子操作。因此，很容易忘记使用原子操作来读写变量。

go.uber.org/atomic 通过隐藏基础类型，使这些操作类型安全。并且，它还提供一个方便的 atomic.Bool 类型。

BadGood
  1. type foo struct {
  2.   running int32  // atomic
  3. }
  5. func (f foo) start() {
  6.   if atomic.SwapInt32(&f.running, 1) == 1 {
  7.      // already running…
  8.      return
  9.   }
  10.   // start the Foo
  11. }
  13. func (f foo) isRunning() bool {
  14.   return f.running == 1  // race!
  15. }
 
  1. type foo struct {
  2.   running atomic.Bool
  3. }
  5. func (f foo) start() {
  6.   if f.running.Swap(true) {
  7.      // already running…
  8.      return
  9.   }
  10.   // start the Foo
  11. }
  13. func (f foo) isRunning() bool {
  14.   return f.running.Load()
  15. }