com.google.code.jerseyclients.asynchttpclient.AsyncHttpClientJerseyClientHandler.java Source code

Java tutorial

Introduction

Here is the source code for com.google.code.jerseyclients.asynchttpclient.AsyncHttpClientJerseyClientHandler.java

Source

package com.google.code.jerseyclients.asynchttpclient;

/*
 * Olivier Lamy
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.
 */

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.net.SocketTimeoutException;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.concurrent.Future;
import java.util.concurrent.TimeoutException;

import javax.ws.rs.core.MultivaluedMap;

import org.apache.commons.io.IOUtils;
import org.apache.commons.lang.StringUtils;
import org.apache.commons.lang.time.StopWatch;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.google.code.jerseyclients.JerseyHttpClientConfig;
import com.ning.http.client.AsyncHttpClient;
import com.ning.http.client.AsyncHttpClient.BoundRequestBuilder;
import com.ning.http.client.FluentCaseInsensitiveStringsMap;
import com.ning.http.client.PerRequestConfig;
import com.ning.http.client.ProxyServer;
import com.ning.http.client.Response;
import com.sun.jersey.api.client.ClientHandlerException;
import com.sun.jersey.api.client.ClientRequest;
import com.sun.jersey.api.client.ClientResponse;
import com.sun.jersey.api.client.ClientResponse.Status;
import com.sun.jersey.api.client.CommittingOutputStream;
import com.sun.jersey.api.client.TerminatingClientHandler;
import com.sun.jersey.core.header.InBoundHeaders;

/**
 * created 1 oct. 2010
 * @author <a href="mailto:olamy@apache.org">Olivier Lamy</a> 
 * @version $Id$
 */
public class AsyncHttpClientJerseyClientHandler extends TerminatingClientHandler {
    private Logger log = LoggerFactory.getLogger(getClass());

    private AsyncHttpClient client;

    private JerseyHttpClientConfig jerseyHttpClientConfig;

    public AsyncHttpClientJerseyClientHandler(AsyncHttpClient client, JerseyHttpClientConfig config) {
        this.client = client;
        this.jerseyHttpClientConfig = config;
    }

