holmes-actions/src/main/java/org/onap/holmes/common/config/MicroServiceConfig.java

   1 /**\r
   2  * Copyright  2017-2021 ZTE Corporation.\r
   3  * <p>\r
   4  * Licensed under the Apache License, Version 2.0 (the "License");\r
   5  * you may not use this file except in compliance with the License.\r
   6  * You may obtain a copy of the License at\r
   7  * <p>\r
   8  * http://www.apache.org/licenses/LICENSE-2.0\r
   9  * <p>\r
  10  * Unless required by applicable law or agreed to in writing, software\r
  11  * distributed under the License is distributed on an "AS IS" BASIS,\r
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  13  * See the License for the specific language governing permissions and\r
  14  * limitations under the License.\r
  15  */\r
  16 package org.onap.holmes.common.config;\r
  17 \r
  18 import com.google.gson.JsonArray;\r
  19 import com.google.gson.JsonObject;\r
  20 import com.google.gson.JsonParser;\r
  21 import org.apache.commons.lang3.StringUtils;\r
  22 import org.onap.holmes.common.constant.AlarmConst;\r
  23 import org.onap.holmes.common.utils.JerseyClient;\r
  24 import org.slf4j.Logger;\r
  25 import org.slf4j.LoggerFactory;\r
  26 \r
  27 import static org.onap.holmes.common.utils.CommonUtils.getEnv;\r
  28 \r
  29 public class MicroServiceConfig {\r
  30 \r
  31     final static public String CONSUL_ADDR_SUF = ":8500/v1/catalog/service/";\r
  32     final static public String CONSUL_HOST = "CONSUL_HOST";\r
  33     final static public String HOSTNAME = "HOSTNAME";\r
  34     final static public String POD_IP = "POD_IP";\r
  35     final static public String CONFIG_BINDING_SERVICE = "CONFIG_BINDING_SERVICE";\r
  36     final static public String MSB_ADDR = "MSB_ADDR";\r
  37     final static public String MSB_IAG_SERVICE_HOST = "MSB_IAG_SERVICE_HOST";\r
  38     final static public String MSB_IAG_SERVICE_PORT = "MSB_IAG_SERVICE_PORT";\r
  39     final static public String BASE_URL = "BASE_URL";\r
  40     final static public String PRE_ADDR = "PRE_ADDR";\r
  41     final static public String POST_ADDR = "POST_ADDR";\r
  42     final static public String AAI_ADDR = "AAI_BASEADDR";\r
  43     final static public String PROTOCOL_HTTP = "http";\r
  44     final static public String PROTOCOL_HTTPS = "https";\r
  45     final static public int PLAIN_PORT = 80;\r
  46     final static public int TLS_PORT = 443;\r
  47 \r
  48     final static public Logger log = LoggerFactory.getLogger(MicroServiceConfig.class);\r
  49 \r
  50     public static String getConsulAddrInfo() {\r
  51         return "http://" + getEnv(CONSUL_HOST) + CONSUL_ADDR_SUF;\r
  52     }\r
  53 \r
  54     public static String getServiceAddrInfoFromDcaeConsulByHostName(String hostname) {\r
  55         String ret = null;\r
  56         String queryString = getConsulAddrInfo() + hostname;\r
  57         log.info("Query the " + hostname + " address using the URL: " + queryString);\r
  58         try {\r
  59             JsonArray addrArray = JsonParser.parseString(execQuery(queryString)).getAsJsonArray();\r
  60             if (addrArray.size() > 0) {\r
  61                 JsonObject addrJson = addrArray.get(0).getAsJsonObject();\r
  62                 if (addrJson != null && addrJson.get("ServiceAddress") != null\r
  63                         && addrJson.get("ServicePort") != null) {\r
  64                     ret = "http://" + addrJson.get("ServiceAddress").getAsString() + ":" + addrJson\r
  65                             .get("ServicePort").getAsString();\r
  66                 }\r
  67             } else {\r
  68                 log.info("No service info is returned from DCAE Consul. Hostname: {}", hostname);\r
  69             }\r
  70         } catch (Exception e) {\r
  71             log.warn(e.getMessage(), e);\r
  72         }\r
  73         log.info("The " + hostname + " address is " + ret);\r
  74         return ret;\r
  75     }\r
  76 \r
  77     private static String execQuery(String queryString) {\r
  78         return JerseyClient.newInstance().get(queryString);\r
  79     }\r
  80 \r
  81     public static String getServiceConfigInfoFromCBS(String hostname) {\r
  82         String url = getServiceAddrInfoFromDcaeConsulByHostName(getEnv(CONFIG_BINDING_SERVICE)) + "/service_component/" + hostname;\r
  83         String ret = execQuery(url);\r
  84         log.info("The query url is: " + url + ". The corresponding configurations are " + ret);\r
  85         return ret;\r
  86     }\r
  87 \r
  88     public static String getMsbServerAddrWithHttpPrefix() {\r
  89         String[] addrInfo = getMsbIpAndPort();\r
  90         String ret = addrInfo[0] + ":" + addrInfo[1];\r
  91         if (!ret.startsWith(AlarmConst.HTTP) || !ret.startsWith(AlarmConst.HTTPS)) {\r
  92             ret = AlarmConst.HTTP + ret;\r
  93         }\r
  94         return ret;\r
  95     }\r
  96 \r
  97     public static String getAaiAddr() {\r
  98         boolean tlsEnabled = Boolean.valueOf(getEnv("ENABLE_ENCRYPT"));\r
  99 \r
 100         return String.format("%s://%s%s%s.%s:%d",\r
 101                 tlsEnabled ? PROTOCOL_HTTPS : PROTOCOL_HTTP,\r
 102                 nullToEmptyString(getEnv(PRE_ADDR)),\r
 103                 nullToEmptyString(getEnv(AAI_ADDR)),\r
 104                 nullToEmptyString(getEnv(POST_ADDR)),\r
 105                 nullToEmptyString(getEnv(BASE_URL)),\r
 106                 tlsEnabled ? TLS_PORT : PLAIN_PORT);\r
 107     }\r
 108 \r
 109     public static String[] getMsbIpAndPort() {\r
 110         return new String[]{getEnv(MSB_IAG_SERVICE_HOST), getEnv(MSB_IAG_SERVICE_PORT)};\r
 111     }\r
 112 \r
 113     public static String[] getMicroServiceIpAndPort() {\r
 114         String info = getEnv(POD_IP);\r
 115         if (info != null) {\r
 116             return split(info);\r
 117         } else {\r
 118             return split(getEnv(HOSTNAME));\r
 119         }\r
 120     }\r
 121 \r
 122     private static String[] split(String addr) {\r
 123         String ip;\r
 124         String port = "80";\r
 125         if (addr.lastIndexOf(":") == -1) {\r
 126             ip = addr;\r
 127         } else if (addr.lastIndexOf(":") < 5 && addr.indexOf("://") != -1) {\r
 128             ip = addr.substring(addr.indexOf("//") + 2);    //remove the http(s):// prefix\r
 129         } else {\r
 130             ip = addr.substring(addr.indexOf("://") != -1 ? addr.indexOf("//") + 2 : 0, addr.lastIndexOf(":"));\r
 131             port = addr.substring(addr.lastIndexOf(":") + 1);\r
 132         }\r
 133         return new String[]{ip, port};\r
 134     }\r
 135 \r
 136     private static String nullToEmptyString(String input) {\r
 137         return input == null ? StringUtils.EMPTY : input;\r
 138     }\r
 139 \r
 140 }\r