org.zalando.riptide.spring.RetryPolicyFactoryBean Maven / Gradle / Ivy

Go to download
package org.zalando.riptide.spring;

import net.jodah.failsafe.RetryPolicy;
import org.springframework.beans.factory.FactoryBean;
import org.zalando.riptide.failsafe.RetryAfterDelayFunction;
import org.zalando.riptide.failsafe.RetryException;
import org.zalando.riptide.faults.TransientFaultException;
import org.zalando.riptide.spring.RiptideSettings.Retry;

import javax.annotation.Nullable;
import java.util.Optional;
import java.util.concurrent.TimeUnit;

import static java.time.Clock.systemUTC;
import static java.util.concurrent.TimeUnit.MILLISECONDS;

final class RetryPolicyFactoryBean implements FactoryBean {

    private final RetryPolicy retryPolicy = new RetryPolicy().withMaxRetries(0);

    public void setConfiguration(final Retry config) {
        Optional.ofNullable(config.getFixedDelay())
                .ifPresent(delay -> delay.applyTo(retryPolicy::withDelay));

        Optional.ofNullable(config.getBackoff()).ifPresent(backoff -> {
            final TimeSpan delay = backoff.getDelay();
            final TimeSpan maxDelay = backoff.getMaxDelay();
            final TimeUnit unit = MILLISECONDS;

            @Nullable final Double delayFactor = backoff.getDelayFactor();

            if (delayFactor == null) {
                retryPolicy.withBackoff(delay.to(unit), maxDelay.to(unit), unit);
            } else {
                retryPolicy.withBackoff(delay.to(unit), maxDelay.to(unit), unit, delayFactor);
            }
        });

        retryPolicy.withMaxRetries(Optional.ofNullable(config.getMaxRetries()).orElse(-1));

        Optional.ofNullable(config.getMaxDuration())
                .ifPresent(duration -> duration.applyTo(retryPolicy::withMaxDuration));

        Optional.ofNullable(config.getJitterFactor())
                .ifPresent(retryPolicy::withJitter);

        Optional.ofNullable(config.getJitter())
                .ifPresent(jitter -> jitter.applyTo(retryPolicy::withJitter));

        retryPolicy.retryOn(TransientFaultException.class);
        retryPolicy.retryOn(RetryException.class);
        retryPolicy.withDelay(new RetryAfterDelayFunction(systemUTC()));
    }

    @Override
    public RetryPolicy getObject() {
        return retryPolicy;
    }

    @Override
    public Class getObjectType() {
        return RetryPolicy.class;
    }

    @Override
    public boolean isSingleton() {
        return true;
    }

}