golang线程池怎么实现

   2025-02-15 8580
核心提示:在Go语言中,可以使用goroutine和channel来实现线程池。下面是一个简单的线程池实现示例:package mainimport (fmtsync)type Thr

在Go语言中,可以使用goroutinechannel来实现线程池。下面是一个简单的线程池实现示例:

package mainimport ("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 := ipool.AddTask(func() {fmt.Printf("Task %d is running\n", taskNum)})}// 等待所有任务完成pool.Wait()}

在上面的示例中,ThreadPool结构体表示线程池,包含workerNum表示工作线程的数量,taskQueue表示任务队列,wgsync.WaitGroup用于等待所有任务完成。

NewThreadPool函数用于创建一个新的线程池,Start方法启动线程池中的工作线程,worker方法用于执行任务,AddTask方法用于往线程池中添加任务,Wait方法用于等待所有任务完成。

main函数中,创建一个线程池并启动,然后循环添加任务到线程池中,最后调用Wait方法等待所有任务完成。

当运行上述代码时,你会看到输出结果中的任务是并发执行的,最多同时执行3个任务,这就是线程池的作用。

 
 
更多>同类维修知识
推荐图文
推荐维修知识
点击排行
网站首页  |  关于我们  |  联系方式  |  用户协议  |  隐私政策  |  网站留言