这篇文章主要介绍“GoLang中的sync包Once如何使用”,在日常操作中,相信很多人在GoLang中的sync包Once如何使用问题上存在疑惑,小编查阅了各式资料,整理出简单好用的操作方法,希望对大家解答”GoLang中的sync包Once如何使用”的疑惑有所帮助!接下来,请跟着小编一起来学习吧!
One简介
Once 包主要用于在并发执行代码的时候,某部分代码只会被执行 一次。
Once 的使用也非常简单,Once 只有一个 Do 方法,接收一个无参数无返回值的函数类型的参数 f,不管调用多少次 Do 方法,参数 f 只在第一次调用 Do 方法时执行。
示例
我们有一个Msg 参数,多个协程都会用到他,但是这个参数只用初始化一次就可以。
package main import ( "fmt" "sync" "time" ) var msg string func main() { var one sync.Once for i := 0; i < 5; i++ { go func(i int) { one.Do(func() { fmt.Printf("%d 执行初始化! ", i) msg = "Your Need Data" }) fmt.Println(msg) }(i) } time.Sleep(3* time.Second) }
执行结果如下:
可以看到初始化的代码只被4号线程执行了一次, 其他协程都是直接读的初始化的数据,并没有执行初始化的函数。
注意
不要在
Do()方法的参数方法中再次调用
Do()方法,因为执行这个
Do()方法的参数方法的时候,One 会持有一个锁,如果再参数方法中再次调用
Do()方法,就会等待这个锁释放, 导致参数方法无法执行完毕,然后外层的Do 方法就一直无法释放锁,最后就成了死锁。
错误示例:
package main import ( "fmt" "sync" ) var msg string var one sync.Once func main() { one.Do(fun1) } func fun1(){ fmt.Println("我是 fun1") one.Do(fun2) } func fun2(){ fmt.Println("我是 fun2") }
执行结果:
可以知道再
fun1()中使用
Do()方法调用
fun2的时候形成了死锁, 因为在
fun1()执行过程中已将持有了该锁,需要
fun1()执行完毕才会释放,然后因为使用
Do()方法执行
fun2()也会请求这个锁, 会一直等待,导致
fun1()不可能执行完, 也不可能释放锁。成了死锁。
源码解读
查看源码
func (o *Once) Do(f func()) { if atomic.LoadUint32(&o.done) == 0 { // Outlined slow-path to allow inlining of the fast-path. o.doSlow(f) } } func (o *Once) doSlow(f func()) { o.m.Lock() defer o.m.Unlock() if o.done == 0 { defer atomic.StoreUint32(&o.done, 1) f() } }
使用一个原子类作为标识,加锁校验和操作原子类,保证只会被一个协程执行。
Do调用了
doSlow, 在
doSlow中有
defer关键字,表示执行函数和释放锁是倒序执行,必须先执行完毕
if判断和里面的
f()才能释放锁。