在Go语言中可以使用goroutine和channel来实现并发处理同样的数据。下面是一个示例代码:
package main
import (
"fmt"
"sync"
)
func process(data int, wg *sync.WaitGroup, result chan<- int) {
// 处理数据的逻辑
// ...
// 将处理结果发送到结果通道
result <- data * 2
// 完成一个goroutine的工作,减少WaitGroup的计数
wg.Done()
}
func main() {
// 创建一个WaitGroup来等待所有goroutine完成
var wg sync.WaitGroup
// 创建一个结果通道
result := make(chan int)
// 设置并发的goroutine数量
concurrency := 5
// 启动goroutine进行并发处理
for i := 0; i < concurrency; i++ {
// 增加WaitGroup的计数
wg.Add(1)
// 启动一个goroutine进行处理
go process(i, &wg, result)
}
// 等待所有goroutine完成
wg.Wait()
// 关闭结果通道
close(result)
// 从结果通道读取结果并打印
for r := range result {
fmt.Println("Result:", r)
}
}
在上面的示例代码中,我们通过使用sync.WaitGroup
来等待所有的goroutine完成。每个goroutine在处理完数据后,将结果发送到结果通道result
中。最后,我们使用for range
循环从结果通道中读取结果并进行打印。
需要注意的是,如果不使用sync.WaitGroup
来等待所有goroutine完成的话,可能会导致主goroutine提前退出,从而导致一些goroutine的工作被中断。因此,为了确保所有的goroutine都能完成工作,我们需要使用sync.WaitGroup
来协调它们的执行。