com.arangodb.async.internal.utils.CompletableFutureUtils Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of arangodb-java-driver-shaded Show documentation

ArangoDB Java Driver Shaded

There is a newer version: 7.13.0

Show newest version

package com.arangodb.async.internal.utils;

import java.util.concurrent.*;

public class CompletableFutureUtils {

    private static final ScheduledExecutorService timeoutScheduler = Executors.newSingleThreadScheduledExecutor(r -> {
                Thread t = Executors.defaultThreadFactory().newThread(r);
                t.setDaemon(true);
                return t;
            }
    );

    private CompletableFutureUtils() {
    }

    public static  CompletableFuture orTimeout(CompletableFuture completableFuture, long timeout,
                                                     TimeUnit unit) {
        ScheduledFuture timeoutTask = timeoutScheduler.schedule(() ->
                completableFuture.completeExceptionally(new TimeoutException()), timeout, unit);
        completableFuture.whenComplete((v, e) -> timeoutTask.cancel(false));
        return completableFuture;
    }

}