go协程池

发表于 更新于 分类于 本文字数： 2.1k 阅读时长 ≈ 4 分钟

目的学习go协程池

  • 网上go协程池大同小异，理解原理就可以了，我是根据bilibili上一个视频操作的，讲的还挺好的，自行搜索下，不过因为这个视频中任务队列是没有长度限制的，所以在那个基础上进行改编。

看下协程池的原理

avatar

协程池结构体,并且建立有缓存的channel

1
2
3
4
5
6
7
8
9
10
11
12
type Workpool struct {
	Jobchannel chan *Task
	Workernum int
}
// maxsize 工人多少个，maxlen 多长的管道
func NewWorkpool(maxsize int,maxlen int) *Workpool {
	w := Workpool{
		Jobchannel:      make(chan *Task,maxlen),
		Workernum:       maxsize,
	}
	return &w
}

Task 任务，这里我是派发了十个任务，然后休息两秒，然后再派发

1
2
3
4
5
func (t *Task) Execute(){
	fmt.Println("任务id为：",t.Id,",当前时间为：",time.Now())
	t.f()
	time.Sleep(2 * time.Second)
}

完整代码如下

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
package cache

import (
	"fmt"
	"sync"
	"time"
)

type Task struct {
	f func() error
	Id int
}
func Newtask(arg_f func() error,id int) *Task{
	t :=Task{
		f:	arg_f,
		Id:	id,
	}
	return &t
}
func (t *Task) Execute(){
	fmt.Println("任务id为：",t.Id,",当前时间为：",time.Now())
	t.f()
	time.Sleep(2 * time.Second)
}
type Workpool struct {
	Jobchannel chan *Task
	Workernum int
}
// maxsize 工人多少个，maxlen 多长的管道
func NewWorkpool(maxsize int,maxlen int) *Workpool {
	w := Workpool{
		Jobchannel:      make(chan *Task,maxlen),
		Workernum:       maxsize,
	}
	return &w
}


func (p *Workpool) AllocateJobTask(maxnumjobs int) {
	for i:=0;i<maxnumjobs;i++{
		t := Newtask(func() error {
			//任务是打印当前时间
			//fmt.Println("任务id为：",i,",当前时间为：",time.Now())
			return nil
		},i)
		p.Jobchannel <- t
	}
	close(p.Jobchannel)
}
func (p *Workpool) CreateWorkPool(){
	var wg sync.WaitGroup
	for i:=0; i<p.Workernum;i++  {
		go p.Work(&wg,i)
		wg.Add(1)
	}
	wg.Wait()

}
func (p *Workpool) Work(wg *sync.WaitGroup,id int){
	for job :=range p.Jobchannel{
		job.Execute()
		fmt.Println("我是工人:",id,"正在工作job:",job.Id)
	}
	wg.Done()
}

func GetCacheChannel(){
	startTime :=time.Now()
	w :=NewWorkpool(10,10)
	maxnumjobs :=100
	//先初始化线程池
	go w.AllocateJobTask(maxnumjobs)
	w.CreateWorkPool()

	endTime := time.Now()
	diff := endTime.Sub(startTime)
	fmt.Println("total time taken ", diff.Seconds(), "seconds")
}

写完这段代码，网上还有其他协程池的原理，大同小异