简单的用go协程统计8000内的素数

package main

import (
    "time"
    "fmt"
)

//向管道放入80000个数
func createData(size int, intDataChan chan int)  {
    if size < 1 {
        return
    }

    for i := 1; i <= size; i++ {
        intDataChan <- i
    }

    close(intDataChan)
}

func getSuShu(intDataChan chan int, primeChan chan int, exitChan chan bool)  {

    var flag bool
    for {
        time.Sleep(time.Millisecond * 10)
        num, ok := <- intDataChan
        if !ok {//取不到数据
            break
        }
        flag = true //假设是素数
        for i := 2; i <= num /2; i++{
            if num % i == 0 {
                flag = false //确定不是素数
                break
            }
        }

        if flag {
            primeChan <- num
        }
    }

    fmt.Println("\n 有一个协程取不到数据，退出")

    exitChan <- true //标记退出
}

func main() {
    var maxInt  = 8000
    intDataChan := make(chan int, 1000)
    primeChan := make(chan int, maxInt / 3)
    exitChan := make(chan bool, 4)

    go createData(maxInt, intDataChan)

    for i := 0; i < 4; i++ {
        go getSuShu(intDataChan, primeChan, exitChan)
    }

    go func() {
        for i := 0; i < 4; i++ {
            <- exitChan //阻塞直到四个协程都完成工作
        }

        close(primeChan)//关闭管道
    }()

    for {
        res, ok := <- primeChan
        if !ok {
            break
        }

        fmt.Print(res, "  ")
    }
}

转载于:https://www.cnblogs.com/hirampeng/p/11184288.html