com.github.bingoohuang.springrestclient.utils.RestFuture Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-rest-client Show documentation

convenient spring rest client for rest services

There is a newer version: 0.0.23

package com.github.bingoohuang.springrestclient.utils;

import com.mashape.unirest.http.HttpResponse;

import java.io.InputStream;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;

public class RestFuture implements Future, ResponseAware {
    private final Future> future;
    private final Class beanClass;
    private final RestReq restReq;
    private HttpResponse response;

    public RestFuture(Future> future,
                      final Class beanClass,
                      final RestReq restReq) {
        this.future = future;
        this.beanClass = beanClass;
        this.restReq = restReq;
    }

    @Override
    public boolean cancel(boolean mayInterruptIfRunning) {
        return future.cancel(mayInterruptIfRunning);
    }

    @Override
    public boolean isCancelled() {
        return future.isCancelled();
    }

    @Override
    public boolean isDone() {
        return future.isDone();
    }

    @Override
    public HttpResponse getResponse() {
        if (response != null) return response;

        try {
            return future.get();
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    @Override
    public T get() throws InterruptedException, ExecutionException {
        return processReturn(future.get());
    }

    @Override
    public T get(long timeout, TimeUnit unit) throws InterruptedException, ExecutionException, TimeoutException {
        return processReturn(future.get(timeout, unit));
    }

    private T processReturn(HttpResponse response) throws ExecutionException {
        this.response = response;

        if (restReq.isSuccessful(response)) {
            Object body = RestClientUtils.nullOrBody(response);
            if (body == null) return null;
            if (beanClass == InputStream.class) return (T) response.getRawBody();

            T bean = (T) Beans.unmarshal(body.toString(), beanClass);
            return bean;
        }

        try {
            throw restReq.processStatusExceptionMappings(response);
        } catch (Throwable throwable) {
            throw new ExecutionException(throwable);
        }
    }
}