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)

推薦閱讀: