com.nesscomputing.httpclient.factory.httpclient4.InternalResponse Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ness-httpclient Show documentation
Ness HTTP client component.
There is a newer version: 2.0.5
/**
 * Copyright (C) 2012 Ness Computing, Inc.
 *
 * 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.
 */
package com.nesscomputing.httpclient.factory.httpclient4;

import com.nesscomputing.httpclient.HttpClientResponse;
import com.nesscomputing.logging.Log;

import java.io.IOException;
import java.io.InputStream;
import java.net.URI;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;

import javax.annotation.Nonnull;

import org.apache.commons.io.input.NullInputStream;
import org.apache.commons.lang3.StringUtils;
import org.apache.http.Header;
import org.apache.http.HeaderIterator;
import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.StatusLine;
import org.apache.http.client.methods.HttpRequestBase;
import org.apache.http.util.EntityUtils;

import com.google.common.collect.Lists;

/**
 * Apache HttpClient4 implementation of {@link HttpClientResponse}.
 */
final class InternalResponse implements HttpClientResponse
{
    private static final Log LOG = Log.findLog();

    private final HttpRequestBase httpRequest;
    private final HttpResponse httpResponse;

    InternalResponse(final HttpRequestBase httpRequest, final HttpResponse httpResponse)
    {
        this.httpRequest = httpRequest;
        this.httpResponse = httpResponse;
    }

    @Override
    public String getContentType()
    {
        if (httpResponse != null) {
            final HttpEntity entity = httpResponse.getEntity();
            if (entity != null) {
                final Header contentType = entity.getContentType();
                return (contentType == null) ? null : contentType.getValue();
            }
        }
        return null;
    }

    @Override
    public Long getContentLength()
    {
        if (httpResponse != null) {
            final Header header = httpResponse.getFirstHeader("Content-Length");
            if (header != null) {
                final String contentLen = StringUtils.trimToEmpty(header.getValue());
                LOG.debug("Response content length header is '%s'", contentLen);
                try {
                    return Long.parseLong(contentLen);
                }
                catch (NumberFormatException nfe) {
                    LOG.warnDebug(nfe, "Could not parse '%s'", contentLen);
                }
            }
        }
        return null;
    }

    @Override
    public String getCharset()
    {
        if (httpResponse != null) {
            final HttpEntity entity = httpResponse.getEntity();
            if (entity != null) {
                return EntityUtils.getContentCharSet(entity);
            }
        }
        return null;
    }

    @Override
    public String getHeader(final String name)
    {
        if (httpResponse != null) {
            final Header header = httpResponse.getFirstHeader(name);
            return (header == null) ? null : header.getValue();
        }
        return null;
    }

    @Override
    @Nonnull
    public List getHeaders(final String name)
    {
        final List values = new ArrayList();
        if (httpResponse != null) {
            for(HeaderIterator it = httpResponse.headerIterator(name); it.hasNext(); ) {
                final Header header = it.nextHeader();
                values.add(header.getValue());
            }
        }
        return values;
    }

	@Override
    @Nonnull
	public Map> getAllHeaders() {
        Map> headerMap = new TreeMap>(String.CASE_INSENSITIVE_ORDER);

        Header[] headers = httpResponse.getAllHeaders();

        for (Header header : headers) {
            String name = header.getName();
            String value = header.getValue();

            List valuesForThisHeader = headerMap.get(name);

            if (valuesForThisHeader == null) {
                valuesForThisHeader = Lists.newLinkedList();
                headerMap.put(name, valuesForThisHeader);
            }
            valuesForThisHeader.add(value);
        }

        return headerMap;
    }

    @Override
    public InputStream getResponseBodyAsStream() throws IOException
    {
        if (httpResponse != null) {
            final HttpEntity httpEntity = httpResponse.getEntity();
            if (httpEntity != null) {
                return httpEntity.getContent();
            }
        }
        return new NullInputStream(0);
    }

    @Override
    public int getStatusCode()
    {
        if (httpResponse != null) {
            final StatusLine statusLine = httpResponse.getStatusLine();
            return (statusLine == null) ? 0 : statusLine.getStatusCode();
        }
        return 0;
    }

    @Override
    public String getStatusText()
    {
        if (httpResponse != null) {
            final StatusLine statusLine = httpResponse.getStatusLine();
            return (statusLine == null) ? "" : statusLine.getReasonPhrase();
        }
        return "";
    }

    @Override
    public URI getUri()
    {
        return (httpRequest == null) ? null : httpRequest.getURI();
    }

    @Override
    public boolean isRedirected()
    {
        final int statusCode = getStatusCode();
        return (statusCode >= 300) && (statusCode <= 399);
    }

    @Override
    public String toString() {
        return String.format("InternalResponse [getContentType()=%s, getContentLength()=%s," +
            		" getCharset()=%s, getAllHeaders()=%s, getStatusCode()=%s," +
            		" getStatusText()=%s, getUri()=%s, isRedirected()=%s]",
                getContentType(), getContentLength(), getCharset(),
                getAllHeaders(), getStatusCode(), getStatusText(),
                getUri(), isRedirected());
    }
}