golang中sync.Once的示例分析

這篇文章將為大家詳細講解有關golang中sync.Once的示例分析，小編覺得挺實用的，因此分享給大家做個參考，希望大家閱讀完這篇文章后可以有所收獲。

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

我們可以用下面的代碼實現一個線程安全的單例模式

package singleton

import (
    "fmt"
    "sync"
)

type object struct {
    name string
}

var once sync.Once
var obj *object //單例指針

//公開方法 外包調用
func Instance() *object {
    once.Do(getObj)
    return obj
}

func getObj() {
    if obj == nil {
        obj = new(object)
        //可以做其他初始化事件
    }
}

//單例測試
func (obj *object) Test() {
    fmt.Println(obj.name)
}

如果我們要自己實現這么一個功能如何做呢？

1. 定義一個status變量用來描述是否已經執行過了

2. 使用sync.Mutex 或者sync.Atomic實現線程安全的獲取status狀態， 根據狀態判斷是否執行特定的函數

然后看下sync.Once實際是如何實現的

// Once is an object that will perform exactly one action.
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()
    //這里需要再次重新判斷下，因為 atomic.LoadUint32取出狀態值到  o.m.Lock() 之間是有可能存在其它gotoutine改變status的狀態值的
    if o.done == 0 {
        f()
        atomic.StoreUint32(&o.done, 1)
    }
}

也有網友寫出了更簡潔的代碼,不知道官方為什么沒有采用下面的實現方式。

type Once struct {
    done int32
}

func (o *Once) Do(f func()) {
    if atomic.LoadInt32(&o.done) == 1 {
        return
    }
    // Slow-path.
    if atomic.CompareAndSwapInt32(&o.done, 0, 1) {
        f()
    }
}

關于“golang中sync.Once的示例分析”這篇文章就分享到這里了，希望以上內容可以對大家有一定的幫助，使各位可以學到更多知識，如果覺得文章不錯，請把它分享出去讓更多的人看到。