我们在消费RabbitMQ消息的过程中,有时候可能会想先暂停消费一段时间,然后过段时间再启动消费者,这个需求怎么实现呢?我们可以借助RabbitListenerEndpointRegistry这个类来实现,它的全类名是org.springframework.amqp.rabbit.listener.RabbitListenerEndpointRegistry,通过这个类可以实现全部队列消息的启动、停止消费,也可以实现指定队列消息的启动、停止消费。具体的原因感兴趣的话可以参考一下我前面的这篇博客(17)不重启服务动态调整RabbitMQ消费者数量,里面有相应的源码分析。
停止、启动全部队列消费
RabbitListenerEndpointRegistry类提供了start()方法和stop()方法,可以看到底层都是通过调用getListenerContainers()获取到所有队列的消费监听容器列表,然后遍历挨个调用对应的start()方法和stop()方法。
@Override
public void start() {
for (MessageListenerContainer listenerContainer : getListenerContainers()) {
startIfNecessary(listenerContainer);
}
}
@Override
public void stop() {
for (MessageListenerContainer listenerContainer : getListenerContainers()) {
listenerContainer.stop();
}
}
我们只需要获取到RabbitListenerEndpointRegistry对象,然后调用其start()方法和stop()方法即可实现启动/停止所有队列消费。
实现代码如下所示:
@Resource
RabbitListenerEndpointRegistry rabbitListenerEndpointRegistry;
@RequestMapping(value = "/startStopAllConsumer")
@ApiOperation(value = "启动/暂停全部队列消息消费")
public Response startStopAllConsumer(@RequestParam(value = "consumeSwitch", required = true) boolean consumeSwitch) {
log.info("启动/暂停全部队列消息消费,consumeSwitch:{}",consumeSwitch);
if(consumeSwitch){
rabbitListenerEndpointRegistry.start();
}else {
rabbitListenerEndpointRegistry.stop();
}
return Response.success();
}
传入开关参数为false,会停止所有队列消费者消费,调用后控制台看到如下日志
2023-09-04 19:43:11.480 +0800 [TID: N/A] [http-nio-8080-exec-4] INFO c.b.t.m.p.w.PayCashierMockController:67 – 启动/暂停全部队列消息消费,consumeSwitch:false
2023-09-04 19:43:11.服务器托管网556 +0800 [TID: N/A] [http-ni服务器托管网o-8080-exec-4] INFO o.s.a.r.l.SimpleMessageListenerContainer:586 – Waiting for workers to finish.
2023-09-04 19:43:12.352 +0800 [TID: N/A] [http-nio-8080-exec-4] INFO o.s.a.r.l.SimpleMessageListenerContainer:589 – Successfully waited for workers to finish.
可以看到消息监听容器关闭的日志,然后再传入开关参数为true,调用后会启动所有队列消息消费。
停止、启动指定队列消费
上面提到了RabbitListenerEndpointRegistry.getListenerContainers()可以获取到所有队列的消费监听容器列表,我们可以使用MessageListenerContainer中获取消费的队列名进行判断,以实现指定队列的停止、启动消费。
实现代码如下所示:
@Resource
RabbitListenerEndpointRegistry rabbitListenerEndpointRegistry;
@RequestMapping(value = "/startStopConsumer")
@ApiOperation(value = "启动/暂停指定队列消息消费")
public Response startStopConsumer(@RequestParam(value = "queueName", required = false) String queueName,
@RequestParam(value = "consumeSwitch", required = true) boolean consumeSwitch) {
log.info("启动/暂停指定队列消息消费,consumeSwitch:{},queueName:{}",consumeSwitch,queueName);
//获取所有消息监听容器
Collection listenerContainers = rabbitListenerEndpointRegistry.getListenerContainers();
for (MessageListenerContainer container : listenerContainers) {
SimpleMessageListenerContainer con = (SimpleMessageListenerContainer) container;
//消息监听容器要消费的队列名称集合
List queueNamesList = Arrays.asList(con.getQueueNames());
//判断容器中的队列名称是否包含需要调整的队列名参数
if (queueNamesList.contains(queueName)) {
if(consumeSwitch){
con.start();
}else{
con.stop();
}
}
}
return Response.success();
}
传入开关参数为false,停止pay_work_notify队列消费者消费,调用后控制台看到如下日志
2023-09-04 19:51:37.130 +0800 [TID: N/A] [http-nio-8080-exec-1] INFO c.b.t.m.p.w.PayCashierMockController:80 – 启动/暂停指定队列消息消费,consumeSwitch:false,queueName:pay_work_notify
2023-09-04 19:51:37.200 +0800 [TID: N/A] [http-nio-8080-exec-1] INFO o.s.a.r.l.SimpleMessageListenerContainer:586 – Waiting for workers to finish.
2023-09-04 19:51:37.903 +0800 [TID: N/A] [http-nio-8080-exec-1] INFO o.s.a.r.l.SimpleMessageListenerContainer:589 – Successfully waited for workers to finish.
可以看到消息监听容器关闭的日志,然后再传入开关参数为true,调用后会启动pay_work_notify队列消息消费。
服务器托管,北京服务器托管,服务器租用 http://www.fwqtg.net
相关推荐: 干货:《Android 性能优化实战篇》带你杀穿金九银十
近年来,由于疫情和互联网行业寒冬的影响,Android开发领域的竞争变得更加激烈。各大公司的业务线收缩和裁员成为了常规操作,导致职场环境日益严峻。 作为一个螺丝钉,我们想要卷出重围,性能优化必须了解一下!本文将梳理Android性能优化的知识模块,帮助初学者了…