ai.evolv.httpclients.OkHttpClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ascend-sdk Show documentation
SDK to interact with the Ascend product.
There is a newer version: 0.7.1
package ai.evolv.httpclients;

import ai.evolv.HttpClient;

import java.io.IOException;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.TimeUnit;

import okhttp3.Call;
import okhttp3.Callback;
import okhttp3.ConnectionPool;
import okhttp3.Request;
import okhttp3.Response;
import okhttp3.ResponseBody;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;


public class OkHttpClient implements HttpClient {

    private static final Logger LOGGER = LoggerFactory.getLogger(OkHttpClient.class);

    protected final okhttp3.OkHttpClient httpClient;

    /**
     * Initializes the OhHttp# httpClient.
     * 
     * Note: Default timeout is 1 second
     * 
     */
    public OkHttpClient() {
        this.httpClient = new okhttp3.OkHttpClient.Builder()
            .callTimeout(1, TimeUnit.SECONDS)
            .connectionPool(new ConnectionPool(3, 1000, TimeUnit.MILLISECONDS))
            .build();
    }

    /**
     * Initializes the OhHttp# httpClient.
     *
     * @param timeUnit Specify the unit of the timeout value.
     * @param timeout Specify a request timeout for the httpClient.
     */
    public OkHttpClient(TimeUnit timeUnit, long timeout) {
        this.httpClient = new okhttp3.OkHttpClient.Builder()
            .callTimeout(timeout, timeUnit)
            .connectionPool(new ConnectionPool(3, 1000, TimeUnit.MILLISECONDS))
            .build();
    }

    /**
     * Initializes the OhHttp# httpClient.
     *
     * @param httpClient An instance of okhttp3.OkHttpClient for Ascend to use.
     */
    public OkHttpClient(okhttp3.OkHttpClient httpClient) {
        this.httpClient = httpClient;
    }

    /**
     * Performs a GET request with the given url using the httpClient from
     * okhttp3.
     * @param url a valid url representing a call to the Participant API.
     * @return a Completable future instance containing a response from
     *     the API
     */
    public CompletableFuture get(String url) {
        return getStringCompletableFuture(url, httpClient);
    }

    protected static CompletableFuture getStringCompletableFuture(
        String url, okhttp3.OkHttpClient httpClient) {
        CompletableFuture responseFuture = new CompletableFuture<>();
        final Request request = new Request.Builder()
            .url(url)
            .build();

        httpClient.newCall(request).enqueue(new Callback() {
            @Override
            public void onFailure(Call call, IOException e) {
                responseFuture.completeExceptionally(e);
            }

            @Override
            public void onResponse(Call call, Response response) {
                String body = "";
                try (ResponseBody responseBody = response.body()) {
                    if (responseBody != null) {
                        body = responseBody.string();
                    }

                    if (!response.isSuccessful()) {
                        throw new IOException(String.format("Unexpected response "
                                + "when making GET request: %s using url: %s with body: %s",
                            response, request.url(), body));
                    }

                    responseFuture.complete(body);
                } catch (Exception e) {
                    responseFuture.completeExceptionally(e);
                }
            }
        });

        return responseFuture;
    }
}