Golang并行判断素数

## Golang多核判断素数方式

 package main

 import (
     "bufio"
     "fmt"
     "os"
     "runtime"
     "strconv"
     "syscall"
 )

 func main() {
     //开启真多核
     runtime.GOMAXPROCS(runtime.NumCPU())
     intchan := make(chan int, 10000) //向intchan内写入若干个数字,判断这些数字是否为质数
     final := make(chan int, 10000)   //结果集,质数都放在里面
     exitchan := make(chan bool, 5)   //当收集到5个true时,程序即可结束

     // 开启协程,向intchan放入数据(1-10000)
     go func(intchan chan int) {
         for i := 2; i <= 10000; i++ {
             intchan <- i
         }

     }(intchan)

     //开启四个正经工作的协程,他们来判断谁是质数
     for c := 1; c < 5; c++ {
         go countprime(intchan, final, exitchan, c)
     }

     // 从结果集合中写入文件
     go func(final chan int, exitchan chan bool) {
         file, _ := os.OpenFile("prime.txt", syscall.O_APPEND, 1)
         defer file.Close()
         defer close(final)
         writer := bufio.NewWriter(file)
         n := 500
         for {
             num := <-final
             if num == 10000 {
                 exitchan <- true
                 break
             }
             writer.WriteString(strconv.Itoa(num))
             writer.WriteString(",")
             // 多个协程共同工作,所以返回数不是递增的,所以这种分行方式非常不好用
             if num > n {
                 writer.WriteString("\n")
                 n = num + 500
             }
             writer.Flush()

         }
     }(final, exitchan)

     // 读出5个数据后,代表所有协程都工作完毕,可以关闭程序
     for v := 1; v < 6; v++ {
         <-exitchan
     }
     close(exitchan)
     fmt.Println("程序结束.")
 }

 func countprime(intchan chan int, final chan int, exit chan bool, t int) {
 ISPRIME:
     for {
         num, ok := <-intchan
         if !ok {
             break
         }
         if num == 10000 {
             final <- 10000
             close(intchan)
             break
         } else {
             num2 := num
             for i := 2; i < num2; i++ {
                 //降低时间复杂度,正经判断素数不用这种方式
                 num2 = num / i
                 if num%i == 0 {
                     continue ISPRIME
                 }
             }
             final <- num
         }
     }
     fmt.Println(t, "号协程完成工作")
     exit <- true
 }

缺陷:

1.  判断素数的方法很low,有更好的数学方法降维打击

2.  结束程序的方式很差,有可能(几乎必然)漏掉末尾几个素数无法录入