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