huawei/vnfmadapter/VnfmadapterService/service/src/main/java/org/onap/vfc/nfvo/vnfm/svnfm/vnfmadapter/common/restclient/RestHttpContentExchange.java

   1 /*
   2  * Copyright 2016 Huawei Technologies Co., Ltd.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package org.onap.vfc.nfvo.vnfm.svnfm.vnfmadapter.common.restclient;
  18
  19 import java.io.ByteArrayInputStream;
  20 import java.io.IOException;
  21 import java.io.InputStreamReader;
  22 import java.nio.charset.Charset;
  23 import java.util.Enumeration;
  24 import java.util.HashMap;
  25 import java.util.Map;
  26 import java.util.zip.GZIPInputStream;
  27
  28 import org.apache.commons.lang.StringUtils;
  29 import org.eclipse.jetty.client.ContentExchange;
  30 import org.eclipse.jetty.client.HttpDestination;
  31 import org.eclipse.jetty.http.HttpFields;
  32 import org.eclipse.jetty.http.HttpHeaders;
  33 import org.eclipse.jetty.io.Buffer;
  34 import org.eclipse.jetty.util.StringUtil;
  35 import org.slf4j.Logger;
  36 import org.slf4j.LoggerFactory;
  37
  38 /**
  39  * ContentExchange implementation classe to provide access to response.
  40  * <br/>
  41  * <p>
  42  * </p>
  43  *
  44  * @author
  45  * @version 28-May-2016
  46  */
  47 public class RestHttpContentExchange extends ContentExchange {
  48
  49     private static final Logger LOGGER = LoggerFactory.getLogger(RestHttpContentExchange.class);
  50
  51     private static final String PATH = "path:";
  52
  53     private boolean gzip = false;
  54
  55     private RestfulAsyncCallback callback = null;
  56
  57     /**
  58      * Constructor<br/>
  59      * <p>
  60      * </p>
  61      *
  62      * @since
  63      * @param cacheFields whether to cache response header.
  64      * @param asyncCallback callback method.
  65      */
  66     RestHttpContentExchange(final boolean cacheFields, final RestfulAsyncCallback asyncCallback) {
  67         super(cacheFields);
  68         this.callback = asyncCallback;
  69     }
  70
  71     /**
  72      * Extract message.
  73      * <br/>
  74      *
  75      * @param data GZipped data.
  76      * @return Uncompressed data.
  77      * @throws IOException
  78      * @since
  79      */
  80     public String decompressGzipToStr(final byte[] data) throws IOException {
  81         if(data == null) {
  82             return "";
  83         }
  84         final StringBuilder out = new StringBuilder();
  85         try (ByteArrayInputStream input = new ByteArrayInputStream(data);
  86              GZIPInputStream gzis = new GZIPInputStream(input);
  87              InputStreamReader reader = new InputStreamReader(gzis, Charset.forName(RestfulClientConst.ENCODING));) {
  88             final char[] buff = new char[1024];
  89             for(int n; (n = reader.read(buff)) != -1;) {
  90                 out.append(new String(buff, 0, n));
  91             }
  92         }
  93         return out.toString();
  94
  95     }
  96
  97     /**
  98      * View response headers Content-Encoding values if you need to extract data.<br/>
  99      *
 100      * @param name buffer
 101      * @param value value
 102      * @throws IOException
 103      * @since
 104      */
 105     @Override
 106     protected synchronized void onResponseHeader(final Buffer name, final Buffer value) throws IOException {
 107         super.onResponseHeader(name, value);
 108         final int header = HttpHeaders.CACHE.getOrdinal(name);
 109         if(header == HttpHeaders.CONTENT_ENCODING_ORDINAL) {
 110             final String encoding = StringUtil.asciiToLowerCase(value.toString());
 111             gzip = encoding != null && StringUtils.contains(encoding, "gzip");
 112         }
 113
 114     }
 115
 116     @Override
 117     protected void onResponseComplete() throws IOException {
 118         if(LOGGER.isInfoEnabled()) {
 119             LOGGER.info("Response has Complete:" + PATH + this.getRequestURI().replace("\n", "0x0A"));
 120         }
 121         super.onResponseComplete();
 122         if(callback != null) {
 123             final RestfulResponse rsp = getResponse();
 124             callback.callback(rsp);
 125         }
 126     }
 127
 128     @Override
 129     protected void onRequestCommitted() throws IOException {
 130         if(LOGGER.isInfoEnabled()) {
 131             LOGGER.info("Request Header has been send:" + PATH + this.getRequestURI().replace("\n", "0x0A"));
 132         }
 133         super.onRequestCommitted();
 134     }
 135
 136     @Override
 137     protected void onRequestComplete() throws IOException {
 138         if(LOGGER.isInfoEnabled()) {
 139             LOGGER.info("Request has bend send complete:" + PATH + this.getRequestURI().replace("\n", "0x0A"));
 140         }
 141         super.onRequestComplete();
 142     }
 143
 144     @Override
 145     protected void onException(final Throwable x) {
 146         LOGGER.warn("onException:", x);
 147         super.onException(x);
 148         if(callback != null) {
 149             callback.handleExcepion(x);
 150         }
 151     }
 152
 153     @Override
 154     protected void onConnectionFailed(final Throwable x) {
 155         LOGGER.warn("onConnectionFailed:", x);
 156         super.onConnectionFailed(x);
 157         if(callback != null) {
 158             callback.handleExcepion(x);
 159         }
 160
 161     }
 162
 163     @Override
 164     protected void expire(final HttpDestination destination) {
 165         super.expire(destination);
 166         if(callback != null) {
 167             callback.handleExcepion(new ServiceException("request is expired, status:" + toState(getStatus())));
 168         }
 169     }
 170
 171     public boolean isGzip() {
 172         return gzip;
 173     }
 174
 175     /**
 176      * Get the response as RestfulResponse.
 177      * <br/>
 178      *
 179      * @return response object.
 180      * @throws IOException
 181      * @since
 182      */
 183     public RestfulResponse getResponse() throws IOException {
 184         final RestfulResponse rsp = new RestfulResponse();
 185         rsp.setStatus(this.getResponseStatus());
 186         if(isGzip()) {
 187             final String responseString = decompressGzipToStr(getResponseContentBytes());
 188             rsp.setResponseJson(responseString);
 189         } else {
 190             rsp.setResponseJson(this.getResponseContent());
 191         }
 192
 193         final HttpFields field = this.getResponseFields();
 194         if(field != null) {
 195             final Map<String, String> header = new HashMap<>();
 196
 197             final Enumeration<String> names = field.getFieldNames();
 198             for(final Enumeration<String> e = names; e.hasMoreElements();) {
 199                 final String fieldName = e.nextElement();
 200                 final String fieldValue = field.getStringField(fieldName);
 201                 header.put(fieldName, fieldValue);
 202             }
 203
 204             rsp.setRespHeaderMap(header);
 205         }
 206         return rsp;
 207     }
 208
 209 }