All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • org.zalando.riptide.failsafe.BackupRequestExecutor Maven / Gradle / Ivy

    package org.zalando.riptide.failsafe;

import net.jodah.failsafe.AbstractExecution;
import net.jodah.failsafe.ExecutionResult;
import net.jodah.failsafe.FailsafeFuture;
import net.jodah.failsafe.PolicyExecutor;
import net.jodah.failsafe.util.concurrent.Scheduler;

import java.util.concurrent.Callable;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.Future;
import java.util.concurrent.ScheduledFuture;
import java.util.concurrent.TimeUnit;
import java.util.function.BiConsumer;
import java.util.function.Supplier;

import static org.zalando.riptide.CompletableFutures.forwardTo;

final class BackupRequestExecutor extends PolicyExecutor> {

    BackupRequestExecutor(final BackupRequest policy, final AbstractExecution execution) {
        super(policy, execution);
    }

    @Override
    protected Supplier> supplyAsync(
            final Supplier> supplier,
            final Scheduler scheduler,
            final FailsafeFuture future) {

        return () -> {
            final CompletableFuture original = supplier.get();
            final CompletableFuture backup = new CompletableFuture<>();

            final Future scheduledBackup = delay(scheduler, backup(supplier, backup));

            original.whenComplete(cancel(scheduledBackup));
            backup.whenComplete(cancel(original));

            return anyOf(original, backup);
        };
    }

    private Callable> backup(
            final Supplier> supplier,
            final CompletableFuture target) {

        return () -> supplier.get().whenComplete(forwardTo(target));
    }

    @SuppressWarnings("unchecked")
    private  ScheduledFuture delay(
            final Scheduler scheduler,
            final Callable callable) {

        final long delay = policy.getDelay();
        final TimeUnit unit = policy.getUnit();
        return (ScheduledFuture) scheduler.schedule(callable, delay, unit);
    }

    private  BiConsumer cancel(final Future future) {
        return (result, throwable) -> future.cancel(true);
    }

    @SafeVarargs
    private final  CompletableFuture anyOf(final CompletableFuture... futures) {
        final CompletableFuture any = new CompletableFuture<>();

        for (final CompletableFuture future : futures) {
            future.whenComplete(forwardTo(any));
        }

        return any;
    }

}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy