32 lines
760 B
Java
32 lines
760 B
Java
|
package it.cavallium.dbengine.lucene.searcher;
|
||
|
|
|
||
|
|
import java.time.Duration;
|
||
|
|
import java.util.function.Function;
|
||
|
|
import reactor.core.publisher.Flux;
|
||
|
|
import reactor.core.publisher.Mono;
|
||
|
|
|
||
|
|
public class TimeoutUtil {
|
||
|
|
|
||
|
|
private static final Duration INFINITE = Duration.ofDays(360);
|
||
|
|
|
||
|
|
public static <T> Function<Mono<T>, Mono<T>> timeoutMono(Duration timeout) {
|
||
|
|
return query -> {
|
||
|
|
if (timeout.isZero() || timeout.isNegative() || timeout.compareTo(INFINITE) > 0) {
|
||
|
|
return query;
|
||
|
|
} else {
|
||
|
|
return query.timeout(timeout);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
public static <T> Function<Flux<T>, Flux<T>> timeoutFlux(Duration timeout) {
|
||
|
|
return query -> {
|
||
|
|
if (timeout.compareTo(INFINITE) > 0) {
|
||
|
|
return query;
|
||
|
|
} else {
|
||
|
|
return query.timeout(timeout);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
}
|
||
|
|
}
|