global.maplink.http.HttpAsyncEngineJava11Impl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of http-engine-java11-client Show documentation
The newest version!
package global.maplink.http;

import global.maplink.http.request.*;
import lombok.val;

import java.net.http.HttpClient;
import java.net.http.HttpRequest;
import java.net.http.HttpRequest.BodyPublishers;
import java.net.http.HttpResponse;
import java.net.http.HttpResponse.BodyHandlers;
import java.util.Objects;
import java.util.concurrent.CompletableFuture;
import java.util.function.BiFunction;
import java.util.function.Function;

import static java.net.http.HttpClient.Version.HTTP_1_1;

public class HttpAsyncEngineJava11Impl implements HttpAsyncEngine {

    public static final String HEADER_CONTENT_TYPE = "content-type";

    public static final String HTTP = "http";

    private final HttpClient client = HttpClient.newHttpClient();

    @Override
    public CompletableFuture run(Request request) {
        if (request instanceof GetRequest) return runWithoutBody(request, HttpRequest.Builder::GET);
        if (request instanceof PostRequest) return runWithBody((PostRequest) request, HttpRequest.Builder::POST);
        if (request instanceof PutRequest) return runWithBody((PutRequest) request, HttpRequest.Builder::PUT);

        throw new UnsupportedOperationException(String.format("Tipo de requisição %s não é reconhecida", request.getClass().getName()));
    }

    private HttpRequest.Builder buildBaseRequest(Request request) {
        var builder = HttpRequest.newBuilder(request.getFullURI());
        request.getHeaders().forEach(builder::header);
        if (HTTP.equals(request.getUrl().getProtocol()))
            builder.version(HTTP_1_1);
        return builder;
    }

    private Response translateResponse(HttpResponse response) {
        return new Response(
                response.statusCode(),
                response.headers().firstValue(HEADER_CONTENT_TYPE).orElse(MediaType.Text.PLAIN),
                response.body()
        );
    }

    private CompletableFuture runWithoutBody(
            Request request,
            Function httpVerb
    ) {
        val req = httpVerb.apply(buildBaseRequest(request))
                .build();
        return client
                .sendAsync(req, BodyHandlers.ofByteArray())
                .thenApply(this::translateResponse);
    }

    private CompletableFuture runWithBody(
            WithBodyRequest request,
            BiFunction httpVerb
    ) {
        val req = httpVerb.apply(buildBaseRequest(request), BodyPublishers.ofByteArray(request.getBody()))
                .build();
        return client
                .sendAsync(req, BodyHandlers.ofByteArray())
                .thenApply(this::translateResponse);
    }
}