com.dell.cpsd.common.rabbitmq.retrypolicy.DefaultRetryPolicyAdvice Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of common-rabbitmq Show documentation
There is a newer version: 2.2.0
/**
 * Copyright © 2017 Dell Inc. or its subsidiaries.  All Rights Reserved.
 * Dell EMC Confidential/Proprietary Information
 */


package com.dell.cpsd.common.rabbitmq.retrypolicy;

import com.dell.cpsd.common.logging.ILogger;
import com.dell.cpsd.common.rabbitmq.exceptions.ExceptionLogTransformer;
import com.dell.cpsd.common.rabbitmq.log.RabbitMQLoggingManager;
import com.dell.cpsd.common.rabbitmq.log.RabbitMQMessageCode;
import org.aopalliance.intercept.MethodInterceptor;
import org.aopalliance.intercept.MethodInvocation;
import org.springframework.amqp.rabbit.config.StatefulRetryOperationsInterceptorFactoryBean;
import org.springframework.amqp.rabbit.retry.MessageRecoverer;
import org.springframework.retry.RetryCallback;
import org.springframework.retry.RetryContext;
import org.springframework.retry.RetryPolicy;
import org.springframework.retry.backoff.ExponentialBackOffPolicy;
import org.springframework.retry.interceptor.StatefulRetryOperationsInterceptor;
import org.springframework.retry.listener.RetryListenerSupport;
import org.springframework.retry.support.RetryTemplate;

/**
 * Interceptor for default retry policy
 * 
 * 

 * Copyright © 2017 Dell Inc. or its subsidiaries.  All Rights Reserved.
 * Dell EMC Confidential/Proprietary Information
 * 
 */
public class DefaultRetryPolicyAdvice implements MethodInterceptor
{
    protected final StatefulRetryOperationsInterceptor delegate;

    public DefaultRetryPolicyAdvice(MessageRecoverer messageRecoverer)
    {
        this(messageRecoverer, new DefaultRetryPolicy());
    }

    public DefaultRetryPolicyAdvice(MessageRecoverer messageRecoverer, RetryPolicy retryPolicy)
    {
        RetryTemplate retryTemplate = new RetryTemplate();
        retryTemplate.setBackOffPolicy(createBackOffPolicy());
        retryTemplate.setRetryPolicy(retryPolicy);
        retryTemplate.registerListener(new RetryErrorListener());

        StatefulRetryOperationsInterceptorFactoryBean factory = new StatefulRetryOperationsInterceptorFactoryBean();
        factory.setRetryOperations(retryTemplate);
        factory.setMessageKeyGenerator(new DefaultMessageKeyGenerator());
        factory.setMessageRecoverer(messageRecoverer);

        this.delegate = factory.getObject();
    }

    protected ExponentialBackOffPolicy createBackOffPolicy()
    {
        ExponentialBackOffPolicy backOffPolicy = new ExponentialBackOffPolicy();
        backOffPolicy.setMultiplier(5.0D);
        return backOffPolicy;
    }

    @Override
    public Object invoke(MethodInvocation invocation) throws Throwable
    {
        return delegate.invoke(invocation);
    }

    protected static class RetryErrorListener extends RetryListenerSupport
    {
        private static final ILogger LOGGER = RabbitMQLoggingManager.getLogger(RetryErrorListener.class);

        protected ExceptionLogTransformer exceptionTransformer = new ExceptionLogTransformer();

        @Override
        public  void onError(RetryContext context, RetryCallback callback, Throwable cause)
        {
            // Log only attempt and error. DefaultContainerErrorHandler will log details.
            Integer attempt = context == null ? null : (context.getRetryCount() + 1);
            cause = exceptionTransformer.transform(cause);
            String message = RabbitMQMessageCode.AMQP_ERROR_RETRY_E.getMessageText(attempt, cause.getMessage());
            LOGGER.error(message);
        }
    }
}