Golang並發程式設計：使用Go WaitGroup實現任務調度器

Golang並發程式設計：使用Go WaitGroup實作任務調度器

1. 引言
   在Golang中，實作並發程式設計可以大幅提升程式的效能和效率。而任務調度器是並發程式設計中非常重要的一部分，它可以用於調度並發任務的執行順序和同步任務的完成。本文將介紹如何使用Golang中的WaitGroup來實作一個簡單的任務調度器，並給出具體的程式碼範例。
2. WaitGroup簡介
   WaitGroup是Golang套件sync中的重要類型，它提供了並發任務的同步和等待功能。 WaitGroup有三個主要的方法：Add、Done和Wait。

• Add(n int)：新增 n 個並發任務到WaitGroup。
• Done()：標記一個任務完成。
• Wait()：等待所有任務完成，並阻塞目前的goroutine。

WaitGroup內部維護了一個計數器，用來記錄未完成的任務數量。當呼叫Add方法時，計數器加上指定的值；當呼叫Done方法時，計數器減1；當呼叫Wait方法時，如果計數器大於0，當前goroutine會被阻塞，直到計數器歸零。

3. 任務調度器實作
   下面是一個使用WaitGroup實作任務調度器的範例程式碼：

package main

import (
    "fmt"
    "sync"
    "time"
)

func worker(id int, wg *sync.WaitGroup) {
    defer wg.Done()
    fmt.Printf("Worker %d starting
", id)
    time.Sleep(time.Second)
    fmt.Printf("Worker %d done
", id)
}

func main() {
    var wg sync.WaitGroup
    for i := 1; i <= 5; i++ {
        wg.Add(1)
        go worker(i, &wg)
    }
    wg.Wait()
    fmt.Println("All workers done")
}

在這個範例中，我們使用一個for循環啟動5個worker goroutine，並透過呼叫wg.Add(1)來將任務數量加入WaitGroup。在worker函數中，使用defer關鍵字來確保在函數傳回前呼叫wg.Done()，來減少計數器的值。最後，使用wg.Wait()來等待所有任務完成。

4. 運行結果
   運行這個範例程式碼，你會看到如下的輸出結果：

Worker 1 starting
Worker 2 starting
Worker 3 starting
Worker 4 starting
Worker 5 starting
Worker 3 done
Worker 1 done
Worker 2 done
Worker 5 done
Worker 4 done
All workers done

可以看到，所有的worker goroutine被並發執行，但是輸出的順序不一定是按照啟動順序來的，這是因為goroutine的調度是由Go運行時進行的。

5. 總結
   使用WaitGroup可以方便地實現並發任務的調度和同步。透過呼叫Add方法加入任務數量、呼叫Done方法標記任務完成、呼叫Wait方法來等待任務完成，可以實現任務的同步和順序控制。在實際應用中，任務調度器可以用於處理並發任務、限制並發數量等場景，幫助我們更好地利用Golang的並發能力。

透過本文的介紹和範例程式碼，相信你對如何使用Golang的WaitGroup實現任務調度器有了更深的理解。希望本文對你學習和使用Golang的並發程式設計有幫助！

以上是Golang並發程式設計：使用Go WaitGroup實現任務調度器的詳細內容。更多資訊請關注PHP中文網其他相關文章！