com.github.tcurrie.rest.factory.client.TimeBoxed Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rest.factory Show documentation

Simple rest client-server factory, you give it a url, it does the rest!

There is a newer version: 0.2.70

package com.github.tcurrie.rest.factory.client;

import com.github.tcurrie.rest.factory.v1.RestFactoryException;
import com.github.tcurrie.rest.factory.v1.TimeOut;

import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.function.Supplier;

final class TimeBoxed {
    private TimeBoxed() {
        throw new RestFactoryException("Can not construct instance of Factory class.");
    }

    static  T attempt(final Supplier work, final TimeOut timeout) {
        final ExecutorService executor = Executors.newSingleThreadExecutor();
        final Future future = executor.submit(work::get);
        executor.shutdown();
        try {
            return future.get(timeout.getTime(), timeout.getTimeUnit());
        } catch (final Exception e) {
            throw wrapWithRuntimeException(unwrapExecutionException(e));
        } finally {
            executor.shutdownNow();
        }
    }

    private static RuntimeException wrapWithRuntimeException(final Throwable t) {
        if (RuntimeException.class.isAssignableFrom(t.getClass())) {
            return (RuntimeException) t;
        } else {
            return new RestFactoryException("Failed to complete task.", t);
        }
    }

    private static Throwable unwrapExecutionException(final Throwable e) {
        if (ExecutionException.class.isAssignableFrom(e.getClass())) {
            return e.getCause();
        } else {
            return e;
        }
    }


}