用Go的风格实现素数筛选

Posted 2020-09-29 丹江流

package main

import (
    "fmt"
    "time"
)

const End = 10000

func source(ch chan<- int) {
    for i := 2; i < End; i++ {
        ch <- i
    }
}

func validate(in <-chan int, out chan<- int, fix int) {
    for {
        select {
        case i := <-in:
            if i%fix != 0 {
                out <- i
            }
        case <-time.After(1 * time.Second):
            close(out)
            return
        }
    }
}

// 打印素数
func main() {

    // 逐个获取待检测的数据源
    ch := make(chan int)
    go source(ch)

    for {
        data, ok := <-ch
        if !ok {
            break
        }
        fmt.Println(data)
        out := make(chan int)
        go validate(ch, out, data)
        ch = out
    }
}

 参考:http://tonybai.com/2017/04/20/go-coding-in-go-way/ ， 但他实现的没有关闭chan, 导致 fatal error: all goroutines are asleep - deadlock! ，本方法优化了这个BUG.