RabbitMQ学习之优先级队列

147 阅读2分钟

​ 本文已参与【新人创作礼】活动,一起开启掘金创作之路。

 一、使用场景

在我们系统中有一个订单催付的场景,客户在天猫下的订单,淘宝会及时将订单推送给我们,如
果在用户设定的时间内未付款那么就会给用户推送一条短信提醒,但是,商家对我们来说,会分大客户和小客户,比如像苹果,小米这样大商家一年起码能给我们创造很大的利润,所以理应当然,他们的订单必须得到优先处理,而曾经我们的后端系统是使用 redis 来存放的定时轮询,大家都知道 redis 只能用 List 做一个简简单单的消息队列,并不能实现一个优先级的场景,所以订单量大了后采用 RabbitMQ 进行改造和优化,如果发现是大客户的订单给一个相对比较高的优先级,否则就是默认优先级。

二、如何添加

a.控制台页面添加

​编辑

b.队列中代码添加优先级

Map<String, Object> params = new HashMap();
params.put("x-max-priority", 10);
channel.queueDeclare("hello", true, false, false, params);

​编辑

c.消息中代码添加优先级

AMQP.BasicProperties  properties  =  new
AMQP.BasicProperties().builder().priority(5).build();

d.注意事项

要让队列实现优先级需要做的事情有如下事情:队列需要设置为优先级队列,消息需要设置消息的优先级,消费者需要等待消息已经发送到队列中才去消费因为,这样才有机会对消息进行排序。

三、代码实现 

1.生产者

public class Producer {
    //队列名称
    public static final String QUEUE_NAME="hello";

    public static void main(String[] args) throws IOException, TimeoutException {
        //创建连接工厂
        ConnectionFactory factory = new ConnectionFactory();
        //工厂ip 连接RabbitMQ队列
        factory.setHost("192.168.16.106");
        factory.setUsername("guest");
        factory.setPassword("guest");

        //创建连接
        Connection connection = factory.newConnection();
        //获取信道
        Channel channel = connection.createChannel();
        HashMap<String, Object> arguments = new HashMap<>();
        //官方允许是0-255之间。此处设置10,不要设置过大,浪费cpu内存
        arguments.put("x-max-priority",10);
        /**
         *  生成一个队列
         * 1. 队列名称
         * 2. 队列里面的消息是否持久化(磁盘) 默认消息存储在内存中
         * 3. 该队列是否只供一个消费者进行消费 是否进行共享 true 可以多个消费者消费
         * 4. 是否自动删除 最后一个消费者端开连接以后 该队列是否自动删除 true  自动删除
         * 5. 其他参数
         */
        channel.queueDeclare(QUEUE_NAME,false,false,false,arguments);
        AMQP.BasicProperties properties =
                new AMQP.BasicProperties().builder().priority(7).build();
        //发消息
        for(int i=0;i<20;i++){
            String message = "info--"+i;
            if(i % 5 == 0){
                channel.basicPublish("",QUEUE_NAME,properties,message.getBytes());
            }else {
                channel.basicPublish("",QUEUE_NAME,null,message.getBytes());
            }

        }
        /**
         *  发送一个消息
         * 1. 发送到哪个交换机
         * 2. 路由的 key 是哪个
         * 3. 其他的参数信息
         * 4. 发送消息的消息体
         */
        System.out.println("消息发送完毕");

    }
}

2.消费者 

public class Consumer {
    //队列名称
    public static final String QUEUE_NAME="hello";

    public static void main(String[] args) throws IOException, TimeoutException {
        //创建连接工厂
        ConnectionFactory factory = new ConnectionFactory();
        //工厂ip 连接RabbitMQ队列
        factory.setHost("192.168.16.106");
        factory.setUsername("guest");
        factory.setPassword("guest");
        //创建连接
        Connection connection = factory.newConnection();
        //获取信道
        Channel channel = connection.createChannel();

        // 推送的消息如何进行消费的接口回调
        DeliverCallback deliverCallback = (consumerTag,message) ->{
            System.out.println("接收到的消息:"+new String(message.getBody()));
        };

        //取消消费的一个回调接口 如在消费的时候队列被删除掉了
        CancelCallback cancelCallback = consumerTag ->{
            System.out.println("消费被中断");
        };

        /**
         *  消费者消费消息
         * 1. 消费哪个队列
         * 2. 消费成功之后是否要自动应答 true 代表自动应答 false 手动应答
         * 3. 当一个消息发送过来后的回调接口
         * 4.消费者取消消费的回调
         */
        channel.basicConsume(QUEUE_NAME,true,deliverCallback,cancelCallback);
    }
}

3.运行结果

​编辑

\