com.distelli.gcr.http.GcrHttpResponse Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of gcr-sdk-java Show documentation

Provides GCR API access for various Distelli software components.

There is a newer version: 2.1.3

/*
  $Id: $
  @file GcrHttpResponse.java
  @brief Contains the GcrHttpResponse.java class

  @author Rahul Singh [rsingh]
  Copyright (c) 2013, Distelli Inc., All Rights Reserved.
*/
package com.distelli.gcr.http;

import java.io.IOException;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;

import org.apache.http.Header;
import org.apache.http.HttpResponse;
import org.apache.http.HttpEntity;
import org.apache.http.client.methods.HttpRequestBase;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class GcrHttpResponse
{
    private static final Logger log = LoggerFactory.getLogger(GcrHttpResponse.class);
    private HttpResponse _httpResponse = null;
    private HttpRequestBase _httpRequest = null;
    private Map _headers = null;
    private static final ObjectMapper OBJECT_MAPPER = new ObjectMapper();

    public GcrHttpResponse(HttpRequestBase httpRequest, HttpResponse httpResponse)
    {
        _httpResponse = httpResponse;
        _httpRequest = httpRequest;
        _headers = new HashMap();
        Header[] responseHeaders = _httpResponse.getAllHeaders();
        for(Header header : responseHeaders)
            _headers.put(header.getName(), header.getValue());
    }

    public int getHttpStatusCode() {
        return _httpResponse.getStatusLine().getStatusCode();
    }

    public Map getResponseHeaders() {
        return _headers;
    }

    public String getResponseHeader(String headerName) {
        return _headers.get(headerName);
    }

    public String getResponseAsJsonString()
        throws IOException
    {
        JsonNode jsonNode = getResponseAsJsonNode();
        if(jsonNode == null)
            return null;
        return OBJECT_MAPPER.writeValueAsString(jsonNode);
    }

    public JsonNode getResponseAsJsonNode()
        throws IOException
    {
        InputStream in = getInputStream();
        if(in == null)
            return null;
        return OBJECT_MAPPER.readTree(in);
    }

    public InputStream getInputStream()
        throws IOException
    {
        HttpEntity entity = _httpResponse.getEntity();
        if(entity == null)
            return null;

        InputStream result = entity.getContent();
        if(result != null)
            return result;

        return null;
    }
}