快捷導(dǎo)航

Golang基于sync.Once實(shí)現(xiàn)單例的操作代碼

更新時(shí)間：2022年10月28日 11:14:17 作者：Hugo Lei

這篇文章主要介紹了golang實(shí)現(xiàn)單例的操作代碼,本文介紹基于sync.Once的方式來(lái)實(shí)現(xiàn)單例，熟練掌握這種模式，并理解其底層原理，對(duì)大部分人來(lái)講已經(jīng)完全夠用了，需要的朋友可以參考下

基于sync.Once實(shí)現(xiàn)單例

// 其他package也可見，在其他地方也可以new新對(duì)象
// 但是最終調(diào)用Conn()方法時(shí)，都是用的single這個(gè)單例
// 1
type Driver struct {
	// 小寫字母開頭，外部不可訪問(wèn)，所以new個(gè)Driver新對(duì)象也沒用
	// 2
	conn string
}

// 全局變量，指針默認(rèn)值為nil
// 3
var single *Driver // 單例
var once sync.Once

// 對(duì)外暴露的公共方法
func (s *Driver) Conn() {
	fmt.Printf("conn=%s", single.conn) // do something
}

// 4
func GetDriverSingleton() *Driver {
	// 對(duì)GetDriverSingleton()方法的調(diào)用，都會(huì)執(zhí)行once.Do()方法，只不過(guò)參數(shù)func()只會(huì)被執(zhí)行一次
	// 若并發(fā)執(zhí)行once.Do()，多個(gè)協(xié)程會(huì)阻塞，因內(nèi)部是通過(guò)Mutex來(lái)控制
	once.Do(func() {
		single = new(Driver)
		single.conn = "single conn"
		time.Sleep(50 * time.Millisecond)
	})
	return single
}

單例類型定義Driver

Driver類的方法要支持跨包訪問(wèn)，因此需要以大寫字母開頭。
小寫字母開頭，作用域僅限于包內(nèi)部。

類Field conn

類變量conn需要小寫字母開頭，跨包不可訪問(wèn)，避免在包外被修改。

但是包內(nèi)還是有可能被修改。

once.Do(func() {})

每次調(diào)用GetDriverSingleton()，都會(huì)調(diào)用once.Do()方法，但是在once.Do()方法內(nèi)部，僅會(huì)執(zhí)行一次參數(shù)func(){}，因此就保證了單例唯一初始化。

并發(fā)訪問(wèn)once.Do()

不會(huì)有并發(fā)訪問(wèn)問(wèn)題，因once.Do()內(nèi)部通過(guò)mutex來(lái)控制。

// once.DO()
if atomic.LoadUint32(&o.done) == 0 {
	// Outlined slow-path to allow inlining of the fast-path.
	o.doSlow(f)
}

// doSlow()
func (o *Once) doSlow(f func()) {
	o.m.Lock() // 互斥鎖
	defer o.m.Unlock()
	if o.done == 0 {
		defer atomic.StoreUint32(&o.done, 1)
		f()
	}
}