go 实现 生产者消费者模型

go 生产者消费者模型

package main

import (
	"log"
	"time"
)

// go 生产消费者模型
func main() {

	// 设置缓冲区大小
	ch := make(chan int, 5)

	startTime := time.Now().Unix()
	log.Println("start")
	go provide(ch)
	time.Sleep(time.Second * 3)

	for data := range ch {
		// 接收到的数据为-1, 则表示接收结束, 少加break 会发生deadlock
		if data == -1 {
			log.Println("comsume is over. I'm so full")
			break
		} else {
			log.Printf("comsume %d", data)
		}
	}
	end := time.Now().Unix()
	log.Println("end")
	spend := end - startTime
	log.Printf("spend %ds", spend)
}

func provide(ch chan int) {
	for i := 0; i < 11; i++ {
		// time.Sleep(time.Second * 3)
		// 缓冲区已满才会阻塞当前协程
		ch <- i
		log.Printf("provide %d", i)
	}
	ch <- -1
}

cook

• 与java实现的对比: [生产者消费者模式-Java实现]