Golang中如何使用sync.Once确保初始化代码只执行一次

P粉602998670

发布时间：2025-09-09 09:57:01

402人浏览过

来源于php中文网

原创

使用sync.Once可确保初始化代码在并发环境下仅执行一次。它通过Do方法实现线程安全的懒加载，适用于数据库连接池等需单次初始化的场景；与init()函数不同，sync.Once支持运行时延迟初始化，且天生应对并发，但需注意初始化函数panic会导致永久失败，且无法重试或重置。

golang中如何使用sync.once确保初始化代码只执行一次

在Golang中，要确保一段初始化代码无论在何种并发环境下都只执行一次，最地道且推荐的做法是使用标准库中的

sync.Once

类型。它提供了一个简洁、高效且线程安全的机制，完美解决了单例模式或资源懒加载的初始化问题。

解决方案

sync.Once

的核心思想非常直接：它有一个

Do

方法，接收一个无参数的函数。无论

Do

方法被调用多少次，传入的函数只会被执行一次。这对于那些需要全局唯一实例、或者某个复杂资源只需初始化一次的场景来说，简直是量身定制。

我们来看一个典型的例子，比如我们想确保一个数据库连接池只被初始化一次：

立即学习“go语言免费学习笔记（深入）”；

package main

import (
    "fmt"
    "sync"
    "time"
)

// DBManager 模拟一个数据库连接管理器
type DBManager struct {
    conn string
}

var (
    once    sync.Once
    dbMgr   *DBManager
)

// InitDBManager 模拟数据库连接的初始化过程
func InitDBManager() *DBManager {
    // 模拟耗时操作
    time.Sleep(time.Millisecond * 100)
    fmt.Println("正在初始化数据库连接...")
    dbMgr = &DBManager{conn: "PostgreSQL Connection Pool"}
    return dbMgr
}

// GetDBManager 获取数据库管理器实例
func GetDBManager() *DBManager {
    once.Do(func() {
        dbMgr = InitDBManager()
    })
    return dbMgr
}

func main() {
    var wg sync.WaitGroup
    for i := 0; i < 5; i++ {
        wg.Add(1)
        go func(id int) {
            defer wg.Done()
            fmt.Printf("协程 %d 尝试获取DB管理器...\n", id)
            mgr := GetDBManager()
            fmt.Printf("协程 %d 获取到DB管理器: %s\n", id, mgr.conn)
        }(i)
    }
    wg.Wait()
    fmt.Println("所有协程完成。")
}

在这个例子中，

InitDBManager

函数代表了我们实际的初始化逻辑。

GetDBManager

函数内部通过

once.Do(func() { dbMgr = InitDBManager() })

确保了

InitDBManager

只会被调用一次。即使有多个goroutine同时调用

GetDBManager

，

sync.Once

也会保证只有一个goroutine能成功执行初始化函数，其他goroutine会阻塞直到初始化完成并拿到结果。这种模式，在我看来，是处理懒加载单例最优雅的方式之一。

为什么在并发环境中

sync.Once

是不可或缺的？

在并发编程的世界里，资源初始化往往是个棘手的问题。如果没有

sync.Once

这样的机制，我们可能会面临几种常见的困境。

想象一下，你有一个全局配置对象或者一个日志句柄，它只需要被创建一次。如果多个goroutine几乎同时尝试创建它，会发生什么？

竞态条件（Race Condition）: 多个goroutine可能都会检查到“这个资源还没被创建”，然后各自尝试去创建。这可能导致资源被创建多次，或者更糟的是，导致部分创建失败，或者数据损坏。比如，你可能打开了多个文件句柄，或者初始化了多个数据库连接池，这显然不是我们想要的。
重复初始化开销: 即使重复初始化不会导致程序崩溃，它也带来了不必要的性能开销。初始化操作往往是耗时且占用资源的，比如连接数据库、加载配置文件到内存等。重复执行这些操作，无疑是浪费。
非预期的状态: 如果一个资源在初始化过程中依赖于某些外部状态，而这个外部状态又被多个初始化过程并发修改，那最终资源的状态就可能变得不可预测。

sync.Once

正是为了解决这些问题而生。它内部通过原子操作和互斥锁的巧妙结合，确保了

Do

方法传入的函数只会被执行一次，且这个执行过程是线程安全的。它不仅仅是保证“一次”，更是保证“安全的一次”。在我看来，这不仅仅是技术实现，更是一种编程哲学的体现：化繁为简，将并发的复杂性封装起来，给开发者一个清晰、可靠的接口。相比于手动使用

sync.Mutex

来管理一个

isInitialized

布尔标志，

sync.Once

的代码更简洁，也更不容易出错，因为它替你处理了所有边缘情况。

使用

sync.Once

时有哪些常见的陷阱和注意事项？

虽然

sync.Once

用起来非常方便，但它也不是万能的，或者说，在使用时有一些特定的行为需要我们理解，否则可能会遇到一些“惊喜”。

一个最常被提及的“陷阱”是：如果

Do

