src/main/java/org/onap/clamp/util/HttpConnectionManager.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP CLAMP
   4  * ================================================================================
   5  * Copyright (C) 2019 AT&T Intellectual Property. All rights
   6  *                             reserved.
   7  * ================================================================================
   8  * Licensed under the Apache License, Version 2.0 (the "License");
   9  * you may not use this file except in compliance with the License.
  10  * You may obtain a copy of the License at
  11  *
  12  * http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing, software
  15  * distributed under the License is distributed on an "AS IS" BASIS,
  16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  17  * See the License for the specific language governing permissions and
  18  * limitations under the License.
  19  * ============LICENSE_END============================================
  20  * Modifications copyright (c) 2018 Nokia
  21  * ===================================================================
  22  *
  23  */
  24
  25 package org.onap.clamp.util;
  26
  27 import com.att.eelf.configuration.EELFLogger;
  28 import com.att.eelf.configuration.EELFManager;
  29
  30 import java.io.BufferedReader;
  31 import java.io.DataOutputStream;
  32 import java.io.IOException;
  33 import java.io.InputStreamReader;
  34 import java.net.HttpURLConnection;
  35 import java.net.URL;
  36
  37 import javax.net.ssl.HttpsURLConnection;
  38 import javax.ws.rs.BadRequestException;
  39
  40 import org.apache.commons.io.IOUtils;
  41 import org.onap.clamp.clds.util.LoggingUtils;
  42 import org.springframework.stereotype.Component;
  43
  44 /**
  45  * This class manages the HTTP and HTTPS connections.
  46  */
  47 @Component
  48 public class HttpConnectionManager {
  49     protected static final EELFLogger logger = EELFManager.getInstance().getLogger(HttpConnectionManager.class);
  50     protected static final EELFLogger metricsLogger = EELFManager.getInstance().getMetricsLogger();
  51     private static final String REQUEST_FAILED_LOG = "Request Failed - response payload=";
  52
  53     private String doHttpsQuery(URL url, String requestMethod, String payload, String contentType, String target) throws IOException {
  54         LoggingUtils utils = new LoggingUtils(logger);
  55         logger.info("Using HTTPS URL:" + url.toString());
  56         HttpsURLConnection secureConnection = (HttpsURLConnection) url.openConnection();
  57         secureConnection = utils.invokeHttps(secureConnection, target, requestMethod);
  58         secureConnection.setRequestMethod(requestMethod);
  59         secureConnection.setRequestProperty("X-ECOMP-RequestID", LoggingUtils.getRequestId());
  60         if (payload != null && contentType != null) {
  61             secureConnection.setRequestProperty("Content-Type", contentType);
  62             secureConnection.setDoOutput(true);
  63             try (DataOutputStream wr = new DataOutputStream(secureConnection.getOutputStream())) {
  64                 wr.writeBytes(payload);
  65                 wr.flush();
  66             }
  67         }
  68         int responseCode = secureConnection.getResponseCode();
  69         logger.info("Response Code: " + responseCode);
  70         if (responseCode < 400) {
  71             try (BufferedReader reader = new BufferedReader(new InputStreamReader(secureConnection.getInputStream()))) {
  72                 String responseStr = IOUtils.toString(reader);
  73                 logger.info("Response Content: " + responseStr);
  74                 return responseStr;
  75             }
  76         } else {
  77             // In case of connection failure just check whether there is a
  78             // content or not
  79             try (BufferedReader reader = new BufferedReader(new InputStreamReader(secureConnection.getErrorStream()))) {
  80                 String responseStr = IOUtils.toString(reader);
  81                 logger.error(REQUEST_FAILED_LOG + responseStr);
  82                 throw new BadRequestException(responseStr);
  83             }
  84         }
  85     }
  86
  87     private String doHttpQuery(URL url, String requestMethod, String payload, String contentType, String target) throws IOException {
  88         LoggingUtils utils = new LoggingUtils(logger);
  89         logger.info("Using HTTP URL:" + url);
  90         HttpURLConnection connection = (HttpURLConnection) url.openConnection();
  91         connection = utils.invoke(connection, target, requestMethod);
  92         connection.setRequestMethod(requestMethod);
  93         if (payload != null && contentType != null) {
  94             connection.setRequestProperty("Content-Type", contentType);
  95             connection.setDoOutput(true);
  96             try (DataOutputStream wr = new DataOutputStream(connection.getOutputStream())) {
  97                 wr.writeBytes(payload);
  98                 wr.flush();
  99             }
 100         }
 101         int responseCode = connection.getResponseCode();
 102         logger.info("Response Code: " + responseCode);
 103         if (responseCode < 400) {
 104             try (BufferedReader reader = new BufferedReader(new InputStreamReader(connection.getInputStream()))) {
 105                 String responseStr = IOUtils.toString(reader);
 106                 logger.info("Response Content: " + responseStr);
 107                 utils.invokeReturn();
 108                 return responseStr;
 109             }
 110         } else {
 111             // In case of connection failure just check whether there is a
 112             // content or not
 113             try (BufferedReader reader = new BufferedReader(new InputStreamReader(connection.getErrorStream()))) {
 114                 String responseStr = IOUtils.toString(reader);
 115                 logger.error(REQUEST_FAILED_LOG + responseStr);
 116                 utils.invokeReturn();
 117                 throw new BadRequestException(responseStr);
 118             }
 119         }
 120     }
 121
 122     /**
 123      * This method does a HTTP/HTTPS query with parameters specified.
 124      *
 125      * @param url
 126      *        The string HTTP or HTTPS that mustr be used to connect
 127      * @param requestMethod
 128      *        The Request Method (PUT, POST, GET, DELETE, etc ...)
 129      * @param payload
 130      *        The payload if any, in that case an ouputstream is opened
 131      * @param contentType
 132      *        The "application/json or application/xml, or whatever"
 133      * @return The payload of the answer
 134      * @throws IOException
 135      *         In case of issue with the streams
 136      */
 137     public String doGeneralHttpQuery(String url, String requestMethod, String payload, String contentType, String target)
 138         throws IOException {
 139         URL urlObj = new URL(url);
 140         if (url.contains("https://")) { // Support for HTTPS
 141             return doHttpsQuery(urlObj, requestMethod, payload, contentType, target);
 142         } else { // Support for HTTP
 143             return doHttpQuery(urlObj, requestMethod, payload, contentType, target);
 144         }
 145     }
 146 }