034_go语言中的工作池

代码演示

package main

import "fmt"
import "time"

func worker(id int, jobs <-chan int, results chan<- int) {
	for j := range jobs {
		fmt.Println("worker", id, "processing job", j)
		time.Sleep(time.Second)
		results <- j * 2
	}
}
func main() {
	jobs := make(chan int, 100)
	results := make(chan int, 100)
	for w := 1; w <= 3; w++ {
		go worker(w, jobs, results)
	}
	for j := 1; j <= 9; j++ {
		jobs <- j
	}
	close(jobs)
	for a := 1; a <= 9; a++ {
		<-results
	}
}

　　

代码运行结果

worker 3 processing job 1
worker 2 processing job 3
worker 1 processing job 2
worker 3 processing job 4
worker 1 processing job 5
worker 2 processing job 6
worker 1 processing job 8
worker 2 processing job 7
worker 3 processing job 9

　　

代码解读

• 工作池可以用go协程和通道实现

• 以上例子创造了一个工作池，工作者从通道jobs里领取任务，做完以后将结果发送到results中

• 为了使用工作池，需要总共两个通道，一个用来传输任务，一个用来传输结果

• 上面例子中，启动了3个工作者，一开始是阻塞的，因为还没有传输任务

• 后面往jobs中发送了总共9个任务，然后close掉通道表示这是所有任务了

• 最后收集所有任务的返回值