RabbitMq优先级队列实现

为谁_疯狂

已于 2022-09-06 15:02:12 修改

阅读量1.7k

点赞数 1

分类专栏：工作中。。。文章标签： rabbitmq java spring boot

于 2022-09-06 15:00:11 首次发布

本文链接：https://blog.csdn.net/qq_20741133/article/details/126724406

版权

工作中。。。专栏收录该内容

20 篇文章 1 订阅

订阅专栏

1.springboot配置

设置prefetch=2，就是对应图上的unached的值。acknowledge-mode=manual是设置模式。其他人都是设置simple.acknowledge-mode=manual，而我的是设置direct.acknowledge-mode才好用。

配置参数参考地址：
https://blog.csdn.net/hzzdecsdn/article/details/123548022

spring.rabbitmq.listener.simple.concurrency=1
spring.rabbitmq.listener.simple.max-concurrency=2
spring.rabbitmq.listener.simple.prefetch=2
spring.rabbitmq.listener.direct.acknowledge-mode=manual
# spring.rabbitmq.listener.simple.acknowledge-mode=manual

在这里插入图片描述

2.代码初始化队列

代码中的 args.put(“x-max-priority”, 10); 即为设为带优先级队列的标志。本人项目中是用RabbitAdmin初始化的队列。下面的代码，这里只提供个参考。

    private void bindQ(TopicExchange topicExchange, String queuename, String routingKey, RabbitAdmin rabbitAdmin) {
        Map<String, Object> args = new HashMap<>();
        if (queuename.contains("hky")) {
            args.put("x-max-priority", 10);
        }
        Queue queue = new Queue(queuename, true, false, false, args);

        rabbitAdmin.declareQueue(queue);

        rabbitAdmin.declareBinding(BindingBuilder.bind(queue).to(topicExchange).with(routingKey));

    }

如此初始化出来的队列，就会多出来个Pri的标记。说明创建优先级队列成功了。
在这里插入图片描述
当然，也可以用控制台创建优先级队列，如下图：

3.代码中设置优先级——生产者

生产者发送消息，设置优先级。代码中的setPriority(5)即为优先级信息。越大就越提前执行。
比如我设置了发了3条优先级为5的消息，消息状态为Ready状态，接着发了一个优先级为10的消息，那么Ready中的消息会把优先级为10的排到最前面，最先消费，这样就实现了插队。
（注意：必须是Ready状态，如果都在unacked状态，说明springboot配置prefetch=2未生效）

rabbitTemplate.convertAndSend("hky", "syp1032.hky", msg, message -> {
    message.getMessageProperties().setPriority(5);
    return message;
});

关于发消息这里，我看只要写优先级队列的人，都用amqpTemplate，主要基本都一个模子刻出来的，有点无语。。。像下面这种的写法的，如果是这种方式的，我没用过，但应该也是可以的吧。

amqpTemplate.convertAndSend(queue, payload, message -> {
	message.getMessageProperties().setPriority(priority);
	return message;
});

4.消费者

@Component
public class TestReceiverListener {

//    @RabbitListener(queues = {"syp1032.hky","syp1032.hkyfast"}, ackMode = "MANUAL", concurrency = "1")
    @RabbitListener(queues = "syp1032.hky")
    @Transactional(rollbackOn = Throwable.class)
    public void process(MessageRequestDto dto, @Header(AmqpHeaders.DELIVERY_TAG) long deliveryTag, Channel channel) throws Exception {

        try {
            Thread.sleep(10000);
        } catch (InterruptedException e) {

            e.printStackTrace();
        }
        String parameter = dto.getParameter();
        System.out.println(parameter);
//        try {
//            channel.basicAck(deliveryTag, false);
//        } catch (IOException e) {
//            // TODO Auto-generated catch block
//            e.printStackTrace();
//        }
    }