Go 工作池

在这个例子中，我们来看一下如何使用gorouotine和channel来实现工作池。

  1. package main
  3. import "fmt"
  4. import "time"
  6. // 我们将在worker函数里面运行几个并行实例，这个函数从jobs通道
  7. // 里面接受任务，然后把运行结果发送到results通道。每个job我们
  8. // 都休眠一会儿，来模拟一个耗时任务。
  9. func worker(id int, jobs <-chan int, results chan<- int) {
  10.     for j := range jobs {
  11.         fmt.Println("worker", id, "processing job", j)
  12.         time.Sleep(time.Second)
  13.         results <- j * 2
  14.     }
  15. }
  17. func main() {
  19.     // 为了使用我们的工作池，我们需要发送工作和接受工作的结果，
  20.     // 这里我们定义两个通道，一个jobs，一个results
  21.     jobs := make(chan int, 100)
  22.     results := make(chan int, 100)
  24.     // 这里启动3个worker协程，一开始的时候worker阻塞执行，因为
  25.     // jobs通道里面还没有工作任务
  26.     for w := 1; w <= 3; w++ {
  27.         go worker(w, jobs, results)
  28.     }
  30.     // 这里我们发送9个任务，然后关闭通道，告知任务发送完成
  31.     for j := 1; j <= 9; j++ {
  32.         jobs <- j
  33.     }
  34.     close(jobs)
  36.     // 然后我们从results里面获得结果
  37.     for a := 1; a <= 9; a++ {
  38.         <-results
  39.     }

运行结果

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