ExponentialBackoff.java

import com.datastax.driver.core.exceptions.NoHostAvailableException;
import io.lettuce.core.RedisCommandExecutionException;
import org.slf4j.Logger;

import java.util.concurrent.Callable;

public class ExponentialBackoff {
    private final long delay;
    private final long maxAttempts;
    private final Logger logger;

    private final long MAX_WAIT_TIME_MS = 60000;

    public ExponentialBackoff(Logger logger, long delay, long maxAttempts) {
        this.logger = logger;
        this.delay = delay;
        this.maxAttempts = maxAttempts;
    }

    @SuppressWarnings("InfiniteLoopStatement")
    public void execute(Callable<Void> task) throws Exception {
        int attempt = 0;

        while(true) {
            try {
                task.call();
                attempt = 0;
                Thread.sleep(this.delay);
            } catch (NoHostAvailableException | RedisCommandExecutionException error) {
                if (attempt == this.maxAttempts) {
                    this.logger.error("Long connectivity issue is ongoing");
                } else {
                    attempt += 1;
                }

                long waitTime = this.calculateWaitTime(attempt);
                this.logger.info("Retrying in {}ms...", waitTime);
                Thread.sleep(waitTime);
            }
        }
    }

    private long calculateWaitTime(int attempt) {
        final long wait = ((long) Math.pow(2, attempt)) * Math.max(this.delay, 1);
        return Math.min(wait, MAX_WAIT_TIME_MS);
    }
}