Go語言等待組sync.WaitGrou的使用示例
Go語言中除了可以使用通道(channel)和互斥鎖進行兩個并發(fā)程序間的同步外,還可以使用等待組進行多個任務的同步,等待組可以保證在并發(fā)環(huán)境中完成指定數(shù)量的任務
在 sync.WaitGroup(等待組)類型中,每個 sync.WaitGroup 值在內(nèi)部維護著一個計數(shù),此計數(shù)的初始默認值為零。
等待組有下面幾個方法可用,如下表所示。
| 方法名 | 功能 |
|---|---|
| (wg * WaitGroup) Add(delta int) | 等待組的計數(shù)器 +1 |
| (wg * WaitGroup) Done() | 等待組的計數(shù)器 -1 |
| (wg * WaitGroup) Wait() | 當?shù)却M計數(shù)器不等于 0 時阻塞直到變 0。 |
對于一個可尋址的 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)生。
- 當一個協(xié)程調(diào)用了 wg.Wait() 時,
- 如果此時 wg 維護的計數(shù)為零,則此 wg.Wait() 此操作為一個空操作(noop);
- 否則(計數(shù)為一個正整數(shù)),此協(xié)程將進入阻塞狀態(tài)。當以后其它某個協(xié)程將此計數(shù)更改至 0 時(一般通過調(diào)用 wg.Done()),此協(xié)程將重新進入運行狀態(tài)(即 wg.Wait() 將返回)。
等待組內(nèi)部擁有一個計數(shù)器,計數(shù)器的值可以通過方法調(diào)用實現(xiàn)計數(shù)器的增加和減少。當我們添加了 N 個并發(fā)任務進行工作時,就將等待組的計數(shù)器值增加 N。每個任務完成時,這個值減 1。同時,在另外一個 goroutine 中等待這個等待組的計數(shù)器值為 0 時,表示所有任務已經(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 {
// 每一個任務開始時, 將等待組增加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)
}
// 等待所有的任務完成
wg.Wait()
fmt.Println("over")
}
代碼說明如下:
- 第 12 行,聲明一個等待組,對一組等待任務只需要一個等待組,而不需要每一個任務都使用一個等待組。
- 第 15 行,準備一系列可訪問的網(wǎng)站地址的字符串切片。
- 第 22 行,遍歷這些字符串切片。
- 第 25 行,將等待組的計數(shù)器加1,也就是每一個任務加 1。
- 第 28 行,將一個匿名函數(shù)開啟并發(fā)。
- 第 31 行,在匿名函數(shù)結(jié)束時會執(zhí)行這一句以表示任務完成。wg.Done() 方法等效于執(zhí)行 wg.Add(-1)。
- 第 34 行,使用 http 包提供的 Get() 函數(shù)對 url 進行訪問,Get() 函數(shù)會一直阻塞直到網(wǎng)站響應或者超時。
- 第 37 行,在網(wǎng)站響應和超時后,打印這個網(wǎng)站的地址和可能發(fā)生的錯誤。
- 第 40 行,這里將 url 通過 goroutine 的參數(shù)進行傳遞,是為了避免 url 變量通過閉包放入匿名函數(shù)后又被修改的問題。
- 第 44 行,等待所有的網(wǎng)站都響應或者超時后,任務完成,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)文章希望大家以后多多支持腳本之家!
相關(guān)文章
Golang限流器time/rate設計與實現(xiàn)詳解
在?Golang?庫中官方給我們提供了限流器的實現(xiàn)golang.org/x/time/rate,它是基于令牌桶算法(Token?Bucket)設計實現(xiàn)的,下面我們就來看看他的具體使用吧2024-03-03
使用go語言實現(xiàn)查找兩個數(shù)組的異同操作
這篇文章主要介紹了使用go語言實現(xiàn)查找兩個數(shù)組的異同操作,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-12-12

