關於golang利用channel和goroutine完成統計素數的思路

1. 需求

要求統計1-200000的數字中,哪些是素數?這個問題在本章開篇就提出來瞭,可以使用goroutine和channel來完成

2.分析思路

傳統的方法,就是使用一個循環,循環的判斷各個數是不是素數【ok】

使用並發/並行的方式,將統計素數的任務分配給多個(x個)goroutine去完成,完成任務時間短

分析思路圖:

 代碼實現:

package main
import (
	"fmt"
	"time"
)
// 需求:
// 要求統計1-200000的數字,哪些是素數?這個問題在本章開篇就提出瞭,現在我們有goroutine和channel的知識後,就可以完成瞭【測試數據:80000】
// 分析思路:
// 傳統的方法,就是使用一個循環,循環的判斷各個數是不是素數【ok】
// 使用並發/並行的方式,將統計素數的任務分配給多個(x個)goroutine去完成,完成任務時間短
 
// 1. 向intChan放入1-8000的數字
func putNum(intChan chan int) {
	for i := 1; i <= 20000; i++ {
		intChan <- i
	}
	// 關閉intChan
	close(intChan)
}
 
// 從intChan中取出數據,並判斷是否為素數,如果是,就放入到primeChan
func primeNum1(intChan chan int, primeChan chan int, exitChan chan bool) {
	// 使用for循環
	var flag bool
	for {
		time.Sleep(time.Millisecond * 10)
		num, ok := <- intChan
 
		if !ok { //intChan取不到,且關閉瞭管道
			break
		}
		flag = true // 假設是素數
		// 判斷num是不是素數
		for i := 2; i < num; i++ {
			if num % i == 0 { // 說明該num不是素數
				flag = false
				break
			}
		}
		if flag {
			// 將這個數放到primeChan
			primeChan<- num
		}
	}
	fmt.Println("有一個prieNum 協程因為取不到數據,退出")
	// 這裡我們還不能關閉primeChan
	// 向exitChan寫入true
	exitChan <- true
}
func main() {
	var intChan chan int = make(chan int, 1000)
	var primeChan chan int = make(chan int,2000)
	// 標識退出的管道
	exitChan := make(chan bool,4)
	// 開啟一個協程,向intChan放入 1-8000個數
	go putNum(intChan)
	// 開啟4個協程,從 intChan中取出數據,並判斷是否為素數,如果是就放入到primeChan
	for i := 0; i < 10000; i++ {
		go primeNum1(intChan,primeChan,exitChan)
	}
	// 這裡我們主線程,進行處理
	go func() {
		for i := 0; i < 4; i++ {
			<-exitChan
		}
		// 當我們從exitChan中取出瞭4個結果,就可以放心的關閉primeChan
		close(primeChan)
	}()
	// 遍歷我們的primeChan,把結果取出
	for {
		res, ok := <- primeChan
		if !ok {
			break
		}
		// 將結果取出
		fmt.Printf("素數=%d\n",res)
	}
	fmt.Println("main線程退出")
}

結論:使用go協程結合channel後,執行的速度,會大大 提高

go中可以輕松開啟1萬個協程 

到此這篇關於golang利用channel和goroutine完成統計素數的文章就介紹到這瞭,更多相關golang統計素數內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!

推薦閱讀: