com.almende.util.ApacheHttpClient.java Source code

Introduction

Here is the source code for com.almende.util.ApacheHttpClient.java

Source

/*
 * Copyright: Almende B.V. (2014), Rotterdam, The Netherlands
 * License: The Apache Software License, Version 2.0
 */
package com.almende.util;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;

import org.apache.http.client.CookieStore;
import org.apache.http.client.HttpClient;
import org.apache.http.client.params.ClientPNames;
import org.apache.http.client.params.CookiePolicy;
import org.apache.http.cookie.Cookie;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.params.CoreConnectionPNames;
import org.apache.http.params.HttpParams;

import com.almende.eve.state.State;
import com.almende.eve.state.StateBuilder;
import com.almende.util.jackson.JOM;
import com.fasterxml.jackson.databind.node.ObjectNode;

/**
 * The Class ApacheHttpClient.
 */
public final class ApacheHttpClient {
    private static final Logger LOG = Logger.getLogger(ApacheHttpClient.class.getCanonicalName());
    private static DefaultHttpClient httpClient = null;
    static {
        new ApacheHttpClient();
    }

    /**
     * Instantiates a new apache http client.
     * 
     */
    private ApacheHttpClient() {

        // generate httpclient
        httpClient = new DefaultHttpClient();

        // Set cookie policy and persistent cookieStore
        try {
            httpClient.setCookieStore(new MyCookieStore());
        } catch (final Exception e) {
            LOG.log(Level.WARNING, "Failed to initialize persistent cookieStore!", e);
        }
        final HttpParams params = httpClient.getParams();

        params.setParameter(ClientPNames.COOKIE_POLICY, CookiePolicy.BROWSER_COMPATIBILITY);
        params.setParameter(CoreConnectionPNames.SO_TIMEOUT, 60000);
        params.setParameter(CoreConnectionPNames.CONNECTION_TIMEOUT, 10000);
        params.setParameter(CoreConnectionPNames.STALE_CONNECTION_CHECK, false);
        httpClient.setParams(params);
    }

    /**
     * Gets the.
     * 
     * @return the default http client
     */
    public static HttpClient get() {
        return httpClient;
    }

    /**
     * The Class MyCookieStore.
     */
    class MyCookieStore implements CookieStore {
        // TODO: make COOKIESTORE config parameters

        /** The Constant COOKIESTORE. */
        static final String COOKIESTORE = "_CookieStore";

        /** The my state. */
        private State myState = null;

        /**
         * Instantiates a new my cookie store.
         * 
         * @throws IOException
         *             Signals that an I/O exception has occurred.
         */
        MyCookieStore() throws IOException {
            // TODO: use Config service
            final ObjectNode params = JOM.createObjectNode();
            params.put("class", "com.almende.eve.state.memory.MemoryStateService");
            params.put("id", COOKIESTORE);

            myState = new StateBuilder().withConfig(params).build();
        }

        /*
         * (non-Javadoc)
         * 
         * @see
         * org.apache.http.client.CookieStore#addCookie(org.apache.http.cookie
         * .Cookie)
         */
        @Override
        public void addCookie(final Cookie cookie) {
            myState.put(Integer.valueOf(COOKIESTORE.hashCode()).toString(), cookie);
        }

        /*
         * (non-Javadoc)
         * 
         * @see org.apache.http.client.CookieStore#getCookies()
         */
        @Override
        public List<Cookie> getCookies() {
            final List<Cookie> result = new ArrayList<Cookie>(myState.size());
            for (final String entryKey : myState.keySet()) {
                result.add(myState.get(entryKey, Cookie.class));
            }
            return result;
        }

        /*
         * (non-Javadoc)
         * 
         * @see org.apache.http.client.CookieStore#clearExpired(java.util.Date)
         */
        @Override
        public boolean clearExpired(final Date date) {
            boolean result = false;

            for (final String entryKey : myState.keySet()) {
                final Cookie cookie = myState.get(entryKey, Cookie.class);
                if (cookie.isExpired(date)) {
                    myState.remove(entryKey);
                    result = true;
                }
            }
            return result;
        }

        /*
         * (non-Javadoc)
         * 
         * @see org.apache.http.client.CookieStore#clear()
         */
        @Override
        public void clear() {
            myState.clear();
        }

        /**
         * Gets the my state.
         * 
         * @return the my state
         */
        public State getMyState() {
            return myState;
        }

        /**
         * Sets the my state.
         * 
         * @param myState
         *            the new my state
         */
        public void setMyState(final State myState) {
            this.myState = myState;
        }
    }
}