    /**
     * @see com.sun.jersey.api.client.ClientHandler#handle(com.sun.jersey.api.client.ClientRequest)
     */
    public ClientResponse handle(final ClientRequest clientRequest) throws ClientHandlerException {

        final BoundRequestBuilder boundRequestBuilder = getBoundRequestBuilder(clientRequest);

        PerRequestConfig perRequestConfig = new PerRequestConfig();
        perRequestConfig.setRequestTimeoutInMs(this.jerseyHttpClientConfig.getReadTimeOut());

        if (this.jerseyHttpClientConfig.getProxyInformation() != null) {
            ProxyServer proxyServer = new ProxyServer(jerseyHttpClientConfig.getProxyInformation().getProxyHost(),
                    jerseyHttpClientConfig.getProxyInformation().getProxyPort());
            perRequestConfig = new PerRequestConfig(proxyServer, this.jerseyHttpClientConfig.getReadTimeOut());
        }

        boundRequestBuilder.setPerRequestConfig(perRequestConfig);

        if (this.jerseyHttpClientConfig.getApplicationCode() != null) {
            boundRequestBuilder.addHeader(this.jerseyHttpClientConfig.getApplicationCodeHeader(),
                    this.jerseyHttpClientConfig.getApplicationCode());
        }
        if (this.jerseyHttpClientConfig.getOptionnalHeaders() != null) {
            for (Entry<String, String> entry : this.jerseyHttpClientConfig.getOptionnalHeaders().entrySet()) {
                boundRequestBuilder.addHeader(entry.getKey(), entry.getValue());
            }
        }

        if (StringUtils.equalsIgnoreCase("POST", clientRequest.getMethod())) {

            if (clientRequest.getEntity() != null) {
                final RequestEntityWriter re = getRequestEntityWriter(clientRequest);

                ByteArrayOutputStream baos = new ByteArrayOutputStream();
                try {
                    re.writeRequestEntity(new CommittingOutputStream(baos) {
                        @Override
                        protected void commit() throws IOException {
                            writeOutBoundHeaders(clientRequest.getHeaders(), boundRequestBuilder);
                        }
                    });
                } catch (IOException ex) {
                    throw new ClientHandlerException(ex);
                }

                boundRequestBuilder.setBody(new ByteArrayInputStream(baos.toByteArray()));

            }
        } else {
            writeOutBoundHeaders(clientRequest.getHeaders(), boundRequestBuilder);
        }
        try {
            StopWatch stopWatch = new StopWatch();
            stopWatch.reset();
            stopWatch.start();
            Future<Response> futureResponse = boundRequestBuilder.execute();
            Response response = futureResponse.get();
            int httpReturnCode = response.getStatusCode();
            stopWatch.stop();
            log.info("time to call rest url " + clientRequest.getURI() + ", " + stopWatch.getTime() + " ms");
            // in case of empty content returned we returned an empty stream
            // to return a null object
            if (httpReturnCode == Status.NO_CONTENT.getStatusCode()) {
                new ClientResponse(httpReturnCode, getInBoundHeaders(response), IOUtils.toInputStream(""),
                        getMessageBodyWorkers());
            }
            return new ClientResponse(httpReturnCode, getInBoundHeaders(response),
                    response.getResponseBodyAsStream() == null ? IOUtils.toInputStream("")
                            : response.getResponseBodyAsStream(),
                    getMessageBodyWorkers());
        } catch (Exception e) {
            if (e.getCause() != null && (e.getCause() instanceof TimeoutException)) {
                throw new ClientHandlerException(new SocketTimeoutException());
            }
            throw new ClientHandlerException(e);
        }
    }

    /**
     * @param httpResponse
     * @return
     */
    private InBoundHeaders getInBoundHeaders(Response response) {
        InBoundHeaders headers = new InBoundHeaders();
        FluentCaseInsensitiveStringsMap headersMap = response.getHeaders();
        if (headersMap == null || headersMap.isEmpty()) {
            return headers;
        }

        for (String key : headersMap.keySet()) {
            List<String> list = headersMap.get(key);
            headers.put(key, list);
        }
        return headers;
    }

    private void writeOutBoundHeaders(MultivaluedMap<String, Object> metadata,
            BoundRequestBuilder boundRequestBuilder) {
        for (Map.Entry<String, List<Object>> e : metadata.entrySet()) {
            List<Object> vs = e.getValue();
            if (vs.size() == 1) {
                boundRequestBuilder.addHeader(e.getKey(), headerValueToString(vs.get(0)));
            } else {
                StringBuilder b = new StringBuilder();
                for (Object v : e.getValue()) {
                    if (b.length() > 0) {
                        b.append(',');
                    }
                    b.append(headerValueToString(v));
                }
                boundRequestBuilder.addHeader(e.getKey(), b.toString());
            }
        }
    }

    private BoundRequestBuilder getBoundRequestBuilder(ClientRequest cr) {
        final String strMethod = cr.getMethod();
        final String uri = cr.getURI().toString();
        if (strMethod.equals("GET")) {
            return this.client.prepareGet(uri);
        } else if (strMethod.equals("POST")) {
            return this.client.preparePost(uri);
        } else if (strMethod.equals("PUT")) {
            return this.client.preparePut(uri);
        } else if (strMethod.equals("DELETE")) {
            return this.client.prepareDelete(uri);
        } else if (strMethod.equals("HEAD")) {
            return this.client.prepareHead(uri);
        } else if (strMethod.equals("OPTIONS")) {
            return this.client.prepareOptions(uri);
        } else {
            throw new ClientHandlerException("Method " + strMethod + " is not supported.");
        }
    }

}