消息队列MQ在消费端如何进行限流？

在使用消息队列时，如果生产方在某些时间段产生的消息远远大于消费方的处理能力，则可能造成消费方服务出问题，甚至服务崩溃。这个时候，需要通过消费方限流的方式来规避此类风险。

本贴，以RabbitMQ作为消息中间件，给出限流效果的验证示例：

配置文件（rabbitmq.listener.simple）

#开始手动确认模式

acknowledge-mode: manual

#一个消息端每次最多可处理的消息数量

prefetch: 2

关键消费方代码

@Component

@RabbitListener(queues = "myTestQueue")

@Slf4j

public class BsysMdOrgCodeQueueHandler {

@RabbitHandler

public void process(final String message,final Channel channel,final Message msg) throws IOException{

log.info("receive message：" + message);

try {

log.info("开始执行");

system.out.println("此处进行message数据的消费逻辑");

//关键步骤：进行手动确认

channel.basicAck(msg.getMessageProperties().getDeliveryTag(), false);

log.info("执行结束");

}catch(Exception e) {

channel.basicNack(msg.getMessageProperties().getDeliveryTag(), false, true);

log.info("消息队列消费失败);

}

}

}

验证方式：

（1）需要先注释掉channel.basicAck(msg.getMessageProperties().getDeliveryTag(), false)，会发现因为没有进行ack确认，多条消息到达时，消息队列的总条数条数不会变少，待确认消息队列条数为2；

（2）当取消上述注释后，消息队列可以正常消费。