Spring amqp拒绝侦听器外部的消息

时间:2018-11-22 11:17:19

标签: java rabbitmq spring-amqp

该应用程序使用Java 10,spring amqp和Rabbitmq。

系统有一个死信队列,我们​​在其中发送一些消息(由于数据库不可用,无法按预期处理它们)。

目前,每隔X秒检查一次数据库可用性,并且如果仅可用的话,我们会将消息重新排队到其原始队列中。否则,我们什么也不做,消息会停留在死信队列中。

重新排队到原始队列后,邮件可以再次返回死信队列,并看到x-death标头计数增加。

由于某些原因,我们希望处理计数大于等于5的死信消息,例如,将其他消息重新排队进入死信队列。

我首先需要对消息进行基本确认,以检查x-death计数头,然后如果计数足够大,则将其发送到原始队列,否则重新进入死信队列。

我无法重新排队进入死信队列,因为基本进入不在侦听器内:抛出AmqpRejectAndDontRequeueException不起作用,因为没有在Rabbitmq侦听器对象内抛出异常。

我尝试将异常抛出到receiveAndCallback方法中,但这似乎并不好:

rabbitTemplate.receiveAndReply(queueName, new ReceiveAndReplyCallback<Message, Object>() {

        @Override
        public Object handle(Message message) {
            Long messageXdeathCount = null;
            if (null != message.getMessageProperties() && null != message.getMessageProperties().getHeaders()) {
                List<Map<String, ?>> xdeathHeader =
                        (List<Map<String, ?>>) message.getMessageProperties().getHeaders().get(
                        "x-death");
                if (null != xdeathHeader && null != xdeathHeader.get(0)) {
                    messageXdeathCount = (Long) xdeathHeader.get(0).get("count");
                }
            }
            if (messageXdeathCount == null) {
                messageXdeathCount = 0L;
            }
            if (messageXdeathCount >= 5) {
                resendsMessage(message);
            } else {
                // this does not reject the message
                throw new AmqpRejectAndDontRequeueException("rejected");
            }
            return null;
        }
    });
    return receive;
}

执行此方法后,该消息未如我所愿被拒绝,并且已离开队列(已被确认)。

这是交换和队列声明:

@Bean
public Exchange exchange() {
    TopicExchange exchange = new TopicExchange(EXCHANGE, true, false);
    admin().declareExchange(exchange);
    Map<String, Object> args = new HashMap<String, Object>();
    args.put("x-dead-letter-exchange", EXCHANGE);
    Queue queue = new Queue("queueName", true, false, false, args);
    admin().declareQueue(queue);
    Binding binding = BindingBuilder.bind(queue).to(exchange).with(routingKey).noargs();
    admin().declareBinding(binding);
    return exchange;
}

如何不使用AmqpRejectAndDontRequeueException拒绝死信队列中的消息? 交易所是否可以将x-dead-letter-exchange设置为self?

感谢您的帮助

更新

我尝试了另一种方法,使用通道get和reject:

// exchange creation
@Bean
public Exchange exchange() throws IOException {
    Connection connection = connectionFactory().createConnection();
    Channel channel = channel();
    channel.exchangeDeclare(EXCHANGE, ExchangeTypes.TOPIC, true, false, null);
    Map<String, Object> args = new HashMap<String, Object>();
    args.put("x-dead-letter-exchange", EXCHANGE);
    channel.queueDeclare("queueName", true, false, false, args);
    channel.queueBind("queueName", EXCHANGE, routingKey);
    return exchange;
}

消息获取,确认或拒绝

GetResponse response = channel.basicGet(queueName, false);
Long messageXdeathCount = null;
if(null != response.getProps() && null != response.getProps().getHeaders()) {
    List<Map<String, ?>> xdeathHeader =
            (List<Map<String, ?>>) response.getProps().getHeaders().get("x-death");
    if(null != xdeathHeader && null != xdeathHeader.get(0)) {
        messageXdeathCount = (Long) xdeathHeader.get(0).get("count");
    }
}
if (messageXdeathCount == null) {
    messageXdeathCount = 0L;
}
if (messageXdeathCount >= 5) {
    MessagePropertiesConverter messagePropertiesConverter = new DefaultMessagePropertiesConverter();
    MessageProperties messageProps =
            messagePropertiesConverter.toMessageProperties(response.getProps(),
response.getEnvelope(), "UTF-8");
    resendsMessage(new Message(response.getBody(), messageProps));
    channel.basicAck(response.getEnvelope().getDeliveryTag(), false);
} else {
    if(response.getProps().getHeaders().get("x-death") == null) {
        response.getProps().getHeaders().put("x-death", new ArrayList<>());
    }
    if(((List<Map<String, Object>>) response.getProps().getHeaders().get("x-death")).get(0) == null) {
        ((List<Map<String, Object>>)response.getProps().getHeaders().get("x-death")).add(new HashMap<>());
    }
    ((List<Map<String, Object>>) response.getProps().getHeaders().get("x-death")).get(0).put(
            "count", messageXdeathCount + 1);
    channel.basicReject(response.getEnvelope().getDeliveryTag(), true);
}

首先我意识到这很丑陋,然后无法在获取和拒绝之间更新消息。是否可以使用channel.basicReject并更新x-death计数标头?

2 个答案:

答案 0 :(得分:1)

receiveAndReply()方法当前不提供对所接收消息的确认的控制。随时打开New Feature Request

您可以改用侦听器容器来获得所需的灵活性。

编辑

您可以下拉到Rabbitmq API ...

rabbitTemplate.execute(channel -> {
    // basicGet, basicPublish, ack/nack etc here
});

答案 1 :(得分:0)

我可以使用频道的基本方法:

GetResponse response = channel.basicGet(queueName, false);
Long messageXdeathCount = 0L;
if(null != response.getProps() && null != response.getProps().getHeaders()) {
    List<Map<String, ?>> xdeathHeader =
            (List<Map<String, ?>>) response.getProps().getHeaders().get("x-death");
    if(null != xdeathHeader && null != xdeathHeader.get(0)) {
        for (Map<String, ?> map : xdeathHeader) {
            Long count = (Long) map.get("count");
            messageXdeathCount += count == null ? 0L : count;
        }
    }
}
if (messageXdeathCount >= 5) {
    MessagePropertiesConverter messagePropertiesConverter = new DefaultMessagePropertiesConverter();
    MessageProperties messageProps = messagePropertiesConverter.toMessageProperties(response.getProps(), response.getEnvelope(), "UTF-8");
    resendsMessage(new Message(response.getBody(), messageProps));
    channel.basicAck(response.getEnvelope().getDeliveryTag(), false);
} else {
    channel.basicReject(response.getEnvelope().getDeliveryTag(), false);
}

我的问题的更新部分中的问题在最后一行:

channel.basicGet(queueName, true);

布尔值指示是否应该重新排队该消息:如果不重新排队,它会交换字母并按预期方式增加计数x-death标头。将布尔值更新为false可以解决此问题。