Go语言实现一个协程池

协程的好处:

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

代码实现

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

你可能感兴趣的:(go语言)