golang快速实现协程池

最新推荐文章于 2024-08-03 09:52:51 发布

qq_42457856

最新推荐文章于 2024-08-03 09:52:51 发布

阅读量222

点赞数

分类专栏： golang 文章标签： golang

本文链接：https://blog.csdn.net/qq_42457856/article/details/109503979

版权

golang 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

如何使用golang快速实现协程池

1.下面将通过一个案例来学习协程池的调度和运作，废话不多说，直接开撸。

package main

import (
	"fmt"
	"math/rand"
	"time"
)

func main() {
	rand.Seed(time.Now().UnixNano())
	t := NewTask(func() error{
		fmt.Println(time.Now())
		return nil
	})
	p := NewPool(4)
	go func() {
		for{
			p.AddTask(t)
			time.Sleep(time.Second)
		}

	}()
	p.RunWorker()

	select {

	}

}

type Task struct {
	F func()error
}

func NewTask(f func()error)*Task  {
	return &Task{
		F:f,
	}
}

type Pool struct {
	MaxSize int  //协程数量大小
	Worker []chan *Task  //任务管道
}

func NewPool(maxSize int)*Pool  {
	return &Pool{
		MaxSize:maxSize,
		Worker:make([]chan *Task,maxSize),
	}
}

func (this *Pool)RunWorker()  {
	for i:=0;i<this.MaxSize;i++{
		this.Worker[i] = make(chan *Task,1024)

		go this.Start(i,this.Worker[i])
	}
}

func (this *Pool)Start(index int,tasks chan *Task)  {
	for task:= range tasks{
		fmt.Println("this is workerID:",index)
		task.F()
	}
}

func (this *Pool)AddTask(t *Task)  {

	index := rand.Intn(this.MaxSize)
	this.Worker[index] <- t
}