GO workPool

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函数中,我们创建一个包含5Worker的线程池,并向线程池提交10个任务。
使用WaitGroup来等待所有任务完成。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 本文是《Go系列教程》的第二十篇文章。 什么是缓冲区通道 我们前面所讲的大多数都是基本的非缓冲区通道。正如我们在通...
    大风过岗阅读 382评论 0 0
  • Channel 是什么? channel,通道,本质上是一个通信对象,goroutine 之间可以使用它来通信。从...
    癞痢头阅读 846评论 0 0
  • 并发是编程里面一个非常重要的概念,Go语言在语言层面天生支持并发,这也是Go语言流行的一个很重要的原因。 Go语言...
    羋学僧阅读 665评论 0 6
  • Go语言中的并发编程 并发是编程里面一个非常重要的概念,Go语言在语言层面天生支持并发,这也是Go语言流行的一个很...
    吴佳浩阅读 363评论 0 1
  • go并发编程入门到放弃 并发和并行 并发:一个处理器同时处理多个任务。 并行:多个处理器或者是多核的处理器同时处理...
    yangyunfeng阅读 588评论 0 2