go 协程求素数的代码实现

package main

import (
	_ "a1/utils"
	"fmt"
	_ "time"
)

// 向 initChan 放入 1-8000 个数
func putNum(initChan chan int) {
	for i := 1; i <= 8000; i++ {
		initChan <- i
	}
	// 关闭initChan
	close(initChan)

	// for {
	// 	res, ok := <-initChan
	// 	if !ok {
	// 		break
	// 	}
	// 	fmt.Printf("加入的数字=%d\n", res)
	// }
}

func primeNum(initChan chan int, primeChan chan int, exitChan chan bool) {

	// var num int
	var flag bool
	for {
		// time.Sleep(time.Millisecond * 10)
		num, ok := <-initChan
		if !ok { // initChan取不到内容
			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("有一个primeNum 协程因为取不到数据，退出")
	// 这里还不能关闭 primeNum
	// 向 exitChan 写入true
	exitChan <- true
}

// 协程求素数的代码实现
func main() {

	initChan := make(chan int, 1000)
	primeChan := make(chan int, 2000) // 放入结果
	// 标识退出的管道
	exitChan := make(chan bool, 4) // 4个

	// 开启一个协程，向 initChan 放入 1-8000 个数
	go putNum(initChan)

	// 开启4个协程，从 initChan 取出数据，并判断是否为素数
	// 如果是素数，就放入到primeChan
	for i := 0; i < 4; i++ {
		go primeNum(initChan ,primeChan ,exitChan)
	}

	go func() {
		for i := 0; i < 4; i++ {
			<-exitChan
		}
		// 当我们从exitChan 取出了4个结果，就可以放心的关闭 primeChan
		close(primeChan)
		close(exitChan)
	}()

	// 遍历 primeChan，把结果取出
	for {
		res, ok := <-primeChan
		if !ok {
			break
		}
		fmt.Printf("素数=%d\n", res)
	}
	fmt.Println("main线程退出")
}