Golang中超簡單的協程同步工具

Golang中超簡單的協程同步工具

image

從需求出發

現在有如下需求：多協程處理一批資料，但需要所有協程都跑完才可以繼續下一步，這個需求在日常的處理資料中非常常見。多協程可以最大限度的壓榨CPU和IO，但從業務角度來看，又需要保持一致性，這個時候我們就需要協程同步技術。

流程圖解

從可以看到在3個協程啟動並執行時間都不可控，但為了達到同步的目的，需要等待3個攜程全部跑完再繼續下面的流程。

image

解

Golang提供了使用非常方便的包sync.waitgroup ,這個包就提供了3個方法，

• Add(delta int) 添加攜程記錄

• Done() 移除攜程記錄

• Wait() 同步等待所有記錄的攜程全部結束

代碼

package main import (    "fmt"    "sync"    "time") var wg sync.WaitGroup func main() {    for i := 0; i < 3; i++ {        wg.Add(1) //建立一個協程，記錄一次        go worker(i)    }    wg.Wait() //同步等待所有的協程全部執行完成    fmt.Println("All done !")}func worker(i int) {    fmt.Println(i)    time.Sleep(time.Second * time.Duration(i))    wg.Done() //攜程結束，從屬記錄中移除等同於Add(-1)}

輸出：

210All done !

image