Golang协程池与工作模式实战教程

package main
<p>import (
"fmt"
"sync"
"time"
)</p><p>// Task 表示一个待处理的任务
type Task struct {
ID   int
Data int
}</p><p>// WorkerPool 管理worker和任务分发
type WorkerPool struct {
tasks   chan Task
workers int
wg      sync.WaitGroup
}</p><p>// NewWorkerPool 创建一个新的worker池
func NewWorkerPool(workers, queueSize int) *WorkerPool {
return &WorkerPool{
tasks:   make(chan Task, queueSize),
workers: workers,
}
}</p><p>// Start 启动所有worker
func (wp <em>WorkerPool) Start() {
for i := 0; i < wp.workers; i++ {
wp.wg.Add(1)
go func(workerID int) {
defer wp.wg.Done()
for task := range wp.tasks {
fmt.Printf("Worker %d processing task %d with data %d\n", workerID, task.ID, task.Data)
time.Sleep(time.Millisecond </em> 100) // 模拟处理时间
}
}(i)
}
}</p><p>// Submit 提交任务到池
func (wp *WorkerPool) Submit(task Task) {
wp.tasks <- task
}</p><p>// Stop 关闭任务通道，等待所有worker结束
func (wp *WorkerPool) Stop() {
close(wp.tasks)
wp.wg.Wait()
}</p><p>func main() {
pool := NewWorkerPool(3, 10) // 3个worker，队列最多10个任务
pool.Start()</p><pre class="brush:php;toolbar:false;">// 提交一批任务
for i := 1; i <= 5; i++ {
    pool.Submit(Task{ID: i, Data: i * 10})
}

pool.Stop()
fmt.Println("All tasks completed.")

}