Go 语言入门三部曲（三）：能用 Go 写“生产-消费者”模型

@[toc]

到最后一篇了，让我们再回顾一下这三篇的设定。
第一篇打基础，第二篇学以致用，查缺补漏，第三篇是我认为的 Go 语言能打的部分：协程 的应用。
我对 Go 的理解没那么深，能让我一个小白对 Go 感兴趣，一是云原生工作需要，二就是它能轻易的调动协程。不然它还真不能把我的目光从 C++ 身上挪过去一点。

package main

import (
    "fmt"
    "sync"
)

// 模拟订单对象
type OrderInfo struct {
    id int
}

var num int
var wg sync.WaitGroup

// 生产订单--生产者
func producer(out chan<- OrderInfo) {
    // 业务生成订单
    for i := 0; i < 10000; i++ {
        order := OrderInfo{id: i + 1}
        fmt.Println("生成订单，订单ID为：", order.id)
        out <- order // 写入channel
    }
    // 如果不关闭，消费者就会一直阻塞，等待读
    close(out) // 订单生成完毕，关闭channel
}

func producerDispatch(tasks chan OrderInfo) {

    producer(tasks)
}

// 处理订单--消费者
func consumer(i int, in <-chan OrderInfo) {
    defer wg.Done()
    // 从channel读取订单，并处理
    for order := range in {
        fmt.Println("消费者", i, "读取订单,订单ID为:", order.id)
        num++
    }
}

func main() {
    //消费者个数
    var channelLen = 50

    tasks := make(chan OrderInfo, channelLen)

    go producerDispatch(tasks)

    go consumer(1, tasks)
    wg.Add(1)
    go consumer(2, tasks)
    wg.Add(1)
    go consumer(3, tasks)
    wg.Add(1)
    go consumer(4, tasks)
    wg.Add(1)

    wg.Wait()
    //defer close(tasks)
    fmt.Println("all done")
    fmt.Println(num)
}