怎样用Golang实现工作池模式 演示buffered channel管理并发任务

package main

import (
    "fmt"
    "sync"
    "time"
)

// Task 定义一个任务接口，所有要执行的任务都应该实现它
type Task interface {
    Execute()
}

// SimpleTask 一个简单的任务实现
type SimpleTask struct {
    ID int
}

// Execute 模拟任务执行
func (t *SimpleTask) Execute() {
    fmt.Printf("Worker %d processing task %d\n", t.ID%3, t.ID) // 模拟不同的worker处理
    time.Sleep(time.Millisecond * 100) // 模拟耗时操作
}

// WorkerPool 结构体定义工作池
type WorkerPool struct {
    tasks chan Task // 任务队列，缓冲通道
    wg    sync.WaitGroup
    workers int // 工作协程数量
}

// NewWorkerPool 创建一个新的工作池
func NewWorkerPool(workers, bufferSize int) *WorkerPool {
    return &WorkerPool{
        tasks: make(chan Task, bufferSize),
        workers: workers,
    }
}

// Start 启动工作池中的所有工作协程
func (wp *WorkerPool) Start() {
    for i := 0; i < wp.workers; i++ {
        go wp.worker(i)
    }
}

// worker 工作协程函数
func (wp *WorkerPool) worker(id int) {
    defer wp.wg.Done() // 每个worker退出时，WaitGroup计数减一

    for task := range wp.tasks { // 从任务通道中不断接收任务
        task.Execute()
    }
    fmt.Printf("Worker %d finished.\n", id)
}

// Submit 提交任务到工作池
func (wp *WorkerPool) Submit(task Task) {
    wp.wg.Add(1) // 每提交一个任务，WaitGroup计数加一
    wp.tasks <- task // 将任务发送到任务队列
}

// Wait 等待所有任务完成
func (wp *WorkerPool) Wait() {
    close(wp.tasks) // 关闭任务通道，通知所有worker没有新任务了
    wp.wg.Wait()    // 等待所有worker完成
}

func main() {
    // 创建一个有3个工作协程，任务队列缓冲大小为10的工作池
    pool := NewWorkerPool(3, 10)

    // 启动工作池
    pool.Start()

    // 提交100个任务
    for i := 1; i <= 100; i++ {
        pool.Submit(&SimpleTask{ID: i})
    }

    // 等待所有任务完成
    fmt.Println("All tasks submitted. Waiting for completion...")
    pool.Wait()
    fmt.Println("All tasks completed.")
}

goroutine

goroutine

pool.Wait()

close(wp.tasks)

for task := range wp.tasks

wg.Wait()

sync.WaitGroup

Add

Done

Wait

wg.Add(1)

Add

WaitGroup

Wait()

Add(1)

Done()

Wait()