第9篇 rabbitmq WorkPool源码解析

最新推荐文章于 2025-07-20 09:08:10 发布

原创

最新推荐文章于 2025-07-20 09:08:10 发布 · 445 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#rabbitmq #mq #workpool #源码 #biconsumer

消费者消费消息它有一个工作池的概念，所以现在研究一下WorkPool的源码，看一看可以学到什么技巧。

这个实现是一个线程安全的类
WorkPool<K,W> K ：客户端类型， W : 工作项目类型

1、成员变量

MAX_QUEUE_LENGTH = 1000 队列最大为1000
ready = new SetQueue() 准备好客户端
inProgress = new HashSet() 正在处理客户端
Map<K, VariableLinkedBlockingQueue> = new HashMap 真正的池子，包括注册客户端和它们工作队列
unlimited = new HashSet() 不限制队列大小
enqueueingCallback 入队回调函数（使用BiConsumer)

2、构造函数

定义enqueueingCallback 的函数，入队工作项

public WorkPool(final int queueingTimeout) {
       
       
        if (queueingTimeout > 0) {
       
       
            this.enqueueingCallback = (queue, item) -> {
       
       
                try {
       
       
                    boolean offered = queue.offer(item, queueingTimeout, TimeUnit.MILLISECONDS);
                    if (!offered) {
       
       
                        throw new WorkPoolFullException("Could not enqueue in work pool after " + queueingTimeout + " ms.");
                    }
                } catch (InterruptedException e) {
       
       
                    Thread.