Merge "Add msb register code"

diff --git a/holmes-actions/src/main/java/org/openo/holmes/common/config/MicroServiceConfig.java b/holmes-actions/src/main/java/org/openo/holmes/common/config/MicroServiceConfig.java
new file mode 100644 (file)
index 0000000..ba55562
--- /dev/null
+++ b/holmes-actions/src/main/java/org/openo/holmes/common/config/MicroServiceConfig.java
@@ -0,0 +1,36 @@
+/**\r
+ * Copyright  2017 ZTE Corporation.\r
+ *\r
+ * Licensed under the Apache License, Version 2.0 (the "License");\r
+ * you may not use this file except in compliance with the License.\r
+ * You may obtain a copy of the License at\r
+ *\r
+ *     http://www.apache.org/licenses/LICENSE-2.0\r
+ *\r
+ * Unless required by applicable law or agreed to in writing, software\r
+ * distributed under the License is distributed on an "AS IS" BASIS,\r
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
+ * See the License for the specific language governing permissions and\r
+ * limitations under the License.\r
+ */\r
+package org.openo.holmes.common.config;\r
+\r
+public class MicroServiceConfig {\r
+\r
+  private static String getProperty(String name) {\r
+    String value = System.getenv(name);\r
+    if (value == null) {\r
+      value = System.getProperty(name);\r
+    }\r
+    return value;\r
+  }\r
+\r
+  public static String getMsbServerAddr() {\r
+    return getProperty("MSB_ADDR");\r
+  }\r
+\r
+  public static String getServiceIp() {\r
+    return getProperty("SERVICE_IP");\r
+  }\r
+\r
+}\r

diff --git a/holmes-actions/src/main/java/org/openo/holmes/common/utils/MSBRegisterUtil.java b/holmes-actions/src/main/java/org/openo/holmes/common/utils/MSBRegisterUtil.java
new file mode 100644 (file)
index 0000000..49dff92
--- /dev/null
+++ b/holmes-actions/src/main/java/org/openo/holmes/common/utils/MSBRegisterUtil.java
@@ -0,0 +1,65 @@
+/**\r
+ * Copyright 2017 ZTE Corporation.\r
+ *\r
+ * Licensed under the Apache License, Version 2.0 (the "License");\r
+ * you may not use this file except in compliance with the License.\r
+ * You may obtain a copy of the License at\r
+ *\r
+ *     http://www.apache.org/licenses/LICENSE-2.0\r
+ *\r
+ * Unless required by applicable law or agreed to in writing, software\r
+ * distributed under the License is distributed on an "AS IS" BASIS,\r
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
+ * See the License for the specific language governing permissions and\r
+ * limitations under the License.\r
+ */\r
+\r
+package org.openo.holmes.common.utils;\r
+\r
+import com.fasterxml.jackson.databind.ObjectMapper;\r
+import java.io.IOException;\r
+import lombok.extern.slf4j.Slf4j;\r
+import org.apache.commons.lang3.StringUtils;\r
+import org.apache.http.HttpResponse;\r
+import org.apache.http.client.methods.HttpPost;\r
+import org.apache.http.entity.ByteArrayEntity;\r
+import org.apache.http.impl.client.CloseableHttpClient;\r
+import org.apache.http.impl.client.HttpClients;\r
+import org.jvnet.hk2.annotations.Service;\r
+import org.openo.holmes.common.api.entity.ServiceRegisterEntity;\r
+import org.openo.holmes.common.config.MicroServiceConfig;\r
+import org.openo.holmes.common.constant.AlarmConst;\r
+\r
+@Slf4j\r
+@Service\r
+public class MSBRegisterUtil {\r
+\r
+    public boolean register(ServiceRegisterEntity entity) throws IOException {\r
+        CloseableHttpClient httpClient = HttpClients.createDefault();\r
+        try {\r
+            ObjectMapper mapper = new ObjectMapper();\r
+            String content = mapper.writeValueAsString(entity);\r
+            HttpPost httpPost = new HttpPost("http://" + MicroServiceConfig.getMsbServerAddr()\r
+                    + ":8086/openoapi/microservices/v1/services?createOrUpdate=false");\r
+            if (StringUtils.isNotEmpty(content)) {\r
+                httpPost.setEntity(new ByteArrayEntity(content.getBytes()));\r
+            }\r
+            HttpResponse response;\r
+            try {\r
+                response = httpClient.execute(httpPost);\r
+            } catch (Exception e) {\r
+                log.warn("Registering the service to the bus failure", e);\r
+                return false;\r
+            }\r
+            if (response.getStatusLine().getStatusCode() == AlarmConst.RESPONSE_STATUS_OK) {\r
+                log.info("Registration successful service to the bus :" + response.getEntity());\r
+                return true;\r
+            } else {\r
+                log.warn("Registering the service to the bus failure");\r
+                return false;\r
+            }\r
+        } finally {\r
+            httpClient.close();\r
+        }\r
+    }\r
+}
\ No newline at end of file