GO workPool的線程池實現(xiàn)
Go語言中的線程池是一種用于管理并發(fā)執(zhí)行任務(wù)的設(shè)計模式。
線程池的主要目的是減少創(chuàng)建和銷毀線程的開銷,提高系統(tǒng)性能。 在Go語言中,線程池通常使用goroutine和channel來實現(xiàn)。
以下是一個簡單的Go線程池實現(xiàn):
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()
}說明
在這個例子中,我們創(chuàng)建了 Worker結(jié)構(gòu)體,它包含一個JobQueue通道,用于接收任務(wù)。 NewWorker函數(shù)用于創(chuàng)建一個新的Worker實例。 Run方法啟動一個goroutine,監(jiān)聽JobQueue通道,當(dāng)有任務(wù)到來時,執(zhí)行任務(wù)并通知WaitGroup。
Pool結(jié)構(gòu)體包含一個JobQueue通道和一個WorkerQueue通道。 JobQueue用于接收任務(wù),WorkerQueue用于存儲空閑的Worker。 NewPool函數(shù)用于創(chuàng)建一個新的Pool實例。 Run方法啟動指定數(shù)量的 Worker并將它們添加到 WorkerQueue中。然后啟動一個goroutine,監(jiān)聽 JobQueue通道,當(dāng)有任務(wù)到來時,從 WorkerQueue中取出一個空閑的 Worker,將任務(wù)分配給它。
在main函數(shù)中,我們創(chuàng)建一個包含5個Worker的線程池,并向線程池提交10個任務(wù)。 使用WaitGroup來等待所有任務(wù)完成。
小結(jié)有需要的小伙伴可以加以修改使用
到此這篇關(guān)于GO workPool的線程池實現(xiàn)的文章就介紹到這了,更多相關(guān)GO workPool線程池內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
Go語言服務(wù)器開發(fā)之簡易TCP客戶端與服務(wù)端實現(xiàn)方法
這篇文章主要介紹了Go語言服務(wù)器開發(fā)之簡易TCP客戶端與服務(wù)端實現(xiàn)方法,實例分析了基于Go語言實現(xiàn)的簡易服務(wù)器的TCP客戶端與服務(wù)器端實現(xiàn)技巧,需要的朋友可以參考下2015-02-02
Go語言基礎(chǔ)語法之結(jié)構(gòu)體及方法詳解
結(jié)構(gòu)體類型可以用來保存不同類型的數(shù)據(jù),也可以通過方法的形式來聲明它的行為。本文將介紹go語言中的結(jié)構(gòu)體和方法,以及“繼承”的實現(xiàn)方法2021-09-09

