BlockingSupplierWithRetry.java

import java.util.function.Supplier;

public class BlockingSupplierWithRetry<T> implements Supplier<T> {
	private final Supplier<T> supplier;
	private final int tries;
	private final T defaultValue;
	private final long sleep;
	private final Class<? extends Throwable> throwableClass;

	private BlockingSupplierWithRetry(Supplier<T> supplier, int tries,
			T defaultValue, long sleep,
			Class<? extends Throwable> throwableClass) {
		this.supplier = supplier;
		this.tries = tries;
		this.defaultValue = defaultValue;
		this.sleep = sleep;
		this.throwableClass = throwableClass;
	}

	public static <T> BlockingSupplierWithRetry<T> of(Supplier<T> supplier,
			int tries, T defaultValue, long sleep,
			Class<? extends Throwable> throwableClass) {
		return new BlockingSupplierWithRetry<T>(supplier, tries, defaultValue,
				sleep, throwableClass);
	}

	@Override
	public T get() {
		try {
			return supplier.get();
		} catch (Throwable t) {
			if (throwableClass.isInstance(t)) {
				if (tries <= 0)
					return defaultValue;
				System.out.println(Thread.currentThread().getName()
						+ " >> Exception " + t + " try with tries "
						+ (tries - 1));
				try {
					Thread.sleep(sleep);
				} catch (InterruptedException e) {
					Thread.currentThread().interrupt();
				}
				return BlockingSupplierWithRetry.of(supplier, tries - 1,
						defaultValue, sleep, throwableClass).get();
			}
			throw t;
		}
	}
}