Go语言实现一个协程池

2020-06-18 17:02:59 阅读：267 来源： 互联网

标签：Task 语言 chan worker 协程池任务 func Go Pool

协程的好处：

跨平台
跨体系架构
无需线程上下文切换的开销
无需原子操作锁定及同步的开销。

代码实现

package main

import (
	"fmt"
	"time"
)
//定义一个任务类型Task
type Task struct {
	f func()error //一个task中有具体的业务，业务名叫f
}
//创建一个任务,包装成一个任务
func NewTask(arg_f func()error) *Task{
	t:=Task{
		f:arg_f,
	}
	return &t
}
//Task执行业务的方法
func(t *Task)Execute(){
	t.f()//调用任务中已经绑定好的业务方法
}
//定义协程池类型
type Pool struct {
	EntryChannel chan *Task//对外的Task入口
	JobChannel chan *Task //协成池中的最大worker数量
	worker_num int
}
func NewPool(cap int) *Pool{
	//创建一个Pool
	p:=Pool{
	EntryChannel:make(chan *Task),
	JobChannel:make(chan *Task),
	worker_num:cap,
	}
	return &p
}
//创建一个work去工作
func(p *Pool)worker(worker_ID int){
	//永久的jobchannel取任务
	for task:=range p.JobChannel{//去消息列中取任务
		//执行任务
	task.Execute()
	fmt.Println("work ID",worker_ID,"执行完了一个任务")
	}
}

func(p *Pool)run(){
	//根据work_num来创建worker工作
	//从entrychananel去工作
	for i:=0;i<p.worker_num;i++{
		//每个worker都应该是一个goroutioe
		go p.worker(i)
	}
	for task:=range p.EntryChannel{//不断从入口取任务,将取到的发送给Job
		p.JobChannel<-task
	}
}

func main() {
	//创建一些任务
	t := NewTask(func() error {
		fmt.Println(time.Now())
		return nil

	})
	//创建一个Pool池
	p := NewPool(4)
	task_num:=0
	go func() {
		for {
			p.EntryChannel <- t
			task_num+=1
			fmt.Println("当前一共执行了",task_num,"个任务")
		}
	}()
p.run()
}

标签：Task,语言,chan,worker,协程池,任务,func,Go,Pool
来源： https://blog.csdn.net/weixin_43318506/article/details/106816721

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

Go语言实现一个协程池

协程的好处：

代码实现