在spring-amqp中阅读兔消息的最早入口点是什么？

Question

我将线程本地的兔消息数据存储在MDC中。我希望为传入的兔消息清除旧的上下文数据并添加新的上下文数据，比如从报头中读取某些值或将兔消息的有效负载作为byte[]读取。不幸的是，我经常看到在消息命中带@RabbitHandler注释的方法之前发生异常。有没有一个更早的入口点可以用来建立这个上下文？我不知道反序列化发生之前会发生什么，但理想情况下，我希望在尝试反序列化之前访问消息。也许在某个地方有一个onMessageReceived(byte[] message, Map headers)方法钩子。调用堆栈中越早越好。

Answer 1

@RabbitHandler由AbstractRabbitListenerContainerFactory填充，它可以与自定义的MessageConverter：https://docs.spring.io/spring-amqp/docs/2.0.1.RELEASE/reference/html/_reference.html#message-converters一起提供。从MessagingMessageListenerAdapter.toMessagingMessage()调用它的fromMessage()。这是在MessagingMessageListenerAdapter.onMessage()中完成的。这确实是你可以挂接的非常早的地方。实际上，您仍然有一个原始的org.springframework.amqp.core.Message对象，没有任何转换，并且具有所有可用的头和属性。

嗯，你也可以注入：

/**
 * @param afterReceivePostProcessors the post processors.
 * @see AbstractMessageListenerContainer#setAfterReceivePostProcessors(MessagePostProcessor...)
 */
public void setAfterReceivePostProcessors(MessagePostProcessor... afterReceivePostProcessors) {

出于类似的原因你也在请求。

Answer 2

一种解决方案是使用SimpleMessageListenerContainer而不是@RabbitHandler注释，并使用自定义消息侦听器适配器。

示例：

@Bean
SimpleMessageListenerContainer container(ConnectionFactory connectionFactory,
                                         MessageListenerAdapter listenerAdapter) {
    SimpleMessageListenerContainer container = new SimpleMessageListenerContainer();
    container.setConnectionFactory(connectionFactory);
    container.setQueueNames(queueName);
    container.setMessageListener(listenerAdapter); // custom listener
    container.setMessageConverter(null); // disable default conversion
    return container;
}

@Bean
MessageListenerAdapter listenerAdapter() {
    RawMessageDelegate delegate = new RawMessageDelegate();
    return new MessageListenerAdapter(delegate);
}

public class RawMessageDelegate {

    void handleMessage(Message message) {
        byte[] body = message.getBody();
        MessageProperties properties = message.getMessageProperties();
        Map<String, Object> headers = properties.getHeaders();
        // handle raw data
    }

}