GO workPool的線程池實(shí)現(xiàn)

更新時(shí)間：2023年03月24日 09:24:14 作者：吳佳浩

本文主要介紹了GO workPool的線程池實(shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

Go語(yǔ)言中的線程池是一種用于管理并發(fā)執(zhí)行任務(wù)的設(shè)計(jì)模式。

線程池的主要目的是減少創(chuàng)建和銷毀線程的開(kāi)銷，提高系統(tǒng)性能。在Go語(yǔ)言中，線程池通常使用goroutine和channel來(lái)實(shí)現(xiàn)。

以下是一個(gè)簡(jiǎn)單的Go線程池實(shí)現(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()
}

說(shuō)明

在這個(gè)例子中，我們創(chuàng)建了 Worker結(jié)構(gòu)體，它包含一個(gè)JobQueue通道，用于接收任務(wù)。 NewWorker函數(shù)用于創(chuàng)建一個(gè)新的Worker實(shí)例。 Run方法啟動(dòng)一個(gè)goroutine，監(jiān)聽(tīng)JobQueue通道，當(dāng)有任務(wù)到來(lái)時(shí)，執(zhí)行任務(wù)并通知WaitGroup。

Pool結(jié)構(gòu)體包含一個(gè)JobQueue通道和一個(gè)WorkerQueue通道。 JobQueue用于接收任務(wù)，WorkerQueue用于存儲(chǔ)空閑的Worker。 NewPool函數(shù)用于創(chuàng)建一個(gè)新的Pool實(shí)例。 Run方法啟動(dòng)指定數(shù)量的 Worker并將它們添加到 WorkerQueue中。然后啟動(dòng)一個(gè)goroutine，監(jiān)聽(tīng) JobQueue通道，當(dāng)有任務(wù)到來(lái)時(shí)，從 WorkerQueue中取出一個(gè)空閑的 Worker，將任務(wù)分配給它。