golang的生产者消费者模型示例

golang的生产者消费者模型示例

生产者消费者模型是非常基础的算法，而且golang提供了chan类型，可以很方便的实现。但看了很多golang的示例，代码实现都比较ugly，一类没有好的退出方式，一类是用Sleep盲等事件完成，还有一类引入了sync包来实现事件通知。

参考了golang的官方文档，只使用chan就可以实现生产者和消费者之间的数据和状态同步。

首先，了解chan操作接口和语意，其实就是send、recv和close，类似于文件操作而没有open，文件read通过EOF来返回文件尾，而chan则可以通过额外的返回值(ok)来获取close的信息。

ch <- v       // Send v to channel ch.
v := <-ch     // Receive from ch, and assign value to v.
v, ok := <-ch // Receive from ch, and assign ok to false if closed.

func close(c chan<- Type)

然后，结合goroutine的应用，梳理代码要点。

• 通过chan在生产者和消费者之间传递数据(ch)和同步状态(done)；
• chan作为参数传递时是引用传递，不需要使用指针；
• chan是协程安全的，多个goroutine之间不需要锁；
• chan的close事件可以被recv获取，close事件一定在正常数据读完之后，机制类似于read到EOF；

最后，编写代码并运行。生产者和消费者是并行工作的，而且在工作完成后通过chan同步状态。

package main

import "fmt"

func Producer(ch chan int) {
	for i := 1; i <= 10; i++ {
		ch <- i
	}
	close(ch)
}

func Consumer(id int, ch chan int, done chan bool) {
	for {
		value, ok := <-ch
		if ok {
			fmt.Printf("id: %d, recv: %d\n", id, value)
		} else {
			fmt.Printf("id: %d, closed\n", id)
			break
		}
	}
	done <- true
}

func main() {
	ch := make(chan int, 3)

	coNum := 2
	done := make(chan bool, coNum)
	for i := 1; i <= coNum; i++ {
		go Consumer(i, ch, done)
	}

	go Producer(ch)

	for i := 1; i <= coNum; i++ {
		<-done
	}
}

运行结果：

$ ./chan2 
id: 2, recv: 1
id: 1, recv: 2
id: 1, recv: 4
id: 1, recv: 5
id: 1, recv: 6
id: 2, recv: 3
id: 2, recv: 8
id: 2, recv: 9
id: 2, recv: 10
id: 2, closed
id: 1, recv: 7
id: 1, closed