Go 线程池实现案例

Go 语言并不像其他一些语言（例如 Java 或 C#）那样直接提供一个线程池的概念。相反，Go 使用 goroutines 来实现并发，它是一种比线程更轻量级的并发执行单元。不过，仍然可以实现一个类似线程池的结构，来管理和限制同时运行的 goroutines 的数量。以下是如何在 Go 中实现一个简单的类似线程池的功能的例子：

package mainimport ("fmt""sync""time"
)// WorkerPool 结构体定义
type WorkerPool struct {jobs    chan func()   // 用于接收任务的通道maxJobs int           // 最大并发任务数wg      sync.WaitGroup // 用于等待所有任务完成
}// NewWorkerPool 创建一个新的 WorkerPool
func NewWorkerPool(maxJobs int) *WorkerPool {pool := &WorkerPool{jobs:    make(chan func(), maxJobs),maxJobs: maxJobs,}return pool
}// Start 开始 WorkerPool 的工作
func (p *WorkerPool) Start() {for i := 0; i < p.maxJobs; i++ {go func() {for job := range p.jobs {job()}}()}
}// Submit 提交一个任务到 WorkerPool
func (p *WorkerPool) Submit(job func()) {p.wg.Add(1)p.jobs <- func() {defer p.wg.Done()job()}
}// Wait 等待所有任务完成
func (p *WorkerPool) Wait() {p.wg.Wait()close(p.jobs) // 关闭通道，停止接收新的任务
}func main() {// 创建一个最大并发数为 3 的工作池pool := NewWorkerPool(3)pool.Start()for i := 0; i < 10; i++ {i := i // 创建任务变量的本地副本pool.Submit(func() {fmt.Printf("Starting job %d\n", i)time.Sleep(2 * time.Second) // 模拟耗时操作fmt.Printf("Finished job %d\n", i)})}// 等待所有任务完成pool.Wait()fmt.Println("All jobs completed.")
}

在这个例子中，WorkerPool 结构体有一个 jobs 通道用于接收任务，一个 maxJobs 表示最大并发任务数，和一个 sync.WaitGroup 用于等待所有任务完成。Start 方法启动了 maxJobs 数量的 goroutines，每个 goroutine 不断地从 jobs 通道中接收并执行任务。Submit 方法用于提交新的任务到 jobs 通道，同时增加 WaitGroup 的计数。Wait 方法等待所有任务完成后关闭 jobs 通道。