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

来源:互联网 发布:windows定时启动程序 编辑:程序博客网 时间:2024/06/08 02:23

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

package mainimport "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}

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

0 0