sync.Once 源码分析

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

// Copyright 2009 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package sync

import (
    "sync/atomic"
)

// Once 是一个会执行一个动作的对象。
type Once struct {
    m    Mutex
    done uint32
}

// 当且仅当第一次为这个 Once 实例调用 Do 时，才调用函数 f。
// 换句话说，定义一个 var once Once，如果 once.Do(f) 被调用多次，只有第一次会调用 f，即使 f 在每次调用中都有不同的值。
// 为了执行每个函数，需要新的Once实例。
// Do 是用于初始化的，只能运行一次. 由于 f 是 niladic，因此需要使用函数文字来捕获由 Do 调用的函数的参数：
//  config.once.Do(func() { config.init(filename) })
// 因为没有给 Do 的调用返回，直到对 f 的一次调用返回，如果 f 导致 Do 被调用，它将会死锁。
// 如果 f panic，Do 认为它已经返回了；未来调用 Do 将不再调用 f 。
//
func (o *Once) Do(f func()) {
    if atomic.LoadUint32(&o.done) == 1 {
        return
    }
    // Slow-path.
    o.m.Lock()
    defer o.m.Unlock()
    if o.done == 0 {
        defer atomic.StoreUint32(&o.done, 1)
        f()
    }
}