Golang atomic.StoreUint64() 函数及其示例

在 Go 语言中，我们有一些原子操作来管理共享变量，以确保充分利用并发处理。其中一项功能是在无竞争条件下更新一个变量。这可以通过 atomic.StoreUint64() 函数完成。

函数概述

atomic.StoreUint64() 函数将一个无符号 64 位整数类型的值存储到指定的地址，并返回原始值。函数签名如下：

func StoreUint64(addr *uint64, val uint64)

函数接收两个参数。第一个参数 ‘addr’ 是存储无符号 64 位整数的地址指针，第二个参数 ‘val’ 是一个无符号 64 位整数类型的值，该函数将该值存储在指针地址所指向的内存位置。

函数返回先前存储在地址位置的无符号 64 位整数类型的值。这个函数的重要特点是，在没有竞争的情况下并发调用此函数时保证对变量的存储不会出现冲突。

示例代码

下面是示例代码，该代码演示了如何使用 atomic.StoreUint64() 函数更新一个无符号 64 位整数变量的值：

package main

import (
    "fmt"
    "sync/atomic"
)

func main() {
    var counter uint64

    // 将 counter 的初始值设置为 100
    atomic.StoreUint64(&counter, 100)

    // 使用 StoreUint64() 函数存储一个新值 200
    oldVal := atomic.StoreUint64(&counter, 200)
    fmt.Println(oldVal) // 输出: 100

    // 使用 LoadUint64() 函数读取新值
    newVal := atomic.LoadUint64(&counter)
    fmt.Println(newVal) // 输出: 200
}

在这个示例代码中，我们首先创建了一个 ‘counter’ 变量并将其初始化为 100。接下来，我们使用 atomic.StoreUint64() 函数将其更新为 200。在存储新值之后，我们使用 atomic.LoadUint64() 函数读取新值。最后，我们打印出旧值和新值，以便更好地理解函数。

并发使用示例

接下来，我们将提供一个使用 atomic.StoreUint64() 函数进行并发访问的代码示例。在这个示例代码中，我们将使用 10 个并发 goroutines 更新共享计数器。这个计数器将使用 atomic.StoreUint64() 函数来更新值。

package main

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

func main() {
    var counter uint64
    var wg sync.WaitGroup

    for i := 0; i < 10; i++ {
        wg.Add(1)
        go func() {
            for j := 0; j < 1000; j++ {
                atomic.StoreUint64(&counter, counter+1)
            }
            wg.Done()
        }()
    }

    wg.Wait()
    fmt.Println("Counter = ", counter)
}

在这个示例中，我们启动 10 个 goroutines 并使用同步等待组来等待它们的完成。

每个 goroutine 将调用 atomic.StoreUint64() 函数 1000 次，以更新一个共享计数器 ‘counter’。在每次更新过程中，我们使用加法操作进行更新，从而更好地演示如何在不包含争用的情况下进行存储更新操作。

在所有 goroutine 完成后，我们将打印共享计数器的最终值以验证存储操作的正确性。