Goroutine池

goroutine池的应用

• 本质上是生产者消费者模型

• 可以有效控制goroutine数量，防止暴涨

• 需求：

  • 计算一个数字的各个位数之和，例如数字123，结果为1+2+3=6
  • 随机生成数字进行计算
• 控制台输出结果如下：

job id: 164362 random: 3002688297310473558 result: 81
job id: 164363 random: 8188014692039052954 result: 84
job id: 164364 random: 9199514953162082256 result: 87
job id: 164365 random: 6547609431377245779 result: 96
job id: 164366 random: 5158862497998501304 result: 94
job id: 164367 random: 7591045689235318611 result: 84
job id: 164368 random: 4913457935486468007 result: 93
job id: 164369 random: 6484925446766292271 result: 94
job id: 164370 random: 1630566468428792798 result: 101
job id: 164371 random: 3999715154713986432 result: 96
job id: 164372 random: 8436839935373284876 result: 106
job id: 164373 random: 7590654756672405712 result: 88
job id: 164374 random: 5127823813978664887 result: 103
job id: 164375 random: 5630536624069526117 result: 77
job id: 164376 random: 3445557283367509019 result: 86
job id: 164377 random: 6087330610397339825 result: 83
job id: 164378 random: 3391465925899927215 result: 99

main.go

package main

import (
    "fmt"
    "math/rand"
)

type Job struct {
    Id int
    Random int
}

type Result struct {
    job *Job
    sum int
}

func createGoroutinePool(goNum int, jobChan chan *Job, resultChan chan *Result){
    // 根据goNum开启多个goroutine执行
    for i := 0; i < goNum; i++ {
        go func(jobChan chan *Job, resultChan chan *Result) {
            // 处理每个jobChan中的数据并返回结果
            for job := range jobChan{
                random := job.Random
                var sum int
                // 计算每一位的和
                for random != 0 {
                    tmp := random % 10
                    sum += tmp
                    random /= 10
                }
                // 组织返回结果
                r := &Result{
                    job: job,
                    sum: sum,
                }
                // 将结果写入通道
                resultChan<- r
            }
        }(jobChan, resultChan)
    }
}

func main() {
    // 工作通道
    jobChan := make(chan *Job, 128)
    // 结果通道
    resultChan := make(chan *Result,128)
    // 创建工作池执行任务
    createGoroutinePool(64, jobChan, resultChan)
    // 打印结果通道数据
    go func(resultChan chan *Result){
        for result := range resultChan{
            fmt.Printf("job id: %v random: %v result: %d\n", result.job.Id, result.job.Random, result.sum)
        }
    }(resultChan)

    var id int
    for {
        id++
        r_num := rand.Int()
        job := &Job{
            Id:     id,
            Random: r_num,
        }
        jobChan<-job
    }
}