1. 程式人生 > 程式設計 >解決golang sync.Wait()不執行的問題

解決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)
  }()

其實就是一個小小的同步問題,旁觀者清呀!!!

以上為個人經驗,希望能給大家一個參考,也希望大家多多支援我們。如有錯誤或未考慮完全的地方,望不吝賜教。