Golang中协程（channel+sync.WaitGroup）实现
package main
import (
"fmt"
"sync"
"time"
)
/*
设计个程序，要求：
1、将1-120000数字存入channel1。
2、1-120000数字中为素数的计算出来并且存入channel2。
3、打印这些素数。
注意：三个任务要求同步进行即开启多协程
*/
var wg = new(sync.WaitGroup)
//存入1-120000数字的函数
func storeNum(channel1 chan int) {
for i := 1; i <= 120000; i++ {
channel1 <- i
}
close(channel1)//每次利用for range遍历channel时，得将channel关闭
wg.Done()
}
//计算出素数，存入channel2的函数
func calculateNum(channel1, channel2 chan int, channel3 chan bool) {
for val := range channel1 {
flag := true
for i := 2; i < val; i++ {
if val%i == 0 {
flag = false
}
}
if flag {
channel2 <- val
}
}
channel3 <- true
wg.Done()
}
//打印素数的函数
func printNum(channel2 chan int) {
for val := range channel2 {
fmt.Println(val)
}
wg.Done()
}
func main() {
start := time.Now().Unix()
channel1 := make(chan int, 2000)
channel2 := make(chan int, 5000)
channel3 := make(chan bool, 6)
wg.Add(1)
go storeNum(channel1)
for i := 1; i <= 6; i++ {
wg.Add(1)
go calculateNum(channel1, channel2, channel3)
}
wg.Add(1)
go printNum(channel2)
wg.Add(1)
go func() {
for i := 1; i <= 6; i++ {
<-channel3
}
close(channel2)//每次利用for range遍历channel时，得将channel关闭
wg.Done()
}()
wg.Wait()
end := time.Now().Unix()
fmt.Printf("执行时间为%d\n", end-start)
}