Go 学习笔记（8）：生产者消费者模型

package main

import (
    "fmt"
    "sync"
    "time"
)

var (
    // 定义共享数据的通道
    itemCh = make(chan int, 20)
    cwg    sync.WaitGroup
)

func producter(ch chan int) {
    for i := 0; i < 10; i++ {
        time.Sleep(time.Second) // 模拟真实场景的处理时间
        ch <- i
        fmt.Printf("set num %d to ch\n", i)
    }
    close(ch)
}

func consumer(ch chan int, name string) {
    defer cwg.Done()
    for n := range ch {
        time.Sleep(time.Second * 2) // 模拟真实场景的处理时间
        fmt.Printf("%s:get num %d from ch\n", name, n)
    }
}

func main() {
    go producter(itemCh)

    // 启动多个消费者
    for i := 0; i < 5; i++ {
        go consumer(itemCh, fmt.Sprintf("c-%d", i))
        cwg.Add(1)
    }

    cwg.Wait()

    fmt.Println("main done!")
}

package main

import (
    "fmt"
    "sync"
    "time"
)

var (
    // 定义共享数据的通道
    itemCh = make(chan int, 20)
    // 分别定义生产者和消费者的计数器
    pwg sync.WaitGroup
    cwg sync.WaitGroup
)

func producter(num int, ch chan int, name string) {
    defer pwg.Done()
    time.Sleep(time.Second) // 模拟真实场景的处理时间
    ch <- num
    fmt.Printf("%s:set num %d to ch\n", name, num)
}

func consumer(ch chan int, name string) {
    defer cwg.Done()
    for n := range ch {
        time.Sleep(time.Second * 2) // 模拟真实场景的处理时间
        fmt.Printf("%s:get num %d from ch\n", name, n)
    }
}

func main() {
    // 启动多个生产者
    for i := 0; i < 10; i++ {
        go producter(i, itemCh, fmt.Sprintf("p-%d", i))
        pwg.Add(1)
    }

    // 启动多个消费者
    for i := 0; i < 5; i++ {
        go consumer(itemCh, fmt.Sprintf("c-%d", i))
        cwg.Add(1)
    }

    // 等待生产者执行结束前必须让消费者和生产者都运行起来
    pwg.Wait()
    close(itemCh) // 等所有生产者执行结束就关闭通道

    cwg.Wait()

    fmt.Println("main done!")
}