解決golang sync.Wait()不執行的問題
goroutine 似乎不用解釋太多,可以利用它實現多執行緒,也可以利用它來實現非同步事件。
在使用關鍵字go的過程中,常常會將用到sync.WaitGroup,如下一段程式碼。
package main import ( "fmt" "sync" "time" ) func Run() { var wg = &sync.WaitGroup{} go func() { wg.Add(1) fmt.Println("halo world start") time.Sleep(time.Second * 5) fmt.Println("halo world end") wg.Done() }() // time.Sleep(time.Millisecond * 5) // fmt.Println("server will start") wg.Wait() } func main() { Run() } // output: //
期待的結果是列印 halo world start,5秒後列印halo world end,但是結果就是什麼都沒有,並且程序立即就結束了。
原因
關鍵字go是非同步的,當執行到go,不會立即執行go 後面的內容,而且繼續往下執行。此時wg.Add(1)還沒有來得及執行,wg.Wait()就已經執行,即不會發生等待,程序就結束了。
怎麼解決:
只需要在wg.Wait()前有其他操作,給與足夠的時間讓wg.Add(1)執行即可,
方法一、時間等待,在wg.Wait()前加一句time.Sleep(time.Millisecond*5),既不影響效能,也能讓wg.Add(1)來得及執行
方法二、有IO操作,在wg.Wait()有其他IO操作,比如fmt.Println("server will start"),原因是std的輸出會將程序從使用者態轉向核心態,列印命令發出後,又切回用戶態,這個狀態的轉換是很有消耗的,wg.Add(1)也就有時間執行。
Don't worry
是否有存在擔心,方法一的時間等待,等待的時候不夠長,還是讓wg.Add(1)來不及執行。don't worry.
這裡涉及到goroutine的排程問題,go程序在執行過程中,必須從goroutine佇列中取出一個來執行,當wg.Wait()執行前就算執行time.Sleep(time.Nanosecond),一納秒,一…一…一納秒,wg.Add(1)也來得及執行,因為主goroutine會被切換到睡眠狀態,go程序必須要取一個執行緒來執行,就會取到wg.Add(1)這個執行緒,接下來就順理成章了。
同時方法二也是異曲同工,當發出列印的事件,整個程序都會被切換到就緒態,然後再被cpu執行。
補充:【golang】sync.WaitGroup{}的wait()呼叫位置不同導致意想不到錯誤
協程go多了,總覺的天下我有,沒事就喜歡go一個協程,信手拈來,在專案中寫個如下類似程式碼:
wh := sync.WaitGroup{} out := make(chan string) go func() { wh.Wait() close(out) }() go func() { for i := 0; i < 2; i++ { wh.Add(1) go tt(out) wh.Done() } }()
想著開個協程去wait所有協程組,測試一下通了,沒問題,好牛逼,協程呀!!
可多測試即便就會出現:
send close channel
或者協程定死在某一個,還自已為是的認為自己寫的子方法估計不小心關閉了channel,找了半天只找到在wg.wait()後進行了關閉。就這樣扣了好久,還沒想到自己畫蛇添足的錯誤,經大佬一指點,原來開一個協程,還沒等後一個協程進行wg.add(1)操作,wg.wait()就已經過了,關閉了channel。
只好老老實實寫:
wh := sync.WaitGroup{} out := make(chan string) go func() { for i := 0; i < 2; i++ { wh.Add(1) go tt(out) wh.Done() } wh.Wait() close(out) }()
其實就是一個小小的同步問題,旁觀者清呀!!!
以上為個人經驗,希望能給大家一個參考,也希望大家多多支援我們。如有錯誤或未考慮完全的地方,望不吝賜教。