Go 中 实现基于 channel 的简单 生产者 消费者模型
package main
import (
"fmt"
"strconv"
"sync"
)
// 生产者
func producer(ch chan<- string, wg *sync.WaitGroup) {
defer wg.Done()
for i := 1; i <= 200; i++ {
ch <- "生产者_1 " + strconv.Itoa(i) // 将数据发送到 channel 中
}
// 以便消费者可以在读取完所有数据后停止阻塞并退出
close(ch) // 关闭 channel
}
// 生产者
func producer2(ch2 chan<- string, wg *sync.WaitGroup) {
defer wg.Done()
for i := 1; i <= 100; i++ {
ch2 <- "生产者_2 " + strconv.Itoa(i) // 将数据发送到 channel 中
}
// 以便消费者可以在读取完所有数据后停止阻塞并退出
close(ch2) // 关闭 channel
}
// 消费者
func consumer(ch <-chan string, wg *sync.WaitGroup, id int) {
defer wg.Done()
// 从 channel 中读取数据，直到 channel 关闭
for data := range ch {
fmt.Printf("消费者:%v 数据: %v\n", id, data)
}
}
func main() {
// 创建一个 channel
ch := make(chan string, 100)
ch2 := make(chan string, 100)
var wg sync.WaitGroup
wg.Add(9)
go producer(ch, &wg) // 启动生产者 goroutine
go producer2(ch2, &wg) // 启动生产者 goroutine
// 启动5个消费者 goroutine
go consumer(ch, &wg, 1)
go consumer(ch, &wg, 2)
go consumer(ch, &wg, 3)
go consumer(ch, &wg, 4)
go consumer(ch, &wg, 5)
go consumer(ch2, &wg, 6)
go consumer(ch2, &wg, 7)
// 使用 sync.WaitGroup 等待所有 goroutine 执行完毕
wg.Wait() // 等待所有 goroutine 结束
}
