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

前端之家收集整理的这篇文章主要介绍了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

}

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

猜你在找的Go相关文章