当前位置 博文首页 > go并发实现素数筛的代码

    go并发实现素数筛的代码

    作者:kelianlee 时间:2021-04-29 17:47

    先贴代码

    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)

    js
    下一篇:没有了