X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=holmes-actions%2Fsrc%2Fmain%2Fjava%2Forg%2Fopeno%2Fholmes%2Fcommon%2Futils%2FMSBRegisterUtil.java;h=176107ec9b66f7409a942043214487744bc16c96;hb=4b68e64dbeeeaa365eb1698987e7d3cbfaef1e60;hp=826fbd037efce67734098b2078713e7a556264ba;hpb=b67a0ef5ee0ef5a6d84529b6c52d5072807cc36e;p=holmes%2Fcommon.git

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
index 826fbd0..176107e 100644
--- 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
@@ -16,60 +16,59 @@
 
 package org.openo.holmes.common.utils;
 
-import com.fasterxml.jackson.databind.ObjectMapper;
+import com.eclipsesource.jaxrs.consumer.ConsumerFactory;
 import java.io.IOException;
 import lombok.extern.slf4j.Slf4j;
-import org.apache.commons.lang3.StringUtils;
-import org.apache.http.HttpResponse;
-import org.apache.http.client.methods.HttpPost;
 import org.apache.http.client.methods.HttpRequestBase;
-import org.apache.http.entity.ByteArrayEntity;
-import org.apache.http.impl.client.CloseableHttpClient;
-import org.apache.http.impl.client.HttpClients;
+import org.glassfish.jersey.client.ClientConfig;
 import org.jvnet.hk2.annotations.Service;
 import org.openo.holmes.common.api.entity.ServiceRegisterEntity;
 import org.openo.holmes.common.config.MicroServiceConfig;
-import org.openo.holmes.common.constant.AlarmConst;
+import org.openo.holmes.common.msb.MicroserviceBusRest;
 
 @Slf4j
 @Service
 public class MSBRegisterUtil {
 
-    public boolean register(ServiceRegisterEntity entity) throws IOException {
-        CloseableHttpClient httpClient = HttpClients.createDefault();
-        try {
-            ObjectMapper mapper = new ObjectMapper();
-            String content = mapper.writeValueAsString(entity);
-            HttpPost httpPost = new HttpPost("http://" + MicroServiceConfig.getMsbServerAddr()
-                    + ":8086/api/microservices/v1/services?createOrUpdate=false");
-            if (StringUtils.isNotEmpty(content)) {
-                httpPost.setEntity(new ByteArrayEntity(content.getBytes()));
-            }
-            this.setHeader(httpPost);
-            HttpResponse response;
-            try {
-                response = httpClient.execute(httpPost);
-            } catch (Exception e) {
-                log.warn("Registering the service to the bus failure", e);
-                return false;
-            }
-            if (response.getStatusLine().getStatusCode() == AlarmConst.MICRO_SERVICE_STATUS_SUCCESS) {
-                log.info("Registration successful service to the bus :" + response.getEntity());
-                return true;
+    public void register(ServiceRegisterEntity entity) throws IOException {
+        log.info("start holmes micro service register");
+        boolean flag = false;
+        int retry = 0;
+        while (!flag && retry < 20) {
+            log.info("Holmes microservice register. retry:" + retry);
+            retry++;
+            flag = inner_register(entity);
+            if (!flag) {
+                log.warn("micro service register failed, sleep 30S and try again.");
+                threadSleep(30000);
             } else {
-                log.warn(
-                        "Registering the service to the bus failure:" + response.getStatusLine().getStatusCode() + " " +
-                                response.getStatusLine().getReasonPhrase());
-                return false;
+                log.info("micro service register success!");
+                break;
             }
-        } finally {
-            httpClient.close();
         }
+        log.info("holmes micro service register end.");
     }
 
-    private void setHeader(HttpRequestBase httpRequestBase) {
-        httpRequestBase.setHeader("Content-Type", "text/html;charset=UTF-8");
-        httpRequestBase.setHeader("Accept", "application/json");
-        httpRequestBase.setHeader("Content-Type", "application/json");
+    private boolean inner_register(ServiceRegisterEntity entity) {
+        ClientConfig config = new ClientConfig();
+        try {
+            MicroserviceBusRest resourceserviceproxy = ConsumerFactory.createConsumer(
+                    MicroServiceConfig.getMsbServerAddr(), config, MicroserviceBusRest.class);
+            resourceserviceproxy.registerServce("false", entity);
+        } catch (Exception error) {
+            log.error("microservice register failed!" + error.getMessage());
+            return false;
+        }
+        return true;
+    }
+
+    private void threadSleep(int second) {
+        log.info("start sleep ....");
+        try {
+            Thread.sleep(second);
+        } catch (InterruptedException error) {
+            log.error("thread sleep error.errorMsg:" + error.getMessage());
+        }
+        log.info("sleep end .");
     }
 }
\ No newline at end of file