org.lastbamboo.common.http.client.BaseHttpClientRequester Maven / Gradle / Ivy

Go to download
package org.lastbamboo.common.http.client;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.StringReader;
import java.util.Collection;
import java.util.Map;
import java.util.zip.GZIPInputStream;

import org.apache.commons.httpclient.Header;
import org.apache.commons.httpclient.HttpMethod;
import org.apache.commons.httpclient.HttpStatus;
import org.apache.commons.httpclient.StatusLine;
import org.apache.commons.httpclient.methods.GetMethod;
import org.apache.commons.httpclient.methods.PostMethod;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang.StringUtils;
import org.littleshoot.util.DefaultHttpClient;
import org.littleshoot.util.DefaultHttpClientImpl;
import org.littleshoot.util.Pair;
import org.littleshoot.util.UriUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Issues an HTTP request using HTTP client. 
 */
public class BaseHttpClientRequester
    {

    private final Logger LOG = LoggerFactory.getLogger(getClass());
    private final DefaultHttpClient m_httpClient = 
        new DefaultHttpClientImpl();

    public String post(final String baseUrl, 
        final Collection> parameters) throws IOException, 
        ServiceUnavailableException
        {
        final String url = UriUtils.newUrl(baseUrl, parameters);
        final PostMethod method = new PostMethod(url);
        return request(method);
        }
    
    public String post(final String baseUrl, 
        final Map parameters) throws IOException, 
        ServiceUnavailableException
        {
        final String url = UriUtils.newUrl(baseUrl, parameters);
        final PostMethod method = new PostMethod(url);
        return request(method);
        }
    
    
    public String get(final String baseUrl, 
        final Collection> parameters) 
        throws IOException, ServiceUnavailableException
        {
        final String url = UriUtils.newUrl(baseUrl, parameters);
        final GetMethod method = new GetMethod(url);
        return request(method);
        }

    
    public String get(final String baseUrl, 
        final Map parameters) 
        throws IOException, ServiceUnavailableException
        {
        final String url = UriUtils.newUrl(baseUrl, parameters);
        final GetMethod method = new GetMethod(url);
        return request(method);
        }
    
    private String request(final HttpMethod method) throws IOException, 
        ServiceUnavailableException
        {
        method.setRequestHeader("Accept-Encoding", "gzip");
        InputStream is = null;
        try
            {
            this.m_httpClient.executeMethod(method);
            final int statusCode = method.getStatusCode();
            final StatusLine statusLine = method.getStatusLine();
            final Header encoding = 
                method.getResponseHeader("Content-Encoding");
            if (encoding != null && encoding.getValue().equals("gzip"))
                {
                LOG.debug("Unzipping body...");
                is = new GZIPInputStream(method.getResponseBodyAsStream());
                }
            else
                {
                is = method.getResponseBodyAsStream();
                }
            final String body = IOUtils.toString(is);
            if (StringUtils.isBlank(body))
                {
                // Could easily be a post request, which would not have a body.
                LOG.debug("No response body.  Post request?");
                }
            
            if (statusCode == HttpStatus.SC_SERVICE_UNAVAILABLE)
                {
                final String msg = "Got 503 Service Unavailable " + 
                    method.getURI() + "\n" +
                    statusLine + "\n" + body;
                LOG.warn(msg);
                throw new ServiceUnavailableException(msg);
                }
            if (statusCode != HttpStatus.SC_OK)
                {
                final String msg = "NO 200 OK: " + method.getURI() + "\n" +
                    statusLine + "\n" + body;
                LOG.warn(msg);
                writeToFile(body);
                throw new IOException(msg);
                }
            else
                {
                LOG.debug("Got 200 response...");
                }
            
            return body;
            }
        finally
            {
            IOUtils.closeQuietly(is);
            method.releaseConnection();
            }
        
        }

    private void writeToFile(final String body)
        {
        final File out = new File(getClass().getSimpleName()+"-Error.html");
        if (out.exists())
            {
            out.delete();
            }
        final StringReader is = new StringReader(body);
        try
            {
            IOUtils.copy(is, new FileOutputStream(out));
            }
        catch (final FileNotFoundException e)
            {
            LOG.debug("FNF", e);
            }
        catch (final IOException e)
            {
            LOG.debug("IO", e);
            }
        }

    }