org.expath.httpclient.impl.HttpRequestImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of http-client-java Show documentation
Java Library implementing the core HTTP Client module features
The newest version!
/****************************************************************************/
/*  File:       HttpRequestImpl.java                                        */
/*  Author:     F. Georges - fgeorges.org                                   */
/*  Date:       2009-02-02                                                  */
/*  Tags:                                                                   */
/*      Copyright (c) 2009 Florent Georges (see end of file.)               */
/* ------------------------------------------------------------------------ */


package org.expath.httpclient.impl;

import org.expath.httpclient.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.apache.http.Header;
import org.expath.httpclient.model.Result;

import javax.annotation.Nullable;

/**
 * Implementation of a HTTP request.
 *
 * @author Florent Georges
 */
public class HttpRequestImpl
        implements HttpRequest
{
    public HttpResponse send(Result result, HttpConnection conn, HttpCredentials cred)
            throws HttpClientException
    {
        long start = System.currentTimeMillis();
        if ( myHeaders == null ) {
            myHeaders = new HeaderSet();
        }
        conn.setRequestMethod(myMethod, myBody != null);
        conn.setRequestHeaders(myHeaders);
        if ( myHttpVer != null ) {
            conn.setHttpVersion(myHttpVer);
        }
        if ( myTimeout != null ) {
            conn.setTimeout(myTimeout);
        }
        if ( myGzip ) {
            conn.setGzip(true);
        }
        conn.setChunked(isChunked());
        conn.setPreemptiveAuthentication(isPreemptiveAuthentication());

        conn.setFollowRedirect(myFollowRedirect);
        conn.connect(myBody, cred);
        int status = conn.getResponseStatus();
        String msg = conn.getResponseMessage();
        HttpResponseBody body = null;
        if ( ! myStatusOnly ) {
            ContentType type = getContentType(conn.getResponseHeaders());
            if ( type == null ) {
                // FIXME: We should probably rather fall back to octet-stream...
                LOG.debug("There is no Content-Type, we assume there is no content");
            }
            else {
                body = BodyFactory.makeResponseBody(result, type, conn);
            }
        }
        long stop = System.currentTimeMillis();
        long spent = stop - start;
        HttpResponse resp = new HttpResponse(status, msg, conn.getResponseHeaders(), body, spent);
        result.add(resp);
        return resp;
    }

    private @Nullable ContentType getContentType(final HeaderSet headers)
            throws HttpClientException
    {
        final Header header = headers.getFirstHeader("Content-Type");
        return ContentType.parse(header, myOverrideType, myDefaultCharset);
    }

    @Override
    public String getMethod()
    {
        return myMethod;
    }

    @Override
    public void setMethod(String method)
    {
        myMethod = method;
    }

    @Override
    public String getHref()
    {
        return myHref;
    }

    @Override
    public void setHref(String href)
    {
        myHref = href;
    }

    @Override
    public String getHttpVersion()
    {
        return myHttpVer;
    }

    @Override
    public void setHttpVersion(String ver)
            throws HttpClientException
    {
        if ( HttpConstants.HTTP_1_0.equals(ver) ) {
            myHttpVer = HttpConstants.HTTP_1_0;
        }
        else if ( HttpConstants.HTTP_1_1.equals(ver) ) {
            myHttpVer = HttpConstants.HTTP_1_1;
        }
        else {
            throw new HttpClientException(HttpClientError.HC005, "Unknown HTTP version: '" + ver + "'");
        }
    }

    @Override
    public void setDefaultCharset(final String charset) {
        myDefaultCharset = charset;
    }

    @Override
    public void setOverrideType(String type)
    {
        myOverrideType = type;
    }

    @Override
    public void setHeaders(HeaderSet headers)
    {
        myHeaders = headers;
    }

    @Override
    public void setBody(HttpRequestBody body)
            throws HttpClientException
    {
        myBody = body;
        body.setHeaders(myHeaders);
    }

    @Override
    public void setStatusOnly(boolean only)
    {
        myStatusOnly = only;
    }

    @Override
    public void setFollowRedirect(boolean follow)
    {
        myFollowRedirect = follow;
    }

    @Override
    public void setTimeout(Integer seconds)
    {
        myTimeout = seconds;
    }

    @Override
    public void setGzip(final boolean gzip) {
        myGzip = gzip;
    }

    @Override
    public boolean isChunked() {
        if(myChunked == null) {
            if(myHttpVer != null && myHttpVer.equals(HttpConstants.HTTP_1_0)) {
                return false;
            } else {
                return true;
            }
        } else {
            return myChunked;
        }
    }

    @Override
    public void setChunked(boolean chunked) {
        this.myChunked = chunked;
    }

    @Override
    public boolean isPreemptiveAuthentication() {
        return myPreemptiveAuthentication;
    }

    @Override
    public void setPreemptiveAuthentication(final boolean preemptiveAuthentication) {
        this.myPreemptiveAuthentication = preemptiveAuthentication;
    }

    private String myMethod;
    private String myHref;
    private String myHttpVer;
    private String myDefaultCharset;
    private String myOverrideType;
    private boolean myStatusOnly;
    private boolean myFollowRedirect = true;
    private Integer myTimeout = null;
    private boolean myGzip = false;
    private Boolean myChunked = null;
    private boolean myPreemptiveAuthentication = false;
    private HeaderSet myHeaders;
    private HttpRequestBody myBody;
    private static final Logger LOG = LoggerFactory.getLogger(HttpRequestImpl.class);
}


/* ------------------------------------------------------------------------ */
/*  DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS COMMENT.               */
/*                                                                          */
/*  The contents of this file are subject to the Mozilla Public License     */
/*  Version 1.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.mozilla.org/MPL/.                                            */
/*                                                                          */
/*  Software distributed under the License is distributed on an "AS IS"     */
/*  basis, WITHOUT WARRANTY OF ANY KIND, either express or implied.  See    */
/*  the License for the specific language governing rights and limitations  */
/*  under the License.                                                      */
/*                                                                          */
/*  The Original Code is: all this file.                                    */
/*                                                                          */
/*  The Initial Developer of the Original Code is Florent Georges.          */
/*                                                                          */
/*  Contributor(s): none.                                                   */
/* ------------------------------------------------------------------------ */