Go语言中的线程池是一种用于管理并发执行任务的设计模式。
线程池的主要目的是减少创建和销毁线程的开销,提高系统性能。 在Go语言中,线程池通常使用goroutine和channel来实现。
以下是一个简单的Go线程池实现:
package main
import (
"fmt"
"sync"
)
type Job func()
type Worker struct {
JobQueue chan Job
}
func NewWorker() Worker {
return Worker{JobQueue: make(chan Job)}
}
func (w Worker) Run(wg *sync.WaitGroup) {
go func() {
for job := range w.JobQueue {
job()
wg.Done()
}
}()
}
type Pool struct {
JobQueue chan Job
WorkerQueue chan chan Job
}
func NewPool(maxWorkers int) *Pool {
workerQueue := make(chan chan Job, maxWorkers)
jobQueue := make(chan Job)
return &Pool{JobQueue: jobQueue, WorkerQueue: workerQueue}
}
func (p *Pool) Run() {
for i := 0; i < cap(p.WorkerQueue); i++ {
worker := NewWorker()
worker.Run(&sync.WaitGroup{})
p.WorkerQueue <- worker.JobQueue
}
go func() {
for job := range p.JobQueue {
workerJobQueue := <-p.WorkerQueue
workerJobQueue <- job
}
}()
}
func main() {
pool := NewPool(5)
pool.Run()
wg := sync.WaitGroup{}
wg.Add(10)
for i := 0; i < 10; i++ {
job := func() {
fmt.Println("Executing job")
}
pool.JobQueue <- job
}
wg.Wait()
}
说明
在这个例子中,我们创建了 Worker结构体,它包含一个JobQueue通道,用于接收任务。 NewWorker函数用于创建一个新的Worker实例。 Run方法启动一个goroutine,监听JobQueue通道,当有任务到来时,执行任务并通知WaitGroup。
Pool结构体包含一个JobQueue通道和一个WorkerQueue通道。 JobQueue用于接收任务,WorkerQueue用于存储空闲的Worker。 NewPool函数用于创建一个新的Pool实例。 Run方法启动指定数量的 Worker并将它们添加到 WorkerQueue中。然后启动一个goroutine,监听 JobQueue通道,当有任务到来时,从 WorkerQueue中取出一个空闲的 Worker,将任务分配给它。
在main函数中,我们创建一个包含5个Worker的线程池,并向线程池提交10个任务。 使用WaitGroup来等待所有任务完成。
小结有需要的小伙伴可以加以修改使用
到此这篇关于GO workPool的线程池实现的文章就介绍到这了,更多相关GO workPool线程池内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网!