com.rabbitmq.client.impl.recovery.DefaultRetryHandler Maven / Gradle / Ivy
Show all versions of amqp-client Show documentation
// Copyright (c) 2018 Pivotal Software, Inc. All rights reserved.
//
// This software, the RabbitMQ Java client library, is triple-licensed under the
// Mozilla Public License 1.1 ("MPL"), the GNU General Public License version 2
// ("GPL") and the Apache License version 2 ("ASL"). For the MPL, please see
// LICENSE-MPL-RabbitMQ. For the GPL, please see LICENSE-GPL2. For the ASL,
// please see LICENSE-APACHE2.
//
// This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY KIND,
// either express or implied. See the LICENSE file for specific language governing
// rights and limitations of this software.
//
// If you have any questions regarding licensing, please contact us at
// [email protected].
package com.rabbitmq.client.impl.recovery;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.Objects;
import java.util.function.BiPredicate;
/**
* Composable topology recovery retry handler.
* This retry handler implementations let the user choose the condition
* to trigger retry and the retry operation for each type of recoverable
* entities. The number of attempts and the backoff policy (time to wait
* between retries) are also configurable.
*
* See also {@link TopologyRecoveryRetryHandlerBuilder} to easily create
* instances and {@link TopologyRecoveryRetryLogic} for ready-to-use
* conditions and operations.
*
* @see TopologyRecoveryRetryHandlerBuilder
* @see TopologyRecoveryRetryLogic
* @since 5.4.0
*/
public class DefaultRetryHandler implements RetryHandler {
private static final Logger LOGGER = LoggerFactory.getLogger(DefaultRetryHandler.class);
private final BiPredicate queueRecoveryRetryCondition;
private final BiPredicate exchangeRecoveryRetryCondition;
private final BiPredicate bindingRecoveryRetryCondition;
private final BiPredicate consumerRecoveryRetryCondition;
private final RetryOperation queueRecoveryRetryOperation;
private final RetryOperation exchangeRecoveryRetryOperation;
private final RetryOperation bindingRecoveryRetryOperation;
private final RetryOperation consumerRecoveryRetryOperation;
private final int retryAttempts;
private final BackoffPolicy backoffPolicy;
public DefaultRetryHandler(BiPredicate queueRecoveryRetryCondition,
BiPredicate exchangeRecoveryRetryCondition,
BiPredicate bindingRecoveryRetryCondition,
BiPredicate consumerRecoveryRetryCondition,
RetryOperation queueRecoveryRetryOperation,
RetryOperation exchangeRecoveryRetryOperation,
RetryOperation bindingRecoveryRetryOperation,
RetryOperation consumerRecoveryRetryOperation, int retryAttempts, BackoffPolicy backoffPolicy) {
this.queueRecoveryRetryCondition = queueRecoveryRetryCondition;
this.exchangeRecoveryRetryCondition = exchangeRecoveryRetryCondition;
this.bindingRecoveryRetryCondition = bindingRecoveryRetryCondition;
this.consumerRecoveryRetryCondition = consumerRecoveryRetryCondition;
this.queueRecoveryRetryOperation = queueRecoveryRetryOperation;
this.exchangeRecoveryRetryOperation = exchangeRecoveryRetryOperation;
this.bindingRecoveryRetryOperation = bindingRecoveryRetryOperation;
this.consumerRecoveryRetryOperation = consumerRecoveryRetryOperation;
this.backoffPolicy = backoffPolicy;
if (retryAttempts <= 0) {
throw new IllegalArgumentException("Number of retry attempts must be greater than 0");
}
this.retryAttempts = retryAttempts;
}
@SuppressWarnings("unchecked")
@Override
public RetryResult retryQueueRecovery(RetryContext context) throws Exception {
return doRetry((BiPredicate) queueRecoveryRetryCondition, queueRecoveryRetryOperation, context.queue(), context);
}
@SuppressWarnings("unchecked")
@Override
public RetryResult retryExchangeRecovery(RetryContext context) throws Exception {
return doRetry((BiPredicate) exchangeRecoveryRetryCondition, exchangeRecoveryRetryOperation, context.exchange(), context);
}
@SuppressWarnings("unchecked")
@Override
public RetryResult retryBindingRecovery(RetryContext context) throws Exception {
return doRetry((BiPredicate) bindingRecoveryRetryCondition, bindingRecoveryRetryOperation, context.binding(), context);
}
@SuppressWarnings("unchecked")
@Override
public RetryResult retryConsumerRecovery(RetryContext context) throws Exception {
return doRetry((BiPredicate) consumerRecoveryRetryCondition, consumerRecoveryRetryOperation, context.consumer(), context);
}
protected RetryResult doRetry(BiPredicate condition, RetryOperation operation, RecordedEntity entity, RetryContext context)
throws Exception {
int attempts = 0;
Exception exception = context.exception();
while (attempts < retryAttempts) {
if (condition.test(entity, exception)) {
log(entity, exception, attempts);
backoffPolicy.backoff(attempts + 1);
try {
Object result = operation.call(context);
return new RetryResult(
entity, result == null ? null : result.toString()
);
} catch (Exception e) {
exception = e;
attempts++;
}
} else {
throw exception;
}
}
throw exception;
}
protected void log(RecordedEntity entity, Exception exception, int attempts) {
LOGGER.info("Error while recovering {}, retrying with {} more attempt(s).", entity, retryAttempts - attempts, exception);
}
public interface RetryOperation {
T call(RetryContext context) throws Exception;
default RetryOperation andThen(RetryOperation after) {
Objects.requireNonNull(after);
return (context) -> {
call(context);
return after.call(context);
};
}
}
}