[Golang]实现一个带有等待和超时功能的协程池 - 类似Java中的ExecutorService接口实现

最新推荐文章于 2024-08-29 12:26:46 发布

原创

最新推荐文章于 2024-08-29 12:26:46 发布 · 1.6k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#golang #并发编程

本文介绍了如何在Golang中实现一个类似Java ExecutorService接口功能的协程池，支持指定任务队列长度、工作协程数量，并能启动、停止以及设置等待时间。核心数据结构包括协程池、工作协程和任务。通过加锁确保任务计数正确，使用select处理多情况下的信号优先级，并详细解释了停止协程池时避免遗漏工作协程的策略。

对于支持CSP并发编程模型的Golang而言，要实现一个协程池是非常简单的。对于习惯了基于线程模型完成并发编程的开发同学，可能初次接触会有点难理解，但是俗话说"书读百遍其义自见"，百来行的代码也并不多。

我们的目标是实现一个具有以下特性的协程池(熟悉Java的话，基本上就是实现了ExecutorService接口中的主要方法)：

能够指定任务队列长度和工作协程的数量使用任务队列
能够支持启动和停止
能够等待投递到其中的任务执行完毕(等待时间可以指定)

核心的三个数据结构：

协程池
工作协程
任务

直接上代码：

package pool

import (
   "sync"
   "time"
)

const (
   defaultWorkerQueueLength = 10   // 默认工作协程数量
   defaultJobQueueLength    = 1000 // 默认任务队列长度
)

type Job func()

type TimeoutPool struct {
   workerQueue chan *worker
   jobQueue    chan Job
   jobCount    int
   jobRet      chan struct{}
   stop        chan struct{}
   terminated  chan struct{}
   lock        sync.Mutex
}

// 初始化一个带有执行超时时间的协程池，协程数量：10；任务队列长度1000
func NewTimeoutPoolWithDefaults() *TimeoutPool {
   return NewTimeoutPool(defaultWorkerQueueLength, defaultJobQueueLength)
}

// 初始化一个带有执行超时时间的协程池，指定worker数量以及任务队列长度
func NewTimeoutPool(workerQu

最低0.47元/天解锁文章