golang编写一个简单的生成消费模式

生成消费模式大家在现实生活中经常遇到，如果使用Java编写可以使用阻塞队列如BlockingQueue，Python也有相应的组件，如果使用golang编写使用自带的channel管道更加简单。
下面直接看代码：

package main

import "fmt"

func consumer(ch chan int,done chan string)  {
    for i := range ch{
        fmt.Println("recive",i)
    }
    done<-"ok"

}
func product(ch chan int) {
    for i := 0; i<5; i++{
        ch<-i
    }
    close(ch)
}

func main() {
    fmt.Println("start")
    ch := make(chan int)
    done := make(chan string)
    go consumer(ch,done)
    go product(ch)

    <-done

}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

代码非常简单，分别用goroutine启动两个协程：生产和消费。共享ch，一个往管道中写数据一个往管道中读数据，done是为了让主进程等待消费者结束。