gerrit.onap Code Review - so/libs.git/blob

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * Licensed under the Apache License, Version 2.0 (the "License");
   4  * you may not use this file except in compliance with the License.
   5  * You may obtain a copy of the License at
   6  *
   7  *      http://www.apache.org/licenses/LICENSE-2.0
   8  *
   9  * Unless required by applicable law or agreed to in writing, software
  10  * distributed under the License is distributed on an "AS IS" BASIS,
  11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12  * See the License for the specific language governing permissions and
  13  * limitations under the License.
  14  * ============LICENSE_END=========================================================
  15  */
  16
  17 /*
  18  * ============LICENSE_START==========================================
  19  * ===================================================================
  20  * Copyright (c) 2017 AT&T Intellectual Property. All rights reserved.
  21  * ===================================================================
  22  * Licensed under the Apache License, Version 2.0 (the "License");
  23  * you may not use this file except in compliance with the License.
  24  * You may obtain a copy of the License at
  25  *
  26  *        http://www.apache.org/licenses/LICENSE-2.0
  27  *
  28  * Unless required by applicable law or agreed to in writing, software
  29  * distributed under the License is distributed on an "AS IS" BASIS,
  30  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  31  * See the License for the specific language governing permissions and
  32  * limitations under the License.
  33  * ============LICENSE_END============================================
  34  *
  35  * ECOMP and OpenECOMP are trademarks
  36  * and service marks of AT&T Intellectual Property.
  37  *
  38  */
  39
  40 package com.woorea.openstack.connector;
  41
  42 import org.apache.http.Header;
  43 import org.apache.http.HttpResponse;
  44 import org.apache.log4j.Logger;
  45
  46 import com.fasterxml.jackson.databind.ObjectMapper;
  47 import com.woorea.openstack.base.client.OpenStackResponse;
  48
  49 import java.io.ByteArrayInputStream;
  50 import java.io.ByteArrayOutputStream;
  51 import java.io.IOException;
  52 import java.io.InputStream;
  53 import java.util.HashMap;
  54 import java.util.Map;
  55
  56 public class HttpClientResponse implements OpenStackResponse {
  57
  58     private static Logger LOGGER = Logger.getLogger(HttpClientConnector.class);
  59
  60     private HttpResponse response = null;
  61     private String entityBody = null;
  62
  63     public HttpClientResponse(HttpResponse response)
  64     {
  65         this.response = response;
  66
  67         // Read the body so InputStream can be closed
  68         if (response.getEntity() == null) {
  69             // No body
  70             LOGGER.debug ("No Response Body");
  71             return;
  72         }
  73
  74         ByteArrayOutputStream responseBody = new ByteArrayOutputStream();
  75         try {
  76             response.getEntity().writeTo(responseBody);
  77         } catch (IOException e) {
  78             throw new HttpClientException ("Error Reading Response Body", e);
  79         }
  80         entityBody = responseBody.toString();
  81         LOGGER.debug (entityBody);
  82     }
  83
  84
  85     @Override
  86     public <T> T getEntity (Class<T> returnType) {
  87         // Get appropriate mapper, based on existence of a root element
  88         ObjectMapper mapper = HttpClientConnector.getObjectMapper (returnType);
  89
  90         T resp = null;
  91         try {
  92             resp = mapper.readValue(entityBody, returnType);
  93         } catch (Exception e) {
  94             throw new HttpClientException ("Caught exception in getEntity", e);
  95         }
  96         return resp;
  97     }
  98
  99     @Override
 100     public <T> T getErrorEntity(Class<T> returnType) {
 101         return getEntity(returnType);
 102     }
 103
 104     @Override
 105     public InputStream getInputStream() {
 106            return new ByteArrayInputStream (entityBody.getBytes());
 107     }
 108
 109     @Override
 110     public String header(String name) {
 111         return response.getFirstHeader(name).getValue();
 112     }
 113
 114     @Override
 115     public Map<String, String> headers() {
 116         Map<String, String> headers = new HashMap<String, String>();
 117
 118         Header responseHeaders[] = response.getAllHeaders();
 119         for (Header h : responseHeaders) {
 120             headers.put(h.getName(), h.getValue());
 121         }
 122
 123         return headers;
 124     }
 125
 126 }