org.robertburrelldonkin.template4couchdb.rest.HttpClientRestClient.java Source code

Introduction

Here is the source code for org.robertburrelldonkin.template4couchdb.rest.HttpClientRestClient.java

Source

/*
*    Copyright 2012 Robert Burrell Donkin robertburrelldonkin.name
* 
*   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 org.robertburrelldonkin.template4couchdb.rest;

import java.io.IOException;

import org.apache.http.client.ClientProtocolException;
import org.apache.http.client.HttpClient;
import org.apache.http.client.ResponseHandler;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.conn.ClientConnectionManager;
import org.apache.http.entity.ContentProducer;
import org.apache.http.entity.ContentType;
import org.apache.http.entity.EntityTemplate;
import org.robertburrelldonkin.template4couchdb.IDocumentMarshaller;
import org.robertburrelldonkin.template4couchdb.IDocumentUnmarshaller;
import org.robertburrelldonkin.template4couchdb.IRestClient;

public class HttpClientRestClient implements IRestClient {

    private final ICodecFactory codecFactory;

    private final HttpClient httpClient;

    public HttpClientRestClient(final HttpClient httpClient, final ICodecFactory responseHandlerFactory) {
        super();
        this.httpClient = httpClient;
        this.codecFactory = responseHandlerFactory;
    }

    public <T> T get(final String url, final IDocumentUnmarshaller<T> mapper) {
        final ResponseHandler<T> handler = codecFactory.handlerFor(mapper);
        try {
            return httpClient.execute(new HttpGet(url), handler);
        } catch (ClientProtocolException e) {
            throw new HttpClientRestClientException(e);
        } catch (IOException e) {
            throw new HttpClientRestClientException(e);
        }
    }

    public void shutdown() {
        final ClientConnectionManager connectionManager = this.httpClient.getConnectionManager();
        if (connectionManager != null) {
            connectionManager.shutdown();
        }
    }

    @Override
    public <R, D> R post(String url, IDocumentMarshaller<D> documentMarshaller, D document,
            IDocumentUnmarshaller<R> responseUnmarshaller) {
        final ContentProducer producer = codecFactory.producerFor(documentMarshaller, document);
        final EntityTemplate entity = new EntityTemplate(producer);
        entity.setContentType(ContentType.APPLICATION_JSON.toString());
        final ResponseHandler<R> handler = codecFactory.handlerFor(responseUnmarshaller);
        final HttpPost post = new HttpPost(url);
        post.setEntity(entity);
        final R response;
        try {
            response = httpClient.execute(post, handler);
        } catch (ClientProtocolException e) {
            throw new HttpClientRestClientException(e);
        } catch (IOException e) {
            throw new HttpClientRestClientException(e);
        }
        return response;
    }
}