golang 多协程处理任务

关注

原创最新推荐文章于 2025-07-22 16:02:41 发布 1.2k 阅读 AI 写同款· GEO 优化 ›

[Golang]实现一个带有等待和超时功能的协程池 - 类似Java中的ExecutorService接口实现对于支持CSP并发编程模型的Golang而言，要实现一个协程池是非常简单的。对于习惯了基于线程模型完成并发编程的开发同学，可能初次接触会有点难理解，但是俗话说"书读百遍其义自见"，百来行的代码也并不多。我们的目标是实现一个具有以下特性的协程池(熟悉Java的话，基本上就是实现了ExecutorService接口中的主要方法)：能够指定任务队列长度和工作协程的数量使用任务队列能够支持启动和停... 阅读详情

package main
 
import (
	"fmt"
	"runtime"
)
 
var workers = runtime.NumCPU()
 
type result struct {
	jobname    string
	resultcode int
	resultinfo string
}
 
type job struct {
	jobname string
	results chan<- result
}
 
func main() {
	jobnames := []string{"1", "2", "3", "4", "5", "6", "7", "8"}
	dorequest(jobnames)
}
 
func dorequest(jobnames []string) {
 
	// 定义需要的channels切片
	jobs := make(chan job, workers)
	results := make(chan result, len(jobnames))
	done := make(chan struct{}, workers)
 
 	/**
 	 * 把任务写入 JobCh通道 
 	 **/
 	go func( jobs chan <- job, jobnames[]string, results chan <- result){
 		for _, jobname := range jobnames {
			jobs <- job{jobname, results}
		}
		close(jobs)
 	}( jobs, jobnames, results)

 	/**
 	 * 开启n个协程处理任务
 	 **/
	for i := 0; i < workers; i++ {
		go func( done chan <- struct{}, jobs <- chan job){
			for job := range jobs {
				job.do()
			}
			done <- struct{}{}
		}( done, jobs)
	}
	
	/**
	 * 查看任务是否完成
	 **/
	go func( done <- chan struct{}, results chan result){
		for i := 0; i < workers; i++ {
			<-done
		}
		close(results)
	}( done, results)

	/**
	 * 取出结果
	 **/
	for result := range results {
		fmt.Printf("done: %s,%d,%s\n", result.jobname, result.resultcode, result.resultinfo)
	}
}
func (job job) do() {
	fmt.Printf("... doing work in [%s]\n", job.jobname)
 
	// 模拟处理结果
	if job.jobname == "8" {
		job.results <- result{job.jobname, 0, "ok"}
	} else {
		job.results <- result{job.jobname, -1, "error"}
	}
}