Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
io.gravitee.common.utils.RxHelper Maven / Gradle / Ivy
/*
* Copyright © 2015 The Gravitee team (http://gravitee.io)
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package io.gravitee.common.utils;
import io.reactivex.rxjava3.core.*;
import java.util.Objects;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.function.Predicate;
import lombok.AccessLevel;
import lombok.NoArgsConstructor;
import org.reactivestreams.Publisher;
/**
* @author Jeoffrey HAEYAERT (jeoffrey.haeyaert at graviteesource.com)
* @author GraviteeSource Team
*/
@NoArgsConstructor(access = AccessLevel.PRIVATE)
public class RxHelper {
public static final Predicate TRUE_PREDICATE = t -> true;
public static final Predicate FALSE_PREDICATE = o -> false;
/**
* Returns a {@link FlowableTransformer} that can be used in a composition.
* It basically offers the same behavior as {@link io.reactivex.rxjava3.core.Flowable#mergeWith(Publisher)} but allows completing as soon as one of the sources completes (success, error or disposed).
*
* @param other the other source.
* @param the type of the items emitted.
* @return a {@link FlowableTransformer} that will be applied.
*/
public static FlowableTransformer mergeWithFirst(Flowable other) {
return upstream -> other.materialize().mergeWith(upstream.materialize()).dematerialize(n -> n);
}
/**
* Returns a {@link FlowableTransformer} that can be used in a composition.
* It applies an interval for each element received in the {@link Flowable} upstream in opposition to {@link Flowable#delay(long, TimeUnit)} which will apply the delay only once for the {@link Flowable}
*
* @param delay the delay to apply between each element of the {@link Flowable}
* @param timeUnit the {@link TimeUnit} of the delay
* @return a {@link FlowableTransformer} that will be applied.
*/
public static FlowableTransformer delayElement(int delay, TimeUnit timeUnit) {
return delayElement(delay, timeUnit, FALSE_PREDICATE);
}
/**
* Returns a {@link FlowableTransformer} that can be used in a composition.
* It applies an interval for each element received in the {@link Flowable} upstream in opposition to {@link Flowable#delay(long, TimeUnit)} which will apply the delay only once for the {@link Flowable}
*
* @param delay the delay to apply between each element of the {@link Flowable}
* @param timeUnit the {@link TimeUnit} of the delay
* @param skipDelayPredicate the predicate to test if delay should be skipped, else delay is applied
* @return a {@link FlowableTransformer} that will be applied.
*/
public static FlowableTransformer delayElement(int delay, TimeUnit timeUnit, Predicate skipDelayPredicate) {
return upstream ->
upstream.concatMapSingle(e -> {
if (skipDelayPredicate.test(e)) {
return Single.just(e);
} else {
return Single.just(e).delay(delay, timeUnit);
}
});
}
/**
* Returns a {@link FlowableTransformer} that can be used in a composition.
* It retries the {@link Flowable} X times with delay between each attempt
*
* @param times the attempts number
* @param retryInterval the delay between each retry
* @param timeUnit the {@link TimeUnit} of the backOffDelay
* @return a {@link FlowableTransformer} that will be applied.
*/
public static FlowableTransformer retryFlowable(int times, int retryInterval, TimeUnit timeUnit) {
return retryFlowable(times, retryInterval, timeUnit, TRUE_PREDICATE);
}
/**
* Returns a {@link FlowableTransformer} that can be used in a composition.
* It retries the {@link Flowable} X times with delay between each attempt
*
* @param times the attempts number
* @param retryInterval the delay between each retry
* @param timeUnit the {@link TimeUnit} of the backOffDelay`
* @param retryPredicate the predicate to test if instance of {@link Throwable} has to be retried, else, emits directly the error
* @return a {@link FlowableTransformer} that will be applied.
*/
public static FlowableTransformer retryFlowable(
int times,
int retryInterval,
TimeUnit timeUnit,
Predicate retryPredicate
) {
Objects.requireNonNull(retryPredicate, "retryPredicate is null");
// Negate the retryPredicate to skip the Throwable in operators
final Predicate skipThrowable = retryPredicate.negate();
return upstream ->
upstream.retryWhen(throwables ->
throwables.compose(delayElement(retryInterval, timeUnit, skipThrowable)).compose(takeThenThrow(times, skipThrowable))
);
}
/**
* Same as {@link #retryFlowable(int, int, TimeUnit)} but with a {@link Maybe} instead.
*
* @param times the attempts number
* @param retryInterval the delay between each retry
* @param timeUnit the {@link TimeUnit} of the backOffDelay
* @param the value type.
* @return a {@link MaybeTransformer} that will be applied.
*/
public static MaybeTransformer retryMaybe(int times, int retryInterval, TimeUnit timeUnit) {
return retryMaybe(times, retryInterval, timeUnit, TRUE_PREDICATE);
}
/**
* Same as {@link #retryFlowable(int, int, TimeUnit)} but with a {@link Maybe} instead.
*
* @param times the attempts number
* @param retryInterval the delay between each retry
* @param timeUnit the {@link TimeUnit} of the backOffDelay
* @param the value type.
* @param retryPredicate the predicate to test if instance of {@link Throwable} has to be retried, else, emits directly the error
* @return a {@link MaybeTransformer} that will be applied.
*/
public static MaybeTransformer retryMaybe(
int times,
int retryInterval,
TimeUnit timeUnit,
Predicate retryPredicate
) {
// Negate the retryPredicate to skip the Throwable in operators
final Predicate skipThrowable = retryPredicate.negate();
return upstream ->
upstream.retryWhen(throwables ->
throwables.compose(delayElement(retryInterval, timeUnit, skipThrowable)).compose(takeThenThrow(times, skipThrowable))
);
}
/**
* Same as {@link #retryMaybe(int, int, TimeUnit)} but with a {@link Single} instead.
*
* @param times the attempts number
* @param retryInterval the delay between each retry
* @param timeUnit the {@link TimeUnit} of the backOffDelay
* @param the value type.
* @return a {@link MaybeTransformer} that will be applied.
*/
public static SingleTransformer retrySingle(int times, int retryInterval, TimeUnit timeUnit) {
return retrySingle(times, retryInterval, timeUnit, TRUE_PREDICATE);
}
/**
* Same as {@link #retryMaybe(int, int, TimeUnit)} but with a {@link Single} instead.
*
* @param times the attempts number
* @param retryInterval the delay between each retry
* @param timeUnit the {@link TimeUnit} of the backOffDelay
* @param the value type.
* @param retryPredicate the predicate to test if instance of {@link Throwable} has to be retried, else, emits directly the error
* @return a {@link MaybeTransformer} that will be applied.
*/
public static SingleTransformer retrySingle(
int times,
int retryInterval,
TimeUnit timeUnit,
Predicate retryPredicate
) {
// Negate the retryPredicate to skip the Throwable in operators
// Negate the retryPredicate to skip the Throwable in operators
final Predicate skipThrowable = retryPredicate.negate();
return upstream ->
upstream.retryWhen(throwables ->
throwables.compose(delayElement(retryInterval, timeUnit, skipThrowable)).compose(takeThenThrow(times, skipThrowable))
);
}
/**
* Returns a {@link CompletableTransformer} that can be used in a composition.
* It retries the {@link Flowable} X times with delay between each attempt
*
* @param times the attempts number
* @param retryInterval the delay between each retry
* @param timeUnit the {@link TimeUnit} of the backOffDelay
* @return a {@link CompletableTransformer} that will be applied.
*/
public static CompletableTransformer retry(int times, int retryInterval, TimeUnit timeUnit) {
// By default, we want to retry every throwable
return retry(times, retryInterval, timeUnit, TRUE_PREDICATE);
}
/**
* Returns a {@link CompletableTransformer} that can be used in a composition.
* It retries the {@link Flowable} X times with delay between each attempt
*
* @param times the attempts number
* @param retryInterval the delay between each retry
* @param timeUnit the {@link TimeUnit} of the backOffDelay
* @param retryPredicate the predicate to test if instance of {@link Throwable} has to be retried, else, emits directly the error
* @return a {@link CompletableTransformer} that will be applied.
*/
public static CompletableTransformer retry(int times, int retryInterval, TimeUnit timeUnit, Predicate retryPredicate) {
Objects.requireNonNull(retryPredicate, "retryPredicate is null");
// Negate the retryPredicate to skip the Throwable in operators
final Predicate skipThrowable = retryPredicate.negate();
return upstream ->
upstream.retryWhen(throwables ->
throwables
// No need to delay element if we do not apply retry
.compose(delayElement(retryInterval, timeUnit, skipThrowable))
// No need to check limit if we do not apply retry, error should be emitted directly
.compose(takeThenThrow(times, skipThrowable))
);
}
/**
* Returns a {@link FlowableTransformer} that can be used in a composition.
* It ignores the N first throwable then return the N+1 in error.
*
* @param limit the number of throwables to ignore
* @param throwDirectlyPredicate the predicate to check if limit should be computed or not. If it returns true, then {@link Maybe#error(Throwable)} is immediately emitted
* @return a {@link FlowableTransformer} that will be applied.
*/
private static FlowableTransformer takeThenThrow(final int limit, Predicate throwDirectlyPredicate) {
// By default, check the limit before returning a {@link Maybe#error}
final AtomicInteger tries = new AtomicInteger(0);
return upstream ->
upstream.flatMapMaybe(throwable -> {
if (tries.incrementAndGet() > limit || throwDirectlyPredicate.test(throwable)) {
return Maybe.error(throwable);
} else {
return Maybe.just(throwable);
}
});
}
}
