HTTPClient.java

package com.tw.go.plugin.util;

import org.apache.http.HttpHost;
import org.apache.http.HttpResponse;
import org.apache.http.auth.AuthScope;
import org.apache.http.auth.UsernamePasswordCredentials;
import org.apache.http.client.AuthCache;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.protocol.HttpClientContext;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.auth.BasicScheme;
import org.apache.http.impl.auth.DigestScheme;
import org.apache.http.impl.client.BasicAuthCache;
import org.apache.http.impl.client.BasicCredentialsProvider;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.util.EntityUtils;

import java.net.URI;

public class HTTPClient {
    public String getRequest(String getURL, AuthenticationType authenticationType, String username, String password) throws Exception {
        CloseableHttpClient httpClient = null;
        try {
            HttpGet request = new HttpGet(getURL);

            HttpHost target = getHttpHost(getURL);
            AuthCache authCache = getAuthCache(authenticationType, target);
            HttpClientContext localContext = HttpClientContext.create();
            localContext.setAuthCache(authCache);

            BasicCredentialsProvider credentialsProvider = new BasicCredentialsProvider();
            credentialsProvider.setCredentials(new AuthScope(target), new UsernamePasswordCredentials(username, password));
            httpClient = HttpClients.custom().setDefaultCredentialsProvider(credentialsProvider).build();

            HttpResponse response = httpClient.execute(request, localContext);
            int statusCode = response.getStatusLine().getStatusCode();
            if (statusCode > 204) {
                throw new RuntimeException("Error occurred. Status Code: " + statusCode);
            }
            return EntityUtils.toString(response.getEntity(), "UTF-8");
        } finally {
            if (httpClient != null) {
                try {
                    httpClient.close();
                } catch (Exception e) {
                    // ignore
                }
            }
        }
    }

    public void postRequest(String updateURL, AuthenticationType authenticationType, String username, String password, String requestBody) throws Exception {
        CloseableHttpClient httpClient = null;
        try {
            HttpPost request = new HttpPost(updateURL);
            request.addHeader("content-type", "application/json");
            request.setEntity(new StringEntity(requestBody));

            HttpHost target = getHttpHost(updateURL);
            AuthCache authCache = getAuthCache(authenticationType, target);
            HttpClientContext localContext = HttpClientContext.create();
            localContext.setAuthCache(authCache);

            BasicCredentialsProvider credentialsProvider = new BasicCredentialsProvider();
            credentialsProvider.setCredentials(new AuthScope(target), new UsernamePasswordCredentials(username, password));
            httpClient = HttpClients.custom().setDefaultCredentialsProvider(credentialsProvider).build();

            HttpResponse response = httpClient.execute(request, localContext);
            int statusCode = response.getStatusLine().getStatusCode();
            if (statusCode > 204) {
                throw new RuntimeException("Error occurred. Status Code: " + statusCode);
            }
        } finally {
            if (httpClient != null) {
                try {
                    httpClient.close();
                } catch (Exception e) {
                    // ignore
                }
            }
        }
    }

    private HttpHost getHttpHost(String url) throws Exception {
        URI uri = new URI(url);
        return new HttpHost(uri.getHost(), uri.getPort(), uri.getScheme());
    }

    private AuthCache getAuthCache(AuthenticationType authenticationType, HttpHost target) {
        AuthCache authCache = new BasicAuthCache();
        if (authenticationType == AuthenticationType.BASIC) {
            authCache.put(target, new BasicScheme());
        } else {
            authCache.put(target, new DigestScheme());
        }
        return authCache;
    }
}