org.zalando.riptide.faults.TransientFaultPlugin Maven / Gradle / Ivy

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

import org.zalando.riptide.Plugin;
import org.zalando.riptide.RequestArguments;
import org.zalando.riptide.RequestExecution;

import static org.zalando.fauxpas.FauxPas.partially;

public final class TransientFaultPlugin implements Plugin {

    private final FaultClassifier classifier;

    public TransientFaultPlugin() {
        this(FaultClassifier.createDefault());
    }

    public TransientFaultPlugin(final FaultClassifier classifier) {
        this.classifier = classifier;
    }

    @Override
    public RequestExecution prepare(final RequestArguments arguments, final RequestExecution execution) {
        return () -> execution.execute().exceptionally(partially(classifier::classifyExceptionally));
    }

}