Go語言中除了可以使用通道（channel）和互斥鎖進行兩個并發(fā)程序間的同步外，還可以使用等待組進行多個任務(wù)的同步，等待組可以保證在并發(fā)環(huán)境中完成指定數(shù)量的任務(wù)

在 sync.WaitGroup（等待組）類型中，每個 sync.WaitGroup 值在內(nèi)部維護著一個計數(shù)，此計數(shù)的初始默認值為零。

等待組有下面幾個方法可用，如下表所示。

對于一個可尋址的 sync.WaitGroup 值 wg：

• 我們可以使用方法調(diào)用 wg.Add(delta) 來改變值 wg 維護的計數(shù)。
• 方法調(diào)用 wg.Done() 和 wg.Add(-1) 是完全等價的。
• 如果一個 wg.Add(delta) 或者 wg.Done() 調(diào)用將 wg 維護的計數(shù)更改成一個負數(shù)，一個恐慌將產(chǎn)生。
• 當(dāng)一個協(xié)程調(diào)用了 wg.Wait() 時，
  • 如果此時 wg 維護的計數(shù)為零，則此 wg.Wait() 此操作為一個空操作（noop）；
  • 否則（計數(shù)為一個正整數(shù)），此協(xié)程將進入阻塞狀態(tài)。當(dāng)以后其它某個協(xié)程將此計數(shù)更改至 0 時（一般通過調(diào)用 wg.Done()），此協(xié)程將重新進入運行狀態(tài)（即 wg.Wait() 將返回）。

等待組內(nèi)部擁有一個計數(shù)器，計數(shù)器的值可以通過方法調(diào)用實現(xiàn)計數(shù)器的增加和減少。當(dāng)我們添加了 N 個并發(fā)任務(wù)進行工作時，就將等待組的計數(shù)器值增加 N。每個任務(wù)完成時，這個值減 1。同時，在另外一個 goroutine 中等待這個等待組的計數(shù)器值為 0 時，表示所有任務(wù)已經(jīng)完成。

下面的代碼演示了這一過程：

package main
import (
    "fmt"
    "net/http"
    "sync"
)
func main() {
    // 聲明一個等待組
    var wg sync.WaitGroup
    // 準備一系列的網(wǎng)站地址
    var urls = []string{
        "http://www.github.com/",
        "https://www.qiniu.com/",
        "https://www.golangtc.com/",
    }
    // 遍歷這些地址
    for _, url := range urls {
        // 每一個任務(wù)開始時, 將等待組增加1
        wg.Add(1)
        // 開啟一個并發(fā)
        go func(url string) {
            // 使用defer, 表示函數(shù)完成時將等待組值減1
            defer wg.Done()
            // 使用http訪問提供的地址
            _, err := http.Get(url)
            // 訪問完成后, 打印地址和可能發(fā)生的錯誤
            fmt.Println(url, err)
            // 通過參數(shù)傳遞url地址
        }(url)
    }
    // 等待所有的任務(wù)完成
    wg.Wait()
    fmt.Println("over")
}

代碼說明如下：

• 第 12 行，聲明一個等待組，對一組等待任務(wù)只需要一個等待組，而不需要每一個任務(wù)都使用一個等待組。
• 第 15 行，準備一系列可訪問的網(wǎng)站地址的字符串切片。
• 第 22 行，遍歷這些字符串切片。
• 第 25 行，將等待組的計數(shù)器加1，也就是每一個任務(wù)加 1。
• 第 28 行，將一個匿名函數(shù)開啟并發(fā)。
• 第 31 行，在匿名函數(shù)結(jié)束時會執(zhí)行這一句以表示任務(wù)完成。wg.Done() 方法等效于執(zhí)行 wg.Add(-1)。
• 第 34 行，使用 http 包提供的 Get() 函數(shù)對 url 進行訪問，Get() 函數(shù)會一直阻塞直到網(wǎng)站響應(yīng)或者超時。
• 第 37 行，在網(wǎng)站響應(yīng)和超時后，打印這個網(wǎng)站的地址和可能發(fā)生的錯誤。
• 第 40 行，這里將 url 通過 goroutine 的參數(shù)進行傳遞，是為了避免 url 變量通過閉包放入匿名函數(shù)后又被修改的問題。
• 第 44 行，等待所有的網(wǎng)站都響應(yīng)或者超時后，任務(wù)完成，Wait 就會停止阻塞。

例子

同時開三個協(xié)程去請求網(wǎng)頁， 等三個請求都完成后才繼續(xù) Wait 之后的工作

var wg sync.WaitGroup
var urls = []string{
   "http://www.golang.org/",
   "http://www.google.com/",
   "http://www.somestupidname.com/",
}
for _, url := range urls {
   // Increment the WaitGroup counter. 
   wg.Add(1)
   // Launch a goroutine to fetch the URL. 
   go func(url string) {
      // Decrement the counter when the goroutine completes. 
      defer wg.Done()
      // Fetch the URL. 
      http.Get(url)
   }(url)
}
// Wait for all HTTP fetches to complete. 
wg.Wait()

或者下面的代碼
用于測試 給chan發(fā)送 1千萬次，并接受1千萬次的性能

import (
   "fmt"
    "sync"
    "time"
)

const (
   num = 10000000
)

func main() {
   TestFunc("testchan", TestChan)
}

func TestFunc(name string, f func()) {
   st := time.Now().UnixNano()
   f()
   fmt.Printf("task %s cost %d \r\n", name, (time.Now().UnixNano()-st)/int64(time.Millisecond))
}

func TestChan() {
   var wg sync.WaitGroup
   c := make(chan string)
   wg.Add(1)

   go func() {
      for _ = range c {
      }
      wg.Done()
   }()

   for i := 0; i < num; i++ {
      c <- "123"
   }
   close(c)
   wg.Wait()
}

到此這篇關(guān)于Go語言等待組sync.WaitGrou的使用示例的文章就介紹到這了,更多相關(guān)Go語言等待組sync.WaitGrou內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！ 

最新評論