在Golang中,我們可以使用sync
包中的WaitGroup
來實現并發控制。WaitGroup
用于等待一組goroutine完成執行。
下面是一個使用WaitGroup
來完成任務的示例:
package main
import (
"fmt"
"sync"
"time"
)
func main() {
var wg sync.WaitGroup
// 設置要執行的任務數量
taskCount := 5
wg.Add(taskCount)
for i := 1; i <= taskCount; i++ {
go performTask(i, &wg)
}
// 等待所有任務完成
wg.Wait()
fmt.Println("所有任務已完成")
}
func performTask(id int, wg *sync.WaitGroup) {
defer wg.Done()
fmt.Printf("開始執行任務 %d\n", id)
time.Sleep(2 * time.Second)
fmt.Printf("任務 %d 完成\n", id)
}
在上面的示例中,我們首先創建了一個WaitGroup
實例wg
,然后設置要執行的任務數量為5,通過調用wg.Add(taskCount)
方法來增加等待的goroutine數量。
接下來,我們使用一個循環來創建5個goroutine,并將每個goroutine的任務ID和WaitGroup
實例的指針傳遞給performTask
函數。
在performTask
函數中,我們使用defer
關鍵字來確保在函數執行完成后調用wg.Done()
方法,表示任務已完成。
在每個goroutine中,我們首先打印出任務開始執行的信息,然后通過time.Sleep
函數模擬任務執行的時間。最后,我們打印出任務完成的信息。
最后,我們調用wg.Wait()
方法來等待所有的goroutine完成任務。一旦所有的goroutine都調用了wg.Done()
方法,Wait
方法就會返回,程序繼續執行后面的代碼。
運行上面的代碼,你會看到類似下面的輸出:
開始執行任務 1
開始執行任務 2
開始執行任務 3
開始執行任務 4
開始執行任務 5
任務 3 完成
任務 1 完成
任務 4 完成
任務 5 完成
任務 2 完成
所有任務已完成
從輸出可以看出,5個任務是并發執行的,但是最后的輸出表明所有的任務都已完成。
使用WaitGroup
來完成任務的并發控制,它可以確保在所有的goroutine執行完成之前,主goroutine會一直等待。這在需要等待一組goroutine完成執行后再進行后續操作的場景中非常有用。