当前位置:   article > 正文

golang实现多生产者多消费者模式_go 多生产者 消费者

go 多生产者 消费者

多生产者、多消费者,利用通道传递数据,WaitGroup控制生产者、消费者执行结束。

package main

import (
	"fmt"
	"sync"
)

func main() {
	wgP := &sync.WaitGroup{}
	wgC := &sync.WaitGroup{}
	var num = 5
	ch := make(chan int, num)

	// 消费者
	for i := 0; i < num; i++ {
		wgC.Add(1)
		go Consumer(wgC, ch, i)
	}

	// 生产者
	for i := 0; i < num; i++ {
		wgP.Add(1)
		go Producter(wgP, ch)
	}

	wgP.Wait()

	// 生产者发送结束后, 要关闭通道,否则消费者会阻塞造成协程泄漏
	close(ch)

	wgC.Wait()

}

func Producter(wg *sync.WaitGroup, ch chan int) {
	defer wg.Done()
	for i := 0; i < 10; i++ {
		ch <- i
	}
}

func Consumer(wg *sync.WaitGroup, ch chan int, i int) {
	defer wg.Done()
	for item := range ch {
		fmt.Printf("%d号协程: %d\n", i+1, item)
	}
}

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35
  • 36
  • 37
  • 38
  • 39
  • 40
  • 41
  • 42
  • 43
  • 44
  • 45
  • 46
  • 47
  • 48

运行结果:

2号协程: 0
2号协程: 5
2号协程: 6
2号协程: 0
2号协程: 0
2号协程: 0
3号协程: 2
5号协程: 4
5号协程: 7
1号协程: 1
2号协程: 1
2号协程: 0
2号协程: 1
2号协程: 2
2号协程: 2
2号协程: 2
2号协程: 3
2号协程: 3
2号协程: 2
2号协程: 3
2号协程: 9
1号协程: 1
1号协程: 4
1号协程: 5
1号协程: 6
1号协程: 7
1号协程: 8
1号协程: 9
1号协程: 5
5号协程: 8
5号协程: 5
5号协程: 6
5号协程: 7
5号协程: 8
5号协程: 9
3号协程: 1
3号协程: 4
3号协程: 5
3号协程: 6
3号协程: 7
3号协程: 8
3号协程: 9
2号协程: 4
4号协程: 3
1号协程: 4
1号协程: 9
5号协程: 3
3号协程: 6
2号协程: 7
4号协程: 8
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35
  • 36
  • 37
  • 38
  • 39
  • 40
  • 41
  • 42
  • 43
  • 44
  • 45
  • 46
  • 47
  • 48
  • 49
  • 50
声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop】
推荐阅读
相关标签
  

闽ICP备14008679号