ca.uhn.fhir.rest.client.apache.ApacheRestfulClientFactory.java Source code

Introduction

Here is the source code for ca.uhn.fhir.rest.client.apache.ApacheRestfulClientFactory.java

Source

package ca.uhn.fhir.rest.client.apache;

/*
 * #%L
 * HAPI FHIR - Core Library
 * %%
 * Copyright (C) 2014 - 2016 University Health Network
 * %%
 * 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.
 * #L%
 */

import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;

import org.apache.commons.lang3.StringUtils;
import org.apache.http.HttpHost;
import org.apache.http.auth.AuthScope;
import org.apache.http.auth.UsernamePasswordCredentials;
import org.apache.http.client.CredentialsProvider;
import org.apache.http.client.HttpClient;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.impl.client.BasicCredentialsProvider;
import org.apache.http.impl.client.HttpClientBuilder;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.impl.client.ProxyAuthenticationStrategy;
import org.apache.http.impl.conn.PoolingHttpClientConnectionManager;

import ca.uhn.fhir.context.FhirContext;
import ca.uhn.fhir.rest.api.RequestTypeEnum;
import ca.uhn.fhir.rest.client.RestfulClientFactory;
import ca.uhn.fhir.rest.client.api.Header;
import ca.uhn.fhir.rest.client.api.IHttpClient;

/**
 * A Restful Factory to create clients, requests and responses based on the Apache httpclient library.
 * 
 * @author Peter Van Houte | peter.vanhoute@agfa.com | Agfa Healthcare
 */
public class ApacheRestfulClientFactory extends RestfulClientFactory {

    private HttpClient myHttpClient;
    private HttpHost myProxy;

    /**
     * Constructor
     */
    public ApacheRestfulClientFactory() {
        super();
    }

    /**
     * Constructor
     * 
     * @param theContext
     *            The context
     */
    public ApacheRestfulClientFactory(FhirContext theContext) {
        super(theContext);
    }

    public synchronized HttpClient getNativeHttpClient() {
        if (myHttpClient == null) {

            PoolingHttpClientConnectionManager connectionManager = new PoolingHttpClientConnectionManager(5000,
                    TimeUnit.MILLISECONDS);
            connectionManager.setMaxTotal(getPoolMaxTotal());
            connectionManager.setDefaultMaxPerRoute(getPoolMaxPerRoute());

            // @formatter:off
            RequestConfig defaultRequestConfig = RequestConfig.custom().setSocketTimeout(getSocketTimeout())
                    .setConnectTimeout(getConnectTimeout())
                    .setConnectionRequestTimeout(getConnectionRequestTimeout()).setStaleConnectionCheckEnabled(true)
                    .setProxy(myProxy).build();

            HttpClientBuilder builder = HttpClients.custom().setConnectionManager(connectionManager)
                    .setDefaultRequestConfig(defaultRequestConfig).disableCookieManagement();

            if (myProxy != null && StringUtils.isNotBlank(getProxyUsername())
                    && StringUtils.isNotBlank(getProxyPassword())) {
                CredentialsProvider credsProvider = new BasicCredentialsProvider();
                credsProvider.setCredentials(new AuthScope(myProxy.getHostName(), myProxy.getPort()),
                        new UsernamePasswordCredentials(getProxyUsername(), getProxyPassword()));
                builder.setProxyAuthenticationStrategy(new ProxyAuthenticationStrategy());
                builder.setDefaultCredentialsProvider(credsProvider);
            }

            myHttpClient = builder.build();
            // @formatter:on

        }

        return myHttpClient;
    }

    @Override
    public IHttpClient getHttpClient(StringBuilder theUrl, Map<String, List<String>> theIfNoneExistParams,
            String theIfNoneExistString, RequestTypeEnum theRequestType, List<Header> theHeaders) {
        return new ApacheHttpClient(getNativeHttpClient(), theUrl, theIfNoneExistParams, theIfNoneExistString,
                theRequestType, theHeaders);
    }

    @Override
    public void setProxy(String theHost, Integer thePort) {
        if (theHost != null) {
            myProxy = new HttpHost(theHost, thePort, "http");
        } else {
            myProxy = null;
        }
    }

    /**
     * Only allows to set an instance of type org.apache.http.client.HttpClient
     * @see ca.uhn.fhir.rest.client.IRestfulClientFactory#setHttpClient(ca.uhn.fhir.rest.client.api.IHttpClient)
     */
    @Override
    public synchronized void setHttpClient(Object theHttpClient) {
        this.myHttpClient = (HttpClient) theHttpClient;
    }

    @Override
    protected ApacheHttpClient getHttpClient(String theServerBase) {
        return new ApacheHttpClient(getNativeHttpClient(), new StringBuilder(theServerBase), null, null, null,
                null);
    }

    @Override
    protected void resetHttpClient() {
        this.myHttpClient = null;
    }

}