org.openqa.selenium.remote.internal.HttpClientFactory.java Source code

Introduction

Here is the source code for org.openqa.selenium.remote.internal.HttpClientFactory.java

Source

package org.openqa.selenium.remote.internal;

import org.apache.http.HttpRequest;
import org.apache.http.HttpResponse;
import org.apache.http.ProtocolException;
import org.apache.http.client.HttpClient;
import org.apache.http.client.RedirectStrategy;
import org.apache.http.client.methods.HttpUriRequest;
import org.apache.http.conn.ClientConnectionManager;
import org.apache.http.conn.params.ConnConnectionPNames;
import org.apache.http.conn.routing.HttpRoutePlanner;
import org.apache.http.conn.scheme.PlainSocketFactory;
import org.apache.http.conn.scheme.Scheme;
import org.apache.http.conn.scheme.SchemeRegistry;
import org.apache.http.conn.ssl.SSLSocketFactory;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.impl.conn.ProxySelectorRoutePlanner;
import org.apache.http.impl.conn.tsccm.ThreadSafeClientConnManager;
import org.apache.http.params.BasicHttpParams;
import org.apache.http.params.HttpConnectionParams;
import org.apache.http.params.HttpParams;
import org.apache.http.protocol.HttpContext;

import java.net.ProxySelector;
import java.util.concurrent.TimeUnit;

import static java.util.concurrent.TimeUnit.SECONDS;

/*
Copyright 2007-2011 Selenium committers
    
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.
*/
public class HttpClientFactory {

    private final DefaultHttpClient httpClient;
    private final int TIMEOUT_THREE_HOURS = (int) SECONDS.toMillis(60 * 60 * 3);
    private final ClientConnectionManager gridClientConnectionManager = getClientConnectionManager();

    public HttpClientFactory() {
        httpClient = new DefaultHttpClient(getClientConnectionManager());
        httpClient.setParams(getHttpParams());
        httpClient.setRoutePlanner(getRoutePlanner(httpClient.getConnectionManager().getSchemeRegistry()));
    }

    private static ClientConnectionManager getClientConnectionManager() {
        SchemeRegistry registry = new SchemeRegistry();
        registry.register(new Scheme("http", 80, PlainSocketFactory.getSocketFactory()));
        registry.register(new Scheme("https", 443, SSLSocketFactory.getSocketFactory()));
        ThreadSafeClientConnManager cm = new ThreadSafeClientConnManager(registry);
        cm.setMaxTotal(2000);
        cm.setDefaultMaxPerRoute(2000);
        return cm;
    }

    public HttpClient getHttpClient() {
        return httpClient;
    }

    public HttpClient getGridHttpClient(int connection_timeout, int socket_timeout) {
        DefaultHttpClient gridClient = new DefaultHttpClient(gridClientConnectionManager);
        gridClient.setRedirectStrategy(new MyRedirectHandler());
        gridClient.setParams(getGridHttpParams(connection_timeout, socket_timeout));
        gridClient.setRoutePlanner(getRoutePlanner(gridClient.getConnectionManager().getSchemeRegistry()));
        gridClient.getConnectionManager().closeIdleConnections(100, TimeUnit.MILLISECONDS);

        return gridClient;
    }

    public HttpParams getHttpParams() {
        HttpParams params = new BasicHttpParams();
        HttpConnectionParams.setSoReuseaddr(params, true);
        HttpConnectionParams.setConnectionTimeout(params, 120 * 1000);
        HttpConnectionParams.setSoTimeout(params, TIMEOUT_THREE_HOURS);
        params.setIntParameter(ConnConnectionPNames.MAX_STATUS_LINE_GARBAGE, 0);
        HttpConnectionParams.setStaleCheckingEnabled(params, true);
        return params;
    }

    public HttpRoutePlanner getRoutePlanner(SchemeRegistry registry) {
        return new ProxySelectorRoutePlanner(registry, ProxySelector.getDefault());
    }

    public HttpParams getGridHttpParams(int connection_timeout, int socket_timeout) {
        final HttpParams params = getHttpParams();
        HttpConnectionParams.setSoTimeout(params, socket_timeout > 0 ? socket_timeout : TIMEOUT_THREE_HOURS);
        HttpConnectionParams.setConnectionTimeout(params, connection_timeout > 0 ? connection_timeout : 120 * 1000);
        return params;
    }

    public void close() {
        httpClient.getConnectionManager().shutdown();
        gridClientConnectionManager.shutdown();
    }

    static class MyRedirectHandler implements RedirectStrategy {

        public boolean isRedirected(HttpRequest request, HttpResponse response, HttpContext context)
                throws ProtocolException {
            return false;
        }

        public HttpUriRequest getRedirect(HttpRequest request, HttpResponse response, HttpContext context)
                throws ProtocolException {
            return null;
        }
    }
}