go sync.once用法

這是一個建立於 的文章，其中的資訊可能已經有所發展或是發生改變。

歡迎關注go語言公眾號 每日go語言 golang_everyday

 

sync.once可以控制函數只能被調用一次。不能多次重複調用。範例程式碼：

 

package main

 

import (

"fmt"

"sync"

"time"

)

 

func main() {

o := &sync.Once{}

go do(o)

go do(o)

time.Sleep(time.Second * 2)

}

 

func do(o *sync.Once) {

fmt.Println("Start do")

o.Do(func() {

fmt.Println("Doing something...")

})

fmt.Println("Do end")

}

 

輸出結果：

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鎖定被執行函數，防止被重複執行。