如何使用Go WaitGroup處理並發任務

如何使用Go WaitGroup處理並發任務

在Go語言中，我們可以透過使用sync.WaitGroup來處理並發任務。 sync.WaitGroup可以在處理並發任務時提供一種簡潔而有效的方式來協調協程的執行。

sync.WaitGroup是一個有用的工具，在我們不知道有多少協程需要等待的情況下，是處理並發任務的首選方法。它可以讓我們確保在所有任務都完成之前，主協程不會結束執行。

讓我們來看一個具體的例子，展示如何使用sync.WaitGroup來處理並發任務。

首先，我們需要導入sync套件：

import (
    "fmt"
    "sync"
)

接下來，讓我們建立一個sync.WaitGroup物件：

var wg sync.WaitGroup

然後，我們可以透過呼叫Add方法來新增需要等待的任務數目。在這個範例中，我們將新增兩個任務：

wg.Add(2)

接下來，我們可以啟動兩個協程來執行任務。我們可以將任務封裝在一個匿名函數中，然後將其作為參數傳遞給go關鍵字。

go func() {
    defer wg.Done()
    // 这里是第一个任务的代码逻辑
}()

go func() {
    defer wg.Done()
    // 这里是第二个任务的代码逻辑
}()

在這個例子中，我們將每個協程內部的defer wg.Done()放在go關鍵字之後，以確保任務完成之後能夠通知WaitGroup物件。

最後，我們可以執行Wait方法來阻塞主協程，直到所有任務都完成。這個方法將等待WaitGroup物件中的任務數目減為零。

wg.Wait()

下面是完整的範例程式碼：

package main

import (
    "fmt"
    "sync"
)

func main() {
    var wg sync.WaitGroup
    wg.Add(2)

    go func() {
        defer wg.Done()
        fmt.Println("Task 1 executed")
    }()

    go func() {
        defer wg.Done()
        fmt.Println("Task 2 executed")
    }()

    wg.Wait()
    fmt.Println("All tasks completed")
}

執行這段程式碼，你會看到輸出列印了兩個任務的執行結果，然後列印了"所有任務已完成"。

透過使用sync.WaitGroup，我們可以很方便地處理並發任務，確保所有任務都執行完成之後再繼續主協程的執行。這對於需要在主協程中等待所有任務完成的情況下特別有用，例如爬蟲、並行計算等場景。

總結一下，使用sync.WaitGroup可以幫助我們處理Go語言中的並發任務。它提供了一個簡潔而有效的方式來協調協程的執行，並確保所有任務都完成之後再繼續主協程的執行。希望這篇文章對你理解並發任務處理有幫助。

（總字數：409）

以上是如何使用Go WaitGroup處理並發任務的詳細內容。更多資訊請關注PHP中文網其他相關文章！