go 函數(shù)中的高效并行任務(wù)處理:使用 go 關(guān)鍵字啟動(dòng)并發(fā)例程。使用 sync.waitgroup 計(jì)數(shù)未完成例程的數(shù)量。例程完成時(shí)調(diào)用 wg.done() 遞減計(jì)數(shù)器。主程序使用 wg.wait() 阻塞,直到所有例程完成。實(shí)戰(zhàn)案例:并發(fā)發(fā)送 web 請(qǐng)求并收集響應(yīng)。
Go 函數(shù)中的高效并行任務(wù)處理
Go 語(yǔ)言提供了強(qiáng)大的并行處理能力,使開發(fā)人員能夠輕松地創(chuàng)建并執(zhí)行并發(fā)任務(wù)。本文將討論如何使用 Go 函數(shù)高效地處理并行任務(wù),并提供一個(gè)實(shí)戰(zhàn)案例來(lái)演示它的用法。
Go 函數(shù)中的并發(fā)
Go 函數(shù)提供了兩個(gè)關(guān)鍵字來(lái)支持并發(fā)執(zhí)行:go
和 sync.WaitGroup
。go
關(guān)鍵字用于啟動(dòng)并發(fā)的 Go 例程,而 sync.WaitGroup
用于等待所有例程完成。
等待組
sync.WaitGroup
是一個(gè)計(jì)數(shù)器,用于跟蹤未完成例程的數(shù)量。當(dāng)例程完成時(shí),它會(huì)調(diào)用 wg.Done()
來(lái)遞減計(jì)數(shù)器。主程序可以使用 wg.Wait()
方法來(lái)阻塞,直到所有例程完成。
實(shí)戰(zhàn)案例:并發(fā) Web 請(qǐng)求
考慮一個(gè)場(chǎng)景,我們需要并發(fā)發(fā)送多個(gè) Web 請(qǐng)求并收集響應(yīng)。以下是演示如何使用 Go 函數(shù)進(jìn)行高效并發(fā)任務(wù)處理的代碼:
package main import ( "fmt" "net/http" "sync" ) func main() { // 要發(fā)送的 Web 請(qǐng)求 URL urls := []string{"https://example.com", "https://example2.com", "https://example3.com"} // 創(chuàng)建等待組 var wg sync.WaitGroup for _, url := range urls { // 啟動(dòng)一個(gè)并發(fā)例程來(lái)發(fā)送 Web 請(qǐng)求 wg.Add(1) go func(url string) { // 發(fā)送 GET 請(qǐng)求 resp, err := http.Get(url) if err != nil { fmt.Printf("Error getting %s: %v\n", url, err) } else { fmt.Printf("Status code for %s: %d\n", url, resp.StatusCode) } // 例程完成,遞減等待組計(jì)數(shù) wg.Done() }(url) } // 等待所有例程完成 wg.Wait() }
登錄后復(fù)制
在上面的代碼中:
我們使用 sync.WaitGroup
來(lái)跟蹤未完成例程的數(shù)量。
對(duì)于每個(gè) Web 請(qǐng)求 URL,我們啟動(dòng)一個(gè)并發(fā)的 Go 例程來(lái)發(fā)送請(qǐng)求。
例程使用 wg.Done()
來(lái)表示它已完成。
主程序使用 wg.Wait()
阻塞,直到所有例程完成。
通過(guò)使用這種方法,我們可以并發(fā)發(fā)送多個(gè) Web 請(qǐng)求,并無(wú)需等待每個(gè)請(qǐng)求的響應(yīng)即可繼續(xù)執(zhí)行主程序。