springboot 中使用 RabbitMQ 配置使用优先级队列

RabbitMQ 支持优先级队列，当工作中有一些任务需要紧急优先处理，此时可以使用优先级队列
通过设置 MQ 的 x-max-priority 属性可以将对列设置为优先级队列

配置文件类

@Slf4j
@Getter
@Configuration
public class RabbitMQConfig {

    @Value("${spring.rabbitmq.addresses}")
    private String addresses;

    @Value("${spring.rabbitmq.username}")
    private String username;

    @Value("${spring.rabbitmq.password}")
    private String password;

    @Value("${spring.rabbitmq.virtual-host}")
    private String virtualHost;

    @Value("${spring.rabbitmq.cache.channel.size:100}")
    private int channelCacheSize;       // 单条链接channel 缓存大小

    @Value("${spring.rabbitmq.cache.channel.checkout-timeout:5000}")
    private long channelCheckTimeout;    //获取channel等待时间，毫秒

    @Value("${app.accept-text-queue-name}")
    private String acceptTextQueueName;

    @Bean
    public MessageConverter jsonConverter() {
        return new Jackson2JsonMessageConverter();
    }

    @Bean
    public CachingConnectionFactory connectionFactory() {
        CachingConnectionFactory connectionFactory = new CachingConnectionFactory();
        connectionFactory.setAddresses(addresses);
        connectionFactory.setUsername(username);
        connectionFactory.setPassword(password);
        connectionFactory.setVirtualHost(virtualHost);
        connectionFactory.setChannelCheckoutTimeout(channelCheckTimeout);
        connectionFactory.setChannelCacheSize(channelCacheSize);
        connectionFactory.setPublisherConfirmType(CachingConnectionFactory.ConfirmType.CORRELATED);
        connectionFactory.setPublisherReturns(true);
        return connectionFactory;
    }

    @Bean(name = "jsonRabbitTemplate")
    public RabbitTemplate jsonRabbitTemplate(ConnectionFactory connectionFactory) {
        RabbitTemplate template = new RabbitTemplate(connectionFactory);
        template.setMandatory(true);
        template.setMessageConverter(jsonConverter());
        template.setConfirmCallback((correlationData, ack, cause) -> {
            if (!ack) {
                log.error("confirm message error! correlationData: {}, cause: {}", correlationData, cause);
            }
        });
        template.setReturnsCallback((msg) -> {
            String content = new String(msg.getMessage().getBody(), StandardCharsets.UTF_8);
            String message = MessageFormatter.format("send msg error, routing key:{}, msg:{}", msg.getRoutingKey(), content).getMessage();
            LogssetaPrinter.printError(message);
            log.error(JsonObjectUtil.toJsonQuietly(msg));
        });

        return template;
    }

    @Bean(name = "acceptSseConvertQueue")
    public Queue acceptSseConvertQueue() {
        Map<String, Object> args = new HashMap<>();
        args.put("x-max-priority", 100);// 设置优先级最高为 100，数字越大优先级越高
        return new Queue(acceptTextQueueName, true, false, false, args);
    }

}

设置预取数

在消费者中默认 containerFactory 为 SimpleRabbitListenerContainerFactory ，其中默认预取数为 250，该配置在 AbstractMessageListenerContainer 中的 DEFAULT_PREFETCH_COUNT
如果需要每次消费一个消息，就需要在配置文件中指定预取数为 1， 配置项为

// 设置预取数为 1
spring.rabbitmq.listener.simple.prefetch = 1

设置并发消费数

在 @RabbitListener 中，配置项 concurrency 来配置并发消费数

在生产环境中一般需要设置高可用MQ集群，常见的是镜像队列模式

客户端配置镜像队列方式

CachingConnectionFactory connectionFactory = new CachingConnectionFactory();
        connectionFactory.setAddresses(addresses);
        connectionFactory.setUsername(username);
        connectionFactory.setPassword(password);

其中 addresses 是集群的ip端口列表。例如 "192.168.1.1:5672,192.168.1.2:5672,192.168.1.3:5672"

发送消息确认机制

RabbitMQ 发送消息确认机制是：生产者到交换机确认、交换机到对列确认
消息发送过程是 生产者 -> 交换机 -> 对列。消息丢失可能发生在生产者到交换机之间，也可能发生在交换机发队列之间。为保证消息可靠发送，需要在上述链路进行消息确认。

确认机制配置

connectionFactory.setPublisherConfirmType(CachingConnectionFactory.ConfirmType.CORRELATED); // 发送到交换机确认
connectionFactory.setPublisherReturns(true);   // 发送到对列确认

确认消息回调处理

RabbitTemplate template = new RabbitTemplate(connectionFactory);
        template.setMandatory(true);  // 设置确认消息回调
        template.setMessageConverter(jsonConverter());
     
        // 发送到交换机确认消息回调
        template.setConfirmCallback((correlationData, ack, cause) -> {
            if (!ack) {
                log.error("confirm message error! correlationData: {}, cause: {}", correlationData, cause);
            }
        });
        // 发送到对列确认消息回调
        template.setReturnsCallback((msg) -> {
            String content = new String(msg.getMessage().getBody(), StandardCharsets.UTF_8);
            String message = MessageFormatter.format("send msg error, routing key:{}, msg:{}", msg.getRoutingKey(), content).getMessage();
            System.out.println(message);
        });

此处对于异常消息仅进行打印处理，生产上需要告警、重试等策略。