HttpPirasoGetBridgeRegistryHandler.java

package org.piraso.server.bridge.net;

import org.apache.commons.lang.Validate;
import org.apache.http.*;
import org.apache.http.client.HttpClient;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.message.BasicNameValuePair;
import org.apache.http.protocol.HttpContext;
import org.apache.http.util.EntityUtils;
import org.apache.log4j.Logger;
import org.codehaus.jackson.map.ObjectMapper;
import org.piraso.api.JacksonUtils;
import org.piraso.client.net.AbstractHttpHandler;
import org.piraso.client.net.HttpPirasoException;
import org.piraso.server.service.BridgeRegistry;
import org.xml.sax.SAXException;

import javax.xml.parsers.ParserConfigurationException;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import static org.piraso.api.PirasoConstants.*;

public class HttpPirasoGetBridgeRegistryHandler extends AbstractHttpHandler {
    private static final Logger LOG = Logger.getLogger(HttpPirasoGetBridgeRegistryHandler.class);

    private HttpEntity responseEntity;

    private ObjectMapper mapper;

    private BridgeRegistry registry;

    public HttpPirasoGetBridgeRegistryHandler(HttpClient client, HttpContext context) {
        super(client, context);

        mapper = JacksonUtils.MAPPER;
    }

    @Override
    public void execute() throws IOException, SAXException, ParserConfigurationException {
        try {
            doExecute();
        } finally {
            EntityUtils.consume(responseEntity);
        }
    }

    private void doExecute() throws IOException, SAXException, ParserConfigurationException {
        Validate.notNull(uri, "uri should not be null.");

        LOG.info(String.format("Executing %s...", uri));

        HttpPost post = new HttpPost(uri.getPath());

        List<NameValuePair> params = new ArrayList<NameValuePair>();
        params.add(new BasicNameValuePair(SERVICE_PARAMETER, SERVICE_GET_REGISTRY_PARAMETER_VALUE));

        post.setEntity(new UrlEncodedFormEntity(params, ENCODING_UTF_8));

        HttpResponse response = client.execute(targetHost, post, context);
        StatusLine status = response.getStatusLine();

        if(status.getStatusCode() != HttpStatus.SC_OK) {
            throw new HttpPirasoException(status.toString());
        }

        responseEntity = response.getEntity();
        String contentType = responseEntity.getContentType().getValue().toLowerCase();

        if(!contentType.contains(JSON_CONTENT_TYPE)) {
            throw new HttpPirasoException("Invalid response content type: " + responseEntity.getContentType());
        }

        this.registry = mapper.readValue(responseEntity.getContent(), BridgeRegistry.class);
    }

    public BridgeRegistry getRegistry() {
        return registry;
    }
}