使用Go生成素数的代码

使用Go生成素数

package main

import (
	"fmt"
	"os"
	"strconv"
)

func generate() chan int {
	ch := make(chan int)
	go func() {
		for i := 2; ; i++ {
			ch <- i
		}
	}()

	return ch
}

func filter(in chan int, prime int) chan int {
	out := make(chan int)
	go func() {
		for {
			if i := <-in; i%prime != 0 {
				out <- i
			}
		}
	}()

	return out
}

func sieve() chan int {
	out := make(chan int)
	go func() {
		ch := generate()
		for {
			prime := <-ch
			out <- prime
			ch = filter(ch, prime)
		}
	}()
	return out
}

func isPrime(n int) bool {
	for i := 2; i < n; i++ {
		if n%i == 0 {
			return false
		}
	}

	return true
}

func main() {
	if len(os.Args) > 1 {
		n := os.Args[1]
		fmt.Printf("input=%v\n", n)
		n2, err := strconv.Atoi(n)
		if err != nil {
			panic(err)
		} else {
			fmt.Printf("n2=%d isPrime=%v", n2, isPrime(n2))
		}
	} else {
		primes := sieve()
		for {
			fmt.Println(<-primes)
		}
	}
}

备注

用了并发协程处理，大部分都是对的，但很奇怪有些生成的素数，不是素数。

// 检查是否是素数
func isPrime(n int) bool {
	for i := 2; i < n; i++ {
		if n%i == 0 {
			return false
		}
	}

	return true
}