go并发实现素数筛的代码

先贴代码

package main

import (
    "fmt"
)

// Send the sequence 2, 3, 4, ... to channel 'ch'.
func generate(ch chan int) {
    for i := 2; ; i++ {
        ch <- i // Send 'i' to channel 'ch'.
    }
}

// Copy the values from channel 'in' to channel 'out',
// removing those divisible by 'prime'.
func filter(in, out chan int, prime int) {
    for {
        i := <-in // Receive value of new variable 'i' from 'in'.
        if i == 4{
            fmt.Println(prime)
        }
        if i%prime != 0 {
            out <- i // Send 'i' to channel 'out'.
        }
    }
}

// The prime sieve: Daisy-chain filter processes together.
func main() {
    ch := make(chan int) // Create a new channel.
    go generate(ch)   // Start generate() as a goroutine.
    for i:=0;i<10;i++ {
        prime := <-ch
        fmt.Println(&ch)
        fmt.Print(prime, " ")
        ch1 := make(chan int)
        go filter(ch, ch1, prime)
        ch = ch1
    }
}

 

第一次看这个素数筛一脸懵逼，首先我们要理解不是并发的素数筛，这个自行百度。

接下来我们看张图

这就是上面算法的流程图，我刚开始很难的理解是：为什么generate产生的2，3，4，5....等数一定是先被2筛选呢？我们知道并发有个不确定性，我们如果把generate函数当作生产者，那么2，3，5，7这几个素数过滤器协程就是消费者，根据一般的消费者生产模型，生产者产生的2，3，4，5等各个数被消费者使用的时候应该是不确定性的。

这里的关键在于generate这个channel只连接了2这个协程，并没有跟后面的协程相连接，而最秒的地方在于后面的每个协程的channel都只是与前后的协程相互通信的，这样就达到了一个串联的效果。

关键看代码filter函数的参数in，out这两个channel，in是上一个协程与当前协程所建立的channel，out则是当前协程与下一个协程所建立的channel（因为看main里面ch=ch1)

原文链接：https://blog.csdn.net/leekerian/article/details/114832843