在Go語言中,可以使用WaitGroup和協程池來實現高效的并發編程。
下面是一個使用WaitGroup的示例代碼:
package main
import (
"fmt"
"sync"
)
func main() {
var wg sync.WaitGroup
wg.Add(2)
go func() {
defer wg.Done()
// 第一個協程的任務
fmt.Println("協程1完成")
}()
go func() {
defer wg.Done()
// 第二個協程的任務
fmt.Println("協程2完成")
}()
wg.Wait()
fmt.Println("所有協程完成")
}
輸出結果為:
協程2完成
協程1完成
所有協程完成
下面是一個簡單的協程池的實現示例:
package main
import (
"fmt"
)
func worker(id int, jobs <-chan int, results chan<- int) {
for j := range jobs {
// 處理任務的邏輯
fmt.Println("worker", id, "開始處理任務", j)
results <- j * 2
fmt.Println("worker", id, "完成任務", j)
}
}
func main() {
const numJobs = 5
jobs := make(chan int, numJobs)
results := make(chan int, numJobs)
// 啟動3個協程池
for w := 1; w <= 3; w++ {
go worker(w, jobs, results)
}
// 添加任務到任務通道
for j := 1; j <= numJobs; j++ {
jobs <- j
}
close(jobs)
// 讀取任務結果
for r := 1; r <= numJobs; r++ {
fmt.Println(<-results)
}
}
輸出結果為:
worker 3 開始處理任務 1
worker 1 開始處理任務 2
worker 2 開始處理任務 3
worker 1 完成任務 2
worker 1 開始處理任務 4
worker 2 完成任務 3
worker 3 完成任務 1
worker 1 完成任務 4
worker 2 開始處理任務 5
worker 2 完成任務 5
2
4
6
8
10
這個示例中,我們啟動了3個協程池,每個協程池會不斷從任務通道中獲取任務并處理。任務結果會被發送到結果通道中,最后讀取結果通道中的結果。
通過使用WaitGroup和協程池,我們可以高效地實現并發編程,并充分利用多核處理器的性能。