编写代码实现了一个计算随机数的每个位置数字之和的程序,要求使用goroutine
和channel
构建生产者和消费者模式,可以指定启动的goroutine数量-woker pool
模式。
在工作中我们通常使用workerpool
模式,控制goroutine
的数量,防止goroutine
泄露和暴涨。
一个简易的workerpool
示例代码如下:
package main
import (
"fmt"
"time"
)
func worker(id int, jobs <-chan int, results chan<- int) {
for index := range jobs {
fmt.Printf("worker:%d start job:%d\n", id, index)
time.Sleep(time.Second)
fmt.Printf("worker:%d end job%d\n", id, index)
results <- index * 2
}
}
func main() {
jobs := make(chan int, 100)
results := make(chan int, 100)
//开启3个协程
for w := 1; w <= 3; w++ {
go worker(w, jobs, results)
}
//5 个任务
for i := 0; i < 5; i++ {
jobs <- i //加入任务
}
//关闭协程
close(jobs)
//输出结果
for i := 0; i < 5; i++ {
<-results
}
}