方法中传入的初始化函数发生了panic，
sync.Once
会将其视为初始化已完成，并且不会在后续的调用中重试。这意味着，如果你的初始化逻辑有bug导致panic，或者依赖的外部服务不可用导致panic，那么你的应用程序将永远无法正确地初始化这个资源，后续所有尝试获取该资源的地方都会拿到一个未初始化的（或者说，panic时留下的）状态，这通常会导致更深层次的错误。

STORYD

帮你写出让领导满意的精美文稿

下载

举个例子：

var brokenOnce sync.Once
var brokenResource string

func initBrokenResource() {
    fmt.Println("尝试初始化一个会panic的资源...")
    if true { // 模拟一个总是会panic的条件
        panic("初始化失败：模拟一个严重错误")
    }
    brokenResource = "我本应该被初始化"
}

func GetBrokenResource() string {
    brokenOnce.Do(initBrokenResource)
    return brokenResource
}

func main() {
    // 第一次调用，会panic
    // defer func() {
    //  if r := recover(); r != nil {
    //      fmt.Println("捕获到panic:", r)
    //  }
    // }()
    // fmt.Println(GetBrokenResource()) // 这一行会panic

    // 如果不捕获panic，程序会崩溃。
    // 如果捕获了，那么第二次调用GetBrokenResource()，initBrokenResource()不会再执行
    // brokenResource 仍然是空字符串
}

为了避免这种情况，我的建议是：初始化函数内部应该自行处理所有可能的错误，而不是让它panic。 如果初始化失败，应该返回一个错误，并在

Do

函数内部处理这个错误，比如记录日志，或者设置一个错误状态，而不是直接panic。

另一个需要注意的点是，

sync.Once

不提供任何方式来“重置”。一旦

Do

方法中的函数成功执行了一次，它就永远不会再执行了。这对于真正的单例模式是好事，但如果你需要一个资源在某些条件下可以被重新初始化（比如配置热加载，或者测试场景下需要重置状态），那么

sync.Once

就不适用了。在这种情况下，你需要自己实现一个带有互斥锁和条件判断的初始化逻辑。

最后，虽然不是陷阱，但值得一提的是，

sync.Once

的初始化函数不接受参数，也不返回任何值。这意味着，如果你需要将初始化结果传递出去，或者初始化需要外部参数，你通常需要通过闭包或者全局变量来处理。上面的

dbMgr

例子就是通过全局变量来传递结果的。这在大多数情况下是可行的，但偶尔也会让人觉得不够灵活。

sync.Once

与

init()

函数有什么区别和适用场景？

sync.Once

和Go语言的

init()

函数都能实现“只执行一次”的效果，但它们的设计哲学、执行时机以及适用场景有着本质的区别。理解这些差异，能帮助我们更好地选择合适的工具。

init()

函数

执行时机:
```
init()
```
函数在包被导入（import）时，或者说在程序启动时，
```
main
```
函数执行之前自动执行。每个包可以有多个
```
init()
```
函数，它们会按照文件名的字典序以及文件内声明的顺序依次执行。
并发性:
```
init()
```
函数是严格串行执行的。Go运行时保证了所有
```
init()
```
函数在单线程环境下完成，不存在并发问题。
用途: 通常用于包级别的初始化，比如注册驱动、设置全局配置的默认值、检查程序启动环境等。它适合那些在程序生命周期早期就必须完成，且不依赖于具体业务逻辑调用的初始化任务。

sync.Once

执行时机:
```
sync.Once
```
的
```
Do
```
方法传入的函数，是在你第一次显式调用
```
Do
```
方法时才执行。这是一种“懒加载”（Lazy Loading）机制。
并发性:
```
sync.Once
```
专为并发环境设计。它保证了即使多个goroutine同时调用
```
Do
```
方法，初始化函数也只会被执行一次，且这个过程是线程安全的。
用途: 适用于那些需要延迟初始化、或者在并发环境下确保某个资源（如数据库连接、单例对象）只被创建一次的场景。它与
```
init()
```
函数最大的不同在于，它允许你在程序运行的任何阶段，根据实际需要才触发初始化。

核心区别总结：

主动 vs. 被动:
```
init()
```
是“主动”的，程序启动就执行；
```
sync.Once
```
是“被动”的，只有在第一次调用
```
Do
```
时才执行。
包级别 vs. 实例/资源级别:
```
init()
```
通常作用于整个包；
```
sync.Once
```
更常用于初始化某个特定的实例或资源。
并发处理:
```
init()
```
天然串行，不需要额外处理并发；
```
sync.Once
```
天生就是为解决并发初始化问题而设计的。

适用场景选择：

如果你有一个全局性的、程序启动时就必须准备好的配置或服务注册，且它与具体的业务逻辑调用无关，那么
```
init()
```
函数是更简洁、直接的选择。
如果你有一个资源，它可能在程序的整个生命周期中都不会被用到，或者只有在某个特定条件（比如第一次用户请求）下才需要被创建，并且需要保证并发安全，那么
```
sync.Once
```
就是你的不二之选。它能有效节省启动时间，并在真正需要时才分配资源。

在我看来，