在线文档 > Golang练习 > 工作池
在这个例子中，我们将看看如何使用 协程 和 channel 实现一个工作池。
package main
import (
"fmt"
"time"
)
// 这就是我们的worker，我们将运行多个并发实例。
// 这些worker将在jobs通道上接收工作，并在results上发送相应的结果。
// 我们每个work睡眠一秒钟，以模拟一个耗时的任务。
func worker(id int, jobs <-chan int, results chan<- int) {
for j := range jobs {
fmt.Println("worker", id, "started job", j)
time.Sleep(time.Second)
fmt.Println("worker", id, "finished job", j)
results <- j * 2
}
}
func main() {
// 为了使用我们的worker池，我们需要发送他们的工作并收集他们的结果。
// 我们制作两个通道来实现这一点。
const numJobs = 5
jobs := make(chan int, numJobs)
results := make(chan int, numJobs)
//这启动了3个worker，最开始是被阻塞的，因为还没有运行。
for w := 1; w <= 3; w++ {
go worker(w, jobs, results)
}
// 这里我们发送5个“work”，然后通过`close` 关闭 该通道，以表明这是我们所有的工作。
for j := 1; j <= numJobs; j++ {
jobs <- j
}
close(jobs)
// 最后，我们收集所有工作的结果。
// 这也确保worker 协程已经完成。等待多个 协程 的另一种替代方式是使用WaitGroup。
for a := 1; a <= numJobs; a++ {
<-results
}
}
运行结果如下：
$ time go run worker-pools.go
worker 1 started job 1
worker 2 started job 2
worker 3 started job 3
worker 1 finished job 1
worker 1 started job 4
worker 2 finished job 2
worker 2 started job 5
worker 3 finished job 3
worker 1 finished job 4
worker 2 finished job 5
real 0m2.358s
我们的运行程序显示了不同worker正在执行的 5 个作业。尽管总共做了大约 2 秒的工作，但该程序只需要大约 5 秒，因为有 3 名工人同时运行。