阅读量:1
在Go语言中，可以使用goroutine和channel来实现线程池。下面是一个简单的线程池实现示例：
package main import ( "fmt" "sync" ) type ThreadPool struct { workerNum int // 工作线程数量 taskQueue chan func() // 任务队列 wg sync.WaitGroup // 等待所有任务完成 } func NewThreadPool(workerNum int) *ThreadPool { return &ThreadPool{ workerNum: workerNum, taskQueue: make(chan func()), } } func (tp *ThreadPool) Start() { for i := 0; i < tp.workerNum; i++ { go tp.worker() } } func (tp *ThreadPool) worker() { for task := range tp.taskQueue { task() tp.wg.Done() } } func (tp *ThreadPool) AddTask(task func()) { tp.wg.Add(1) tp.taskQueue <- task } func (tp *ThreadPool) Wait() { tp.wg.Wait() } func main() { // 创建线程池 pool := NewThreadPool(3) // 启动线程池 pool.Start() // 添加任务到线程池 for i := 0; i < 10; i++ { taskNum := i pool.AddTask(func() { fmt.Printf("Task %d is running\n", taskNum) }) } // 等待所有任务完成 pool.Wait() } 在上面的示例中，ThreadPool结构体表示线程池，包含workerNum表示工作线程的数量，taskQueue表示任务队列，wg为sync.WaitGroup用于等待所有任务完成。
NewThreadPool函数用于创建一个新的线程池，Start方法启动线程池中的工作线程，worker方法用于执行任务，AddTask方法用于往线程池中添加任务，Wait方法用于等待所有任务完成。
在main函数中，创建一个线程池并启动，然后循环添加任务到线程池中，最后调用Wait方法等待所有任务完成。
当运行上述代码时，你会看到输出结果中的任务是并发执行的，最多同时执行3个任务，这就是线程池的作用。
