「RabbitMQ」实现消息的优先级

目录

1、在生产者方设置优先级

2、创建消息监听器

3、配置Bean

---

在 Spring Boot 中，要使用 RabbitMQ 的消息优先级功能，可以通过以下步骤进行配置和实现

1、在生产者方设置优先级

@Autowired
private RabbitTemplate rabbitTemplate;
public void sendMessageWithPriority(String message, int priority) {
    Message amqpMessage = MessageBuilder
            .withBody(message.getBytes())
            .setHeader("Priority", priority)
            .build();
    rabbitTemplate.convertAndSend("myexchange", "myroutingkey", amqpMessage);
}

上述代码中，在 RabbitMQ 消息头中添加了一个名为 Priority 的属性，用于表示消息优先级。

2、创建消息监听器

@Component
public class MessageHandler {

    @RabbitListener(
        queues = "myqueue",
        containerFactory = "priorityListenerContainerFactory",
        concurrency = "5"
    )
    public void handleMessage(Message message) {
        System.out.println("Received message with priority: " + message.getMessageProperties().getPriority());
    }

}

上述代码中在 @RabbitListener 注解中设置了三个参数：

• queues：指定需要监听的队列；
• containerFactory：指定容器工厂，用于创建消费者容器并设置优先级队列相关的属性。需要事先在配置文件或 Java 配置类中定义支持优先级队列的 RabbitListenerContainerFactory；
• concurrency：指定并发消费者的数量。

 3、配置Bean

在配置文件或 Java 配置类中定义 RabbitListenerContainerFactory，用于支持优先级队列相关的属性和值：

@Configuration
public class RabbitMQConfig {

    @Autowired
    private ConnectionFactory connectionFactory;

    @Bean
    public RabbitListenerContainerFactory priorityListenerContainerFactory() {
        SimpleRabbitListenerContainerFactory factory = new SimpleRabbitListenerContainerFactory();
        factory.setConnectionFactory(connectionFactory);
        factory.setConcurrentConsumers(5);
        factory.setMaxConcurrentConsumers(10);
        factory.setAfterReceivePostProcessors(
            message -> {
                MessageProperties properties = message.getMessageProperties();
                int priority = (int) properties.getHeaders().get("Priority");
                properties.setPriority(priority);
                return message;
            }
        );
        factory.setPriorityComparator(new PriorityMessageConverter());
        return factory;
    }

    // 自定义的消息转换器
    static class PriorityMessageConverter implements Comparator {
        @Override
        public int compare(Message o1, Message o2) {
            Integer p1 = o1.getMessageProperties().getPriority();
            Integer p2 = o2.getMessageProperties().getPriority();
            return p2.compareTo(p1);
        }
    }

}

上述代码中，我们定义了一个支持优先级队列的 SimpleRabbitListenerContainerFactory，并在其中设置了以下属性：

• • connectionFactory：连接工厂；
  • concurrentConsumers 和 maxConcurrentConsumers：最小和最大的并发消费者数量；
  • afterReceivePostProcessors：一个消息后处理器，用于在接收消息后将 Header 中的优先级设置为 MessageProperties 中的优先级；
  • priorityComparator：设置消息比较器，使具有高优先级的消息先被消费。

另外，需要注意，因为 Spring AMQP 默认使用的是 SimpleMessageConverter，该消息转换器不支持优先级队列相关的属性，因此需要自定义一个消息转换器来实现消息的优先级相关的转换。最后，启动应用程序，生产者发送带有优先级的消息，消费者将按照优先级从高到低消费消息。