Go 并发实战核心编程【一】

Go 并发实战核心编程【一】

微信公众号

1. 需求

• 启动一个goroutine，将1-10000的数字放入chan中

• 启动4个goroutine从chan中读取数字，并计算是不是素数

• 是素数就讲结果放入结果chan中

• 最后遍历结果chan，打印素数集合

2. 思路

这道题思路很简单，首先明确一个点就是这个需求有三种类型的goroutine：

• 第一种类型就是生产者，它主要负责数据的生产；

• 第二种类型就是消费者，他主要负责消费数据做加工；

• 第三种类型就是main goroutine，他主要负责调度这些goroutine并且等待他们退出。

明确思路之后我们在深入分析下，这个需求需要三个chan：

• 第一个就是生产者和消费者之间的通信是需要一个chan的，这个chan就是生产者生产数字放到chan中，消费者从chan取出数字。

• 第二个就是消费者计算完素数之后结果得放入一个chan中，这个chan需要和main goroutine通信，因为main goroutine需要遍历打印这些素数；

• 最后一个就是main goroutine需要等待所有消费goroutine的退出，因为我不知道消费者什么时候结束，所以也需要一个chan和main goroutine通信。

因此这个需求需要三种类型的goroutine和三种类型的chan。

3. 编码

package main

import "fmt"

// 生产者
func producer(intChan chan int) {
 fmt.Println("producer goroutine 开始生产数据啦")
 for i:=1; i<10000; i++ {
  intChan <- i
 }

 //生产完毕就关闭
 close(intChan)
 fmt.Println("producer goroutine 退出啦")
}

// 消费者
func consumer(intChan chan int, primeChan chan int, exitChan chan bool) {
 fmt.Println("consumer goroutine 开始消费数据啦")
 for {
  num, ok := <- intChan
  if !ok {
   break //生产者已经关闭intChan 那么消费者读取到关闭信号之后就应该退出
  }

  //计算素数
  flag := true
  for i:=2; i

4. 小结

• 谁负责生产数据到chan，谁就负责关闭chan

• 谁最后处理完chan中数据，就应该通知调用方自己退出了

• main goroutine一定是程序最后一个goroutine退出的

- END -