在go语言中实现原子操作示例代码

Go  /  管理员 发布于 10个月前   213

在go语言中实现原子操作非常便捷直接调用sync/atomic包即可。

1.先来看一个错误的示例

通过启动 1000 个 goroutine 来模拟并发调用，在函数内部对变量 number 进行自增操作，

 那么可能存在的一个问题是，当多个 goroutine 同时对变量操作时，

只有一个成功了，其他的全部失败，

造成的后果就是变量 最终的值小于 1000 (正常情况应该是等于 1000)。

package main

import (
    "fmt"
    "sync"
)

func main() {
    var number uint32
    var wg sync.WaitGroup
    
    for i := 0; i < 1000; i++ {
        wg.Add(1)
        go func() {
            defer func() {
                wg.Done()
            }()
            number++
        }()
    }
    
    wg.Wait()
    fmt.Printf("number = %d\n", number)
}

// $ go run main.go
// 输出如下，你的输出可能和这里的不一样，多试几次，会发现每次的结果都不一样
/**
  number = 971
*/

2.正确的并发操作

package main

import (
    "fmt"
    "sync"
    "sync/atomic"
)

func main() {
    var number uint32
    var wg sync.WaitGroup
    
    for i := 0; i < 1000; i++ {
        wg.Add(1)
        go func() {
            defer func() {
                wg.Done()
            }()
            //number++
            atomic.AddUint32(&number, 1) // 使用原子操作
        }()
    }
    
    wg.Wait()
    fmt.Printf("number = %d\n", number)
}

// $ go run main.go
// 输出如下，多试几次，会发现结果都是一样的
/**
  number = 1000
*/