com.hubspot.horizon.apache.internal.ApacheHttpResponse Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of HorizonApache Show documentation

There is a newer version: 0.3.1

package com.hubspot.horizon.apache.internal;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.hubspot.horizon.Header;
import com.hubspot.horizon.Headers;
import com.hubspot.horizon.HttpRequest;
import com.hubspot.horizon.internal.AbstractHttpResponse;
import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;

public class ApacheHttpResponse extends AbstractHttpResponse {
  private final HttpRequest request;
  private final int statusCode;
  private final Headers headers;
  private final InputStream responseStream;
  private final ObjectMapper mapper;

  public ApacheHttpResponse(HttpRequest request, HttpResponse apacheResponse, ObjectMapper mapper) throws IOException {
    this.request = request;
    this.statusCode = apacheResponse.getStatusLine().getStatusCode();
    this.headers = convertHeaders(apacheResponse.getAllHeaders());
    this.responseStream = extractResponseStream(apacheResponse);
    this.mapper = mapper;
  }

  private Headers convertHeaders(org.apache.http.Header[] apacheHeaders) {
    List headers = new ArrayList();

    for (org.apache.http.Header apacheHeader : apacheHeaders) {
      headers.add(new Header(apacheHeader.getName(), apacheHeader.getValue()));
    }

    return new Headers(headers);
  }

  private InputStream extractResponseStream(HttpResponse apacheResponse) throws IOException {
    HttpEntity entity = apacheResponse.getEntity();
    if (entity == null) {
      return EmptyInputStream.getInstance();
    } else {
      InputStream responseStream = entity.getContent();
      return responseStream == null ? EmptyInputStream.getInstance() : responseStream;
    }
  }

  @Override
  public ObjectMapper getObjectMapper() {
    return mapper;
  }

  @Override
  public HttpRequest getRequest() {
    return request;
  }

  @Override
  public int getStatusCode() {
    return statusCode;
  }

  @Override
  public Headers getHeaders() {
    return headers;
  }

  @Override
  public InputStream getAsInputStream() {
    return responseStream;
  }
}