什么是RabbitMq Listener的ServletFilter equalent?

时间:2018-04-11 09:03:20

标签: spring-boot rabbitmq spring-rabbitmq

我有一个spring-boot Web应用程序,我为其实现了MDCFilter,它可以在日志文件中添加UUIDMDC日志记录上下文。

Filter类看起来像这样。

public class MDCFilter implements Filter {

  @Override
  public void init(FilterConfig filterConfig) {
  }

  @Override
  public void doFilter(ServletRequest req, ServletResponse resp, FilterChain chain)
      throws IOException, ServletException {

    String requestId = UUID.randomUUID().toString();

    MDC.put(REQUEST_ID_KEY, requestId);
    response.addHeader("trace", requestId);
    try {
      chain.doFilter(req, resp);
    } finally {
      MDC.remove("trace");
    }
  }

  @Override
  public void destroy() {
  }
}

但是最近我们开始通过队列处理流量,我不知道文档是否为消息监听器复制了这种过滤行为。

我的听众会看起来像这样。

@RabbitListener(queues = "${queue1}")
public void receiveMessages(Message message) {
doTheBusinessLogic(message)
}

有人能指出我正确的方向吗?

1 个答案:

答案 0 :(得分:1)

使用容器adviceChain。假设您使用的是Boot 2.0和简单的容器工厂,请覆盖boot的工厂以添加建议...

@SpringBootApplication
public class So49770881Application {

    public static void main(String[] args) {
        SpringApplication.run(So49770881Application.class, args);
    }

    @Bean(name = "rabbitListenerContainerFactory")
    public SimpleRabbitListenerContainerFactory simpleRabbitListenerContainerFactory(
            SimpleRabbitListenerContainerFactoryConfigurer configurer,
            ConnectionFactory connectionFactory) {
        SimpleRabbitListenerContainerFactory factory = new SimpleRabbitListenerContainerFactory();
        configurer.configure(factory, connectionFactory);
        factory.setAdviceChain(new MDCAdvice());
        return factory;
    }

    public static class MDCAdvice implements MethodInterceptor {

        @Override
        public Object invoke(MethodInvocation invocation) throws Throwable {
            // pre process
            try {
                return invocation.proceed();
            }
            finally {
                // post process
            }
        }

    }

}