引言

在Golang中，因為協(xié)程執(zhí)行的順序是不固定的，如果不在代碼里進(jìn)行控制，可能就會導(dǎo)致預(yù)期外的輸出。
本文通過分析一段代碼的執(zhí)行來介紹這種情況，以及可行的控制協(xié)程執(zhí)行順序的方法：

1. sleep()
2. waitGroup

實例分析

代碼

func NewRingBuffer(inCh, outCh chan int) *ringBuffer {return &ringBuffer{inCh:  inCh,outCh: outCh,}
}type ringBuffer struct {inCh  chan intoutCh chan int
}func (r *ringBuffer) Run() {for v := range r.inCh {select {case r.outCh <- v:default:<-r.outCh // pop one item from outchanr.outCh <- v}}close(r.outCh)
}func main() {inCh := make(chan int)outCh := make(chan int, 4)rb := pkg.NewRingBuffer(inCh, outCh)go rb.Run()for i := 0; i < 10; i++ {inCh <- i}close(inCh)//time.Sleep(time.Millisecond * 50)for res := range outCh {fmt.Println(res)}
}

上面代碼的作用是，聲明ringBuffer環(huán)狀緩沖區(qū)；主協(xié)程往inCh寫10個數(shù)的同時，有一個協(xié)程異步的讀取數(shù)據(jù)，并寫到outCh中；
outCh是一個4緩沖區(qū)大小的channel，如果outCh沒滿就直接寫入，否則把outCh的首部元素移除再添加；
因此，預(yù)期輸出應(yīng)該是：6、7、8、9。
但實際的輸出是：5、6、7、8、9。

分析

多出來的“5”，是因為協(xié)程的執(zhí)行順序不可控，當(dāng)主協(xié)程執(zhí)行到

for i := 0; i < 10; i++ {inCh <- i
}close(inCh)

時，此時outCh中是5、6、7、8；run協(xié)程還沒有繼續(xù)執(zhí)行，就開始遍歷outCh：

for res := range outCh {fmt.Println(res)
}

然后輸出阻塞后，run協(xié)程才繼續(xù)執(zhí)行，把9寫到outCh中，因此最后的輸出結(jié)果是5、6、7、8、9.

解決辦法

sleep()

使用sleep()函數(shù)，能讓當(dāng)前協(xié)程讓出CPU，暫停執(zhí)行一段時間。
對于這種方法，可以對上面函數(shù)進(jìn)行如下改造：

close(inCh)
time.Sleep(time.Millisecond * 50)
// 新增
for res := range outCh {fmt.Println(res)
}

WaitGroup

使用waitGroup可以實現(xiàn)協(xié)程間通信，對于這段例子，通過wg保證run()函數(shù)執(zhí)行完后，在對outCh進(jìn)行輸出。
改造方法如下：

func (r *ringBuffer) RunWithWg(wg *sync.WaitGroup) {defer wg.Done()for v := range r.inCh {select {case r.outCh <- v:default:<-r.outCh // pop one item from outchanr.outCh <- v}}close(r.outCh)
}func TestRaceWithWg(t *testing.T) {inCh := make(chan int)outCh := make(chan int, 4)rb := pkg.NewRingBuffer(inCh, outCh)var wg sync.WaitGroupwg.Add(1)go rb.RunWithWg(&wg)go func() {for i := 0; i < 10; i++ {inCh <- i}close(inCh)}()wg.Wait()for res := range outCh {fmt.Println(res)}
}