Go语言文档解析：sync.Once函数实现单次执行-Golang-PHP中文网

Go语言文档解析：sync.Once函数实现单次执行

王林

发布： 2023-11-03 08:28:50

原创

2112 人浏览过

Go语言文档解析：sync.Once函数实现单次执行

Go语言文档解析：sync.Once函数实现单次执行，需要具体代码示例

Go语言中的sync包提供了一些用于同步操作的函数和类型。其中一个非常有用的函数是sync.Once，它可以确保某个操作只执行一次。在本文中，我们将详细解析sync.Once函数的使用，并提供一些具体的代码示例。

sync.Once函数的定义如下：

type Once struct {
    m    Mutex
    done uint32
}

func (o *Once) Do(f func()) {
    if atomic.LoadUint32(&o.done) == 1 {
        return
    }
    o.m.Lock()
    defer o.m.Unlock()
    if o.done == 0 {
        f()
        atomic.StoreUint32(&o.done, 1)
    }
}

登录后复制

可以看到，sync.Once结构体中包含了一个互斥锁（Mutex）和一个done标志，用于记录操作是否已经执行。Once结构体的Do方法是实现单次执行的核心逻辑。

Do方法首先通过原子操作atomic.LoadUint32，检查done标志是否为1。如果是1，表示操作已经执行过，直接返回。否则，获取互斥锁，再次检查done标志是否为0。如果是0，则执行传入的函数f，并通过原子操作atomic.StoreUint32将done标志设为1，确保下次调用时不会再执行f。

下面是一个简单的示例，演示了如何使用sync.Once函数实现单次执行：

package main

import (
    "fmt"
    "sync"
)

var once sync.Once

func main() {
    for i := 0; i < 5; i++ {
        // 只有第一次调用会执行
        once.Do(func() {
            fmt.Println("This will only print once.")
        })
    }
}

登录后复制

运行上述代码，输出结果如下：

This will only print once.

登录后复制

可以看到，尽管循环中调用了多次once.Do方法，但实际上只有第一次调用会执行传入的函数，后续的调用都直接返回，没有再次执行。

sync.Once函数的使用场景非常广泛。比如，在初始化某个全局变量时，我们通常希望只执行初始化一次，而不是每次访问变量都进行初始化。此时，就可以使用sync.Once函数来确保初始化只执行一次。

var (
    data  []string
    once  sync.Once
)

func loadData() {
    // 模拟耗时的数据加载操作
    // 这里简单起见直接赋值
    data = []string{"Hello", "World"}
}

func getData() []string {
    once.Do(loadData)
    return data
}

func main() {
    fmt.Println(getData())
    fmt.Println(getData())
}

登录后复制

运行上述代码，输出结果如下：