前言
一般来说RabbitMQ有个方法channel.basicNack()能够让消息回到队列中,这样可以实现重试。但是这样没有明确重试次数,如果当前的消息一直重试的话,则后面的消息就会堆积起来,导致后面的消息无法消费。这是一个致命的缺点。因此这就需要设置重试次数来解决这种问题。下面提供几种解决方案。
使用redis或者mongo等第三方存储当前重试次数。
在header中添加重试次数,并且使用channel.basicPublish() 方法重新将消息发送出去后将重试次数加1。
使用spring-rabbit中自带的retry功能。
重试
这一节介绍使用spring-rabbit自带的retry功能。
关键类介绍
BackOffPolicy:重试的回退策略,指以何种方式进行下一次重试(第一次重试后什么时候进行第二次重试),比如过了15秒后重试,随机时间重试。
RetryPolicy:重试策略或条件,可以指定超时重试,一直重试,简单重试等。
MessageRecoverer:消息回收类,当所有的重试次数都失败后,就会调用该类的recover方法
RetryTemplate:组合了BackOffPolicy,RetryPolicy,RetryListener,执行重试步骤的具体类。
RetryOperationsInterceptor:方法执行失败的拦截器类,拦截失败后交给RertyTemplate去执行重试。
SimpleMessageListenerContainer:用于管理消费者。
RetryListener:重试过程的监听器,第一次重试调用该类的open,每次重试不成功调用onError,最后一次重试调用close。
第一步
设置消息确认模式为auto,SpringBoot 环境添加以下配置。
spring.rabbitmq.listener.simple.acknowledge-mode=auto
第二步配置重试
private RetryOperationsInterceptor createRetry() {
RetryTemplate retryTemplate = new RetryTemplate();
retryTemplate.registerListener(new RetryListener() {
@Override
public boolean open(RetryContext context, RetryCallback callback) {
// 第一次重试调用
return false;
}
@Override
public void close(RetryContext context, RetryCallback callback, Throwable throwable) {
// 最后一次重试会调用
}
@Override
public void onError(RetryContext context, RetryCallback callback, Throwable throwable) {
// 每次重试失败后都会调用
}
});
retryTemplate.setRetryPolicy(new SimpleRetryPolicy(5));
retryTemplate.setBackOffPolicy(new NoBackOffPolicy());
return RetryInterceptorBuilder.stateless()
.retryOperations(retryTemplate).recoverer(new DefaultMessageRecoverer()).build();
}
第三步
配置SimpleMessageListenerContainer,并抛出异常重试。使用simpleRabbitListenerContainerFactory创建SimpleMessageListenerContainer有个好处就是在application.properties中对SimpleMessageListenerContainer的配置可以生效。比如配置spring.rabbitmq.listener.simple.acknowledge-mode=auto就可以直接生效。
@Bean
public SimpleMessageListenerContainer config(SimpleRabbitListenerContainerFactory simpleRabbitListenerContainerFactory) {
SimpleRabbitListenerEndpoint endpoint = new SimpleRabbitListenerEndpoint();
endpoint.setMessageListener(new MessageListenerAdapter(new ChannelAwareMessageListener() {
@Override
public void onMessage(Message message, Channel channel) throws Exception {
try {
// 处理业务逻辑
}catch (Throwable r) {
// 失败抛出异常重试
throw r;
}
}
}));
endpoint.setId(String.valueOf(UUID.randomUUID()));
SimpleMessageListenerContainer container = simpleRabbitListenerContainerFactory.createListenerContainer(endpoint);
// 配置队列信息
container.setQueueNames("队列1","队列2");
// 配置重试
container.setAdviceChain(createRetry());
return container;
}