golang并发模式之生产者消费者模型(来自《go语言高级编程》)

巫马嘉祯
2023-12-01

golang实现生产者消费者模型其实很简单:

package main

import (
	"fmt"
	"time"
)

//生产者:生成factor证书的序列
func Producer(factor int, out chan<- int) {
	for i := 0; ; i++ {
		out <- i * factor
	}
}

//消费者
func Consumer(in <-chan int) {
	for v := range in {
		fmt.Println(v)
	}
}

func main() {
	ch := make(chan int, 64) //成果队列
	go println(3, ch)        //生成3的倍数的序列
	go println(5, ch)        //生成5的倍数的序列
	go Consumer(ch)          //消费 生成队列

	//运行一定时间后推出
	time.Sleep(5 * time.Second)
}

我们开启了2个Produce生产流水线,分别生成3和5的倍数的序列。然后开启1个Consumer消费者线程,打印获取的结果。通过在main函数休眠一定的时间来让生产者和消费者工作一定的时间。这种考休眠的方式是无法保证稳定的输出结果。
	我们可以让main函数保存在阻塞状态不退出,之有当用户输入ctrl-c时才真正推出程序:
package main

import (
	"fmt"
	"os"
	"os/signal"
	"syscall"
)

//生产者:生成factor证书的序列
func Producer(factor int, out chan<- int) {
	for i := 0; ; i++ {
		out <- i * factor
	}
}

//消费者
func Consumer(in <-chan int) {
	for v := range in {
		fmt.Println(v)
	}
}

func main() {
	ch := make(chan int, 64) //成果队列
	go println(3, ch)        //生成3的倍数的序列
	go println(5, ch)        //生成5的倍数的序列
	go Consumer(ch)          //消费 生成队列

	//Ctrl+c
	sig := make(chan os.Signal, 1)
	signal.Notify(sig, syscall.SIGINT, syscall.SIGTERM)
	fmt.Println("quit(%v)\n", <-sig)
}

这个例子中有2个生产者,并且2个生产者之间并无同步时间可参考,他们是并发的。因此,消费者输出的结果序列的顺序是不确定的,这个没有问题,生产者和消费者可以回想配合工作。
 类似资料: