go sync.once用法

作用

sync.once可以控制函數只能被調用一次。不能多次重複調用。

示例代碼：

package main

import (
    "fmt"
    "sync"
)

func main() {
    o := &sync.Once{}
    var wa sync.WaitGroup
    wa.Add(2)
    go do(o,&wa)
    go do(o,&wa)
    wa.Wait()
}

func do(o *sync.Once, wa *sync.WaitGroup) {
    fmt.Println("Start do")
    o.Do(func() {
        fmt.Println("Doing something...")
    })
    fmt.Println("Do end")
    wa.Done()
}

輸出結果：
Start do
Doing something...
Do end
Start do
Do end

這裏 Doing something 只被調用了一次。

代碼實現

查看go once的源碼實現，也是非常的簡單：

type Once struct {
    m    Mutex
    done uint32
}

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()
    }
}

核心思想是使用原子計數記錄被執行的次數。使用Mutex Lock Unlock鎖定被執行函數，防止被重複執行。

聲明：Nansheng.Su 發表於 2019-05-13 15:51:46 ，共計93字。

轉載請署名：go sync.once用法 | www.sunansheng.com