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线程退出")
}
go 协程求素数的代码实现
最新推荐文章于 2024-06-18 19:45:17 发布