golang协程工作池处理多任务示例

  1. 工作方法实现
Go 复制代码
// 工作线程
// id : 线程号
// jobs : 任务通道 (chan)
// results: 完成结果通道 (chan)
func worker(id int, jobs <-chan int, results chan<- int) {
	//遍历任务
	for j := range jobs {
		fmt.Println("工作协程: ", id, "启动任务: ", j)
		fmt.Println(">>>>>>休眠2秒,模拟工作处理数据中...")
		time.Sleep(time.Second)
		fmt.Println("工作协程: ", id, "结束任务", j)
		results <- j * 2 //任务完成后写入结果到通道
	}
}

2.创建任务通道与任务执行结果通道

Go 复制代码
const numJobs = 5                  //通道容量
	jobarr := make(chan int, numJobs)  //任务通道
	results := make(chan int, numJobs) //任务执行结果通道

3.创建协程工作池

Go 复制代码
//启动3个工作协程
	for w := 1; w <= 3; w++ {
		go worker(w, jobarr, results)
	}

4.向工作通道发送任务

Go 复制代码
//向任务通道发送5个任务
	for j := 1; j <= numJobs; j++ {
		jobarr <- j //发送任务到任务通道
	}
	close(jobarr) //关闭任务通道

5.遍历执行结果:

Go 复制代码
//遍历执行结果
	for a := 1; a <= numJobs; a++ {
		<-results //读取通道数据不做处理
	}

6.完整示例

Go 复制代码
// 工作线程
// id : 线程号
// jobs : 任务通道 (chan)
// results: 完成结果通道 (chan)
func worker(id int, jobs <-chan int, results chan<- int) {
	//遍历任务
	for j := range jobs {
		fmt.Println("工作协程: ", id, "启动任务: ", j)
		fmt.Println(">>>>>>休眠2秒,模拟工作处理数据中...")
		time.Sleep(time.Second)
		fmt.Println("工作协程: ", id, "结束任务", j)
		results <- j * 2 //任务完成后写入结果到通道
	}
}


const numJobs = 5                  //通道容量
	jobarr := make(chan int, numJobs)  //任务通道
	results := make(chan int, numJobs) //任务执行结果通道
	//启动3个工程协程
	for w := 1; w <= 3; w++ {
		go worker(w, jobarr, results)
	}
	//向任务通道发送5个任务
	for j := 1; j <= numJobs; j++ {
		jobarr <- j //发送任务到任务通道
	}
	close(jobarr) //关闭任务通道
	//遍历执行结果
	for a := 1; a <= numJobs; a++ {
		<-results //读取通道数据不做处理
	}

输出结果:

Go 复制代码
工作协程:  3 启动任务:  1
>>>>>>休眠2秒,模拟工作处理数据中...
工作协程:  2 启动任务:  2
>>>>>>休眠2秒,模拟工作处理数据中...
工作协程:  1 启动任务:  3
>>>>>>休眠2秒,模拟工作处理数据中...
工作协程:  2 结束任务 2
工作协程:  2 启动任务:  4
>>>>>>休眠2秒,模拟工作处理数据中...
工作协程:  3 结束任务 1
工作协程:  3 启动任务:  5
>>>>>>休眠2秒,模拟工作处理数据中...
工作协程:  1 结束任务 3
工作协程:  3 结束任务 5
工作协程:  2 结束任务 4

执行流程示例图:

相关推荐
向上的车轮14 分钟前
无需云服务的家庭相册:OpenHarmony 上的 Rust 实践
开发语言·后端·rust
豐儀麟阁贵1 小时前
4.5数组排序算法
java·开发语言·数据结构·算法·排序算法
Jane-6667772 小时前
C语言——栈与队列
c语言·开发语言
“抚琴”的人2 小时前
C# 取消机制(CancellationTokenSource/CancellationToken)
开发语言·c#·wpf·1024程序员节·取消机制
Halo_tjn2 小时前
Java Map集合
java·开发语言·计算机
lsx2024062 小时前
DOM 创建节点
开发语言
richxu202510013 小时前
C语言<<超全.超重要>>知识点总结
c语言·开发语言
2501_916008893 小时前
用多工具组合把 iOS 混淆做成可复用的工程能力(iOS混淆|IPA加固|无源码混淆|Ipa Guard|Swift Shield)
android·开发语言·ios·小程序·uni-app·iphone·swift
胎粉仔3 小时前
Swift 初阶 —— inout 参数 & 数据独占问题
开发语言·ios·swift·1024程序员节
Lizhihao_3 小时前
Python如何写Selenium全攻略
开发语言·python