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
	}
}