基礎教學：Go WaitGroup及其在Golang中的應用-Golang-PHP中文網

基礎教學：Go WaitGroup及其在Golang中的應用

王林

發布： 2023-09-28 09:42:38

原創

1082 人瀏覽過

基础教程：Go WaitGroup及其在Golang中的应用

基礎教學：Go WaitGroup及其在Golang中的應用，需要具體程式碼範例

前言：
在Golang（Go語言）中，寫並發程序是一項常見任務。 Golang提供了豐富的並發原語和API，其中一個重要的工具是WaitGroup（等待組）。本文將為您介紹WaitGroup的概念和用法，並提供了一些具體的程式碼範例。

一、WaitGroup的概念
WaitGroup是Golang中的一種並發原語，它用於等待一組協程（goroutine）的完成。 WaitGroup內部維護一個計數器，初始值為0。當每個協程開始執行時，計數器會加1。當協程結束時，計數器會減1。主執行緒可以透過Wait()方法阻塞，直到計數器為0，也就是所有協程都執行完成。

二、WaitGroup的基本用法
使用WaitGroup需要依序進行以下步驟：

建立WaitGroup物件：首先，我們需要匯入"sync"包，並建立一個WaitGroup物件。
```
import "sync"

var wg sync.WaitGroup
```
登入後複製
新增協程到WaitGroup：對於每個要執行的協程，我們需要增加計數器，並使用go關鍵字在新的協程中執行該任務。
```
wg.Add(1)
go func() {
 // 协程的具体逻辑
 // ...
 wg.Done() // 协程执行完成后减少计数器
}()
```
登入後複製
等待協程的完成：主執行緒可以使用Wait()方法來阻塞，直到計數器為0，也就是所有協程都已經執行完成。
```
wg.Wait()
```
登入後複製

範例程式碼：
下面是一個具體的程式碼範例，示範了WaitGroup在Golang中的應用。此範例使用WaitGroup來等待多個協程完成後再繼續後續的邏輯。

package main
 
import (
    "fmt"
    "sync"
    "time"
)
 
func main() {
    var wg sync.WaitGroup
 
    for i := 0; i < 5; i++ {
        wg.Add(1)
 
        go func(num int) {
            defer wg.Done()
            time.Sleep(time.Second) // 模拟长时间的任务
 
            fmt.Printf("协程 %d 完成
", num)
        }(i)
    }
 
    fmt.Println("等待所有协程完成...")
    wg.Wait()
    fmt.Println("所有协程已完成！")
}

登入後複製

在上面的範例中，我們模擬了5個協程的執行，每個協程都等待1秒鐘。透過WaitGroup，主執行緒會阻塞，直到所有協程執行完成。在輸出中，我們可以看到協程完成的順序並不固定，因為它們是並行執行的。

三、WaitGroup的注意事項
在使用WaitGroup時，需要遵循以下幾點注意事項：