Go素数筛选分析

Go素数筛选分析

1. 素数筛选介绍

学习Go语言的过程中，遇到素数筛选的问题。这是一个经典的并发编程问题，是某大佬的代码，短短几行代码就实现了素数筛选。但是自己看完原理和代码后一脸懵逼（仅此几行能实现素数筛选），然后在网上查询相关资料，依旧似懂非懂。经过1天的分析调试，目前基本上掌握了的原理。在这里介绍一下学习理解的过程。

素数筛选基本原理如下图：

就原理来说还是比较简单的，首先生成从 2 开始的递增自然数，然后依次对生成的第 1, 2, 3, ...个素数 整除，经过全部整除仍有余数的自然数，将会是素数。

大佬的代码如下：

// 返回生成自然数序列的管道: 2, 3, 4, ...
// GenerateNatural 函数内部启动一个 Goroutine 生产序列，返回对应的管道
func GenerateNatural() chan int {
   
	ch := make(chan int)
	go func() {
   
		for i := 2; ; i++ {
   
			ch <- i
		}
	}()
	return ch
}
// 管道过滤器: 将输入序列中是素数倍数的数淘汰，并返回新的管道
// 函数内部启动一个 Goroutine 生产序列，返回过滤后序列对应的管道
func PrimeFilter(in <-chan int, prime int) chan int {
   
	out := make(chan int)
	go func() {
   
		for {
   
			if i := <-in; i%prime != 0 {
   
				out <- i
			}
		}
	}()
	return out
}
func main() {
   
	ch := GenerateNatural() // 自然数序列: 2, 3, 4, ...
	for i := 0; i < 100; i++ {
   
		prime := <-ch // 新出现的素数
		fmt.Printf("%v: %v\n", i+1, prime)
		ch = PrimeFilter(ch, prime) // 基于新素数构造的过滤器
	}
}

main()函数先是调用 GenerateNatural() 生成最原始的从 2 开始的自然数序列。然后开始一个 100 次迭代的循环，希望生成 100 个素数。在每次循环迭代开始的时候，管道中的第一个数必定是素数，我们先读取并打印这个素数。然后基于管道中剩余的数列，并以当前取出的素数为筛子过滤后面的素数。不同的素数筛子对应的管道是串联在一起的。

运行代码，程序正确输出如下：

1: 2
2: 3
3: 5
......
......
98: 521
99: 523
100: 541

2. 代码分析

之前在课本中学习到：chan底层结构 是一个指针，所以我们能在函数间直接传递 channel，而不用传递 channel 的指针。

上述代码fun GenerateNatural()中创建了管道ch := make(chan int)，并创建一个协程（为了便于描述，该协程称为Gen）持续向ch中写入渐增自然数。

当i=0时，main()中prime := <-ch读取该ch（此时prime=2，输出素数2），接着将ch传入PrimeFilter(ch, prime)中。PrimeFilter(ch, prime)创建新协程（称为PF(ch, 2)）持续读取传入的ch（ch中2之前已被取出，从3依次往后读取），同时返回一个新的chan out（当通过过滤器的i向out写入时，此时out仅有写入而没有读取操作，PF(ch, 2)将阻塞在第1次写chan out操作）。与此同时main()中ch = PrimeFilter(ch, 2)将out赋值给ch，此操作给ch赋了新变量。到这里，重点来了：由于在随后的时间里，协程Gen、PF(ch, 2)中仍需要不停写入和读取ch，这里将<