使用Go WaitGroup實現高效率並發處理任務

使用Go WaitGroup實現高效並發處理任務

在Go語言中，使用WaitGroup可以實現高效的並發處理任務。 WaitGroup是一個計數信號量，用於等待一組goroutine完成它們的工作後再繼續執行下一步操作。

WaitGroup有三個主要的方法：Add()、Done()和Wait()。 Add()方法用於增加等待的goroutine的數量，Done()方法表示一個goroutine已經完成，Wait()方法用於阻塞主goroutine，直到所有的goroutine都已完成。

下面將透過一個簡單的範例來說明如何使用WaitGroup並發處理任務。

首先，我們需要建立一個WaitGroup實例：

var wg sync.WaitGroup

然後，我們可以使用Add()方法增加等待的goroutine數量，並在每個goroutine的最後呼叫Done()方法：

func main() {
    // 设置等待的goroutine数量
    wg.Add(2)

    // 启动第一个goroutine
    go doTask1()

    // 启动第二个goroutine
    go doTask2()

    // 等待所有goroutine完成
    wg.Wait()
}

func doTask1() {
    // 模拟任务1
    time.Sleep(time.Second * 1)
    fmt.Println("Task 1 is done!")

    // 调用Done()方法表示任务完成
    wg.Done()
}

func doTask2() {
    // 模拟任务2
    time.Sleep(time.Second * 2)
    fmt.Println("Task 2 is done!")

    // 调用Done()方法表示任务完成
    wg.Done()
}

在上面的範例中，我們建立了兩個goroutine來執行doTask1()和doTask2()函數，每個函數都會模擬一個耗時的任務。主goroutine在啟動完所有的goroutine後呼叫Wait()方法阻塞自己，直到所有的任務都完成。

運行上面的程式碼，可以得到類似以下的輸出：

Task 1 is done!
Task 2 is done!

可以看到，兩個任務是並行執行的，並在完成後分別列印出了相應的資訊。使用WaitGroup可以方便地進行並發處理，並在所有任務完成後恢復主goroutine的執行。

要注意的是，在使用WaitGroup時，必須保證每個任務都會呼叫Done()方法，否則主goroutine會一直阻塞在Wait()方法處，導致程式無法繼續執行。

總結起來，Go WaitGroup是一種簡單有效的並發控制工具，能夠幫助我們實現高效的並發處理任務。透過呼叫Add()、Done()和Wait()方法，我們可以控制goroutine的執行順序，並在所有任務完成後繼續執行後續操作。

希望本文對您理解並發處理任務和使用Go WaitGroup有所幫助！

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