RabbitMq学习笔记(六)高级特性之消费端限流

目录

场景

具体实现

yml

消费者代码

生产者代码

---

场景

在处理秒杀场景时经常会用到rabbitmq削峰限流作用，假设我们的系统能每秒处理1000个请求，如果有上万个请求同时打进来，会造成服务器的瘫痪。

这时就需要在系统之前加一次处理，将请求发送的MQ中，再让A系统以每秒1000的速率去请求mq服务器。

具体实现

• 配置prefetch数量
• 确认方式设置为手动确认

yml

spring.application.name=springboot_rabbitmq
spring.rabbitmq.host=192.168.0.102
spring.rabbitmq.port=5672
spring.rabbitmq.username=admin
spring.rabbitmq.password=admin
spring.rabbitmq.virtual-host=/
spring.rabbitmq.listener.simple.acknowledge-mode=manual
spring.rabbitmq.listener.simple.prefetch=2

消费者代码

@Component
@RabbitListener(queuesToDeclare = @Queue(name = "springboot-limit"))
public class CurrentlimitCustomer {

    @RabbitHandler
    public void receive(String msg, Channel channel, Message message) throws IOException {

        long deliveryTag = message.getMessageProperties().getDeliveryTag();
        try {
            Thread.sleep(1000*10);
            System.out.println("=====限流====>");
            System.out.println(msg);
            System.out.println(channel);
            System.out.println(message);
            //手动签收[参数1:消息投递序号,参数2:批量签收]
            channel.basicAck(deliveryTag, true);
        } catch (Exception e) {

        }
    }
}

注意：消费者如果不设置 channel.basicAck(deliveryTag, true); 进行消息确认，则永远不用从消息队列中再取消息

生产者代码

@Test
    public void test09() throws Exception {
        for (int i = 0; i < 10 ; i++) {
            rabbitTemplate.convertAndSend("springboot-limit",  "限流测试");
        }
        Thread.sleep(1000 * 1000);
    }

效果图