Go WaitGroup的使用方式及實例詳解

Go語言是一種以並發程式設計為核心的程式語言，在處理並發任務時非常有效率。在Go語言的標準函式庫中，有一個非常好用的並發工具－WaitGroup。本文將詳細介紹WaitGroup的使用方式，並透過實例來示範其具體應用。

首先，我們要先了解WaitGroup的作用。 WaitGroup用於等待一組並發任務完成後再繼續執行後續的程式碼。通常情況下，我們需要在主協程等待其他協程完成某個任務再繼續向下執行時，就可以使用WaitGroup。

在Go語言的sync套件下，有一個WaitGroup結構體，它有3個主要的方法，分別是Add()、Done()和Wait()。以下我們逐一介紹它們的作用。

首先是Add()方法，它用來增加等待的任務數。我們可以透過呼叫Add()方法來告訴WaitGroup我們有多少個任務需要等待完成。具體的呼叫方式是wg.Add(n)，其中n表示需要等待的任務數量。例如，如果我們有10個任務需要等待完成，就可以呼叫wg.Add(10)。

然後是Done()方法，它用於標誌當前任務已經完成。每個任務完成時，都需要呼叫Done()方法告訴WaitGroup目前任務已經完成。具體的呼叫方式是wg.Done()。通常情況下，我們會在任務完成後的defer語句中呼叫Done()方法，以確保任務完成後一定會呼叫Done()方法。

最後是Wait()方法，它用來阻塞主協程，直到所有任務都已經完成。當主協程需要等待一組任務完成後再繼續執行後續的程式碼時，可以呼叫Wait()方法。具體的調用方式是wg.Wait()。 Wait()方法會一直阻塞，直到所有任務都已經完成。

下面我們透過一個實例來示範WaitGroup的具體應用。假設我們有一個任務列表，我們需要並發地處理這些任務。在每個任務完成後，我們都會將結果寫入一個結果清單。最後，我們需要等待所有任務都完成後，才能列印出結果清單。程式碼如下所示：

package main

import (
    "fmt"
    "sync"
)

func main() {
    var wg sync.WaitGroup
    taskList := []string{"task1", "task2", "task3", "task4"}

    // 创建结果列表
    resultList := make([]string, len(taskList))

    // 遍历任务列表，开启多个协程处理任务
    for i, task := range taskList {
        wg.Add(1) // 增加等待的任务数

        go func(index int, task string) {
            defer wg.Done() // 标志任务已经完成

            // 模拟任务处理过程
            result := fmt.Sprintf("task %s processed", task)

            // 将结果写入结果列表
            resultList[index] = result
        }(i, task)
    }

    wg.Wait() // 等待所有任务完成

    // 打印结果列表
    for _, result := range resultList {
        fmt.Println(result)
    }
}

在上面的程式碼中，首先我們建立了一個WaitGroup物件wg。然後，我們遍歷任務列表，針對每個任務呼叫了一次wg.Add(1)，表示有一個任務需要等待。在每個任務的處理函數中，我們在任務完成後呼叫了一次wg.Done()。最後，我們在主協程中呼叫了wg.Wait()來等待所有任務都完成。

透過這樣的設計，我們可以並發地處理任務，並且確保在所有任務都完成後，再繼續主協程的後續執行。最後，我們列印出了結果列表，驗證了任務的並發處理和結果的正確性。

總結來說，WaitGroup是Go語言中非常實用的並發工具。透過合理使用WaitGroup，我們可以更好地管理並發任務，確保任務的順序和正確性。希望本文能幫助讀者更好地理解並應用WaitGroup。

以上是Go WaitGroup的使用方式及實例詳解的詳細內容。更多資訊請關注PHP中文網其他相關文章！