io.github.theangrydev.thinhttpclient.apache.ApacheHttpClient.java Source code

Introduction

Here is the source code for io.github.theangrydev.thinhttpclient.apache.ApacheHttpClient.java

Source

/*
 * Copyright 2016 Liam Williams <liam.williams@zoho.com>.
 *
 * This file is part of thin-http-client.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *  http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package io.github.theangrydev.thinhttpclient.apache;

import io.github.theangrydev.thinhttpclient.api.*;
import org.apache.http.HttpEntity;
import org.apache.http.StatusLine;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpEntityEnclosingRequestBase;
import org.apache.http.client.methods.HttpRequestBase;
import org.apache.http.client.methods.HttpUriRequest;
import org.apache.http.entity.ContentType;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClientBuilder;
import org.apache.http.util.EntityUtils;

import java.io.IOException;
import java.net.URI;
import java.net.URL;

import static io.github.theangrydev.thinhttpclient.api.HeaderName.CONTENT_TYPE;
import static io.github.theangrydev.thinhttpclient.api.Headers.headers;
import static io.github.theangrydev.thinhttpclient.api.Response.response;
import static java.nio.charset.StandardCharsets.UTF_8;
import static java.util.Arrays.stream;
import static java.util.stream.Collectors.toList;

@SuppressWarnings("PMD") //TODO: fix
public class ApacheHttpClient implements HttpClient {

    private final CloseableHttpClient httpClient;

    private ApacheHttpClient(CloseableHttpClient httpClient) {
        this.httpClient = httpClient;
    }

    public static ApacheHttpClient apacheHttpClient() {
        return new ApacheHttpClient(HttpClientBuilder.create().build());
    }

    @Override
    public Response execute(Request request) throws IOException {
        HttpUriRequest apacheRequest = adaptRequest(request);
        try (CloseableHttpResponse apacheResponse = httpClient.execute(apacheRequest)) {
            return adaptResponse(apacheResponse);
        }
    }

    @Override
    public void close() throws IOException {
        httpClient.close();
    }

    private HttpUriRequest adaptRequest(Request request) {
        if (request.method.hasBody) {
            String header = request.header(CONTENT_TYPE);
            return HttpRequestWithEntity.httpRequestWithEntity(request.url, request.method, request.body, header);
        } else {
            return HttpRequestWithoutEntity.httpRequestWithoutEntity(request.url, request.method);
        }
    }

    private Response adaptResponse(CloseableHttpResponse apacheResponse) throws IOException {
        StatusLine statusLine = apacheResponse.getStatusLine();
        return response(adaptHeaders(apacheResponse), statusLine.getStatusCode(), adaptBody(apacheResponse));
    }

    private String adaptBody(CloseableHttpResponse apacheResponse) throws IOException {
        HttpEntity entity = apacheResponse.getEntity();
        if (entity == null) {
            return "";
        } else {
            return EntityUtils.toString(entity, UTF_8);
        }
    }

    private Headers adaptHeaders(CloseableHttpResponse apacheResponse) {
        return headers(stream(apacheResponse.getAllHeaders()).map(this::adaptHeader).collect(toList()));
    }

    private Header adaptHeader(org.apache.http.Header header) {
        return Header.header(header.getName(), adaptHeaderValue(header.getValue()));
    }

    private String adaptHeaderValue(String value) {
        if (value == null) {
            return "";
        } else {
            return value;
        }
    }

    private static final class HttpRequestWithoutEntity extends HttpRequestBase {
        private final Method method;

        private HttpRequestWithoutEntity(Method method) {
            this.method = method;
        }

        static HttpRequestWithoutEntity httpRequestWithoutEntity(URL url, Method method) {
            HttpRequestWithoutEntity httpRequest = new HttpRequestWithoutEntity(method);
            httpRequest.setURI(URI.create(url.toExternalForm()));
            return httpRequest;
        }

        @Override
        public String getMethod() {
            return method.name;
        }
    }

    private static final class HttpRequestWithEntity extends HttpEntityEnclosingRequestBase {
        private final Method method;

        private HttpRequestWithEntity(Method method) {
            this.method = method;
        }

        static HttpRequestWithEntity httpRequestWithEntity(URL url, Method method, String body,
                String contentType) {
            HttpRequestWithEntity httpRequest = new HttpRequestWithEntity(method);
            httpRequest.setURI(URI.create(url.toExternalForm()));
            if (!body.isEmpty()) {
                httpRequest.setEntity(new StringEntity(body, ContentType.parse(contentType)));
            }
            return httpRequest;
        }

        @Override
        public String getMethod() {
            return method.name;
        }
    }
}