在Golang中,可以使用Go WaitGroup來同步并發操作。WaitGroup是一個計數器,用于等待一組goroutine完成任務。
當我們需要異步處理多個任務時,可以使用WaitGroup來等待所有任務完成。下面是一種巧妙地結合異步處理和WaitGroup的方法:
var wg sync.WaitGroup
wg.Add(numTasks)
for _, task := range tasks {
go func(t Task) {
// 處理任務 t
// ...
// 任務完成時調用Done方法
wg.Done()
}(task)
}
wg.Wait()
這樣,我們就可以同時處理多個任務,并在所有任務完成后繼續執行后續操作。
完整的示例代碼如下:
package main
import (
"fmt"
"sync"
"time"
)
type Task struct {
id int
}
func main() {
numTasks := 10
tasks := make([]Task, numTasks)
var wg sync.WaitGroup
wg.Add(numTasks)
for i := 0; i < numTasks; i++ {
tasks[i] = Task{id: i}
go func(t Task) {
// 模擬處理任務
time.Sleep(1 * time.Second)
fmt.Printf("Task %d done\n", t.id)
// 任務完成時調用Done方法
wg.Done()
}(tasks[i])
}
// 等待所有任務完成
wg.Wait()
fmt.Println("All tasks done")
}
輸出結果:
Task 1 done
Task 2 done
Task 5 done
Task 8 done
Task 4 done
Task 9 done
Task 6 done
Task 7 done
Task 3 done
Task 0 done
All tasks done
通過巧妙地結合異步處理和WaitGroup,我們可以高效地處理并發任務,并在所有任務完成后繼續執行后續操作。