holmes-actions/src/main/java/org/onap/holmes/common/utils/MsbRegister.java

   1 /**
   2  * Copyright 2017-2020 ZTE Corporation.
   3  * <p>
   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  * <p>
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  * <p>
  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.holmes.common.utils;
  18
  19 import org.apache.commons.lang3.StringUtils;
  20 import org.jvnet.hk2.annotations.Service;
  21 import org.onap.holmes.common.config.MicroServiceConfig;
  22 import org.onap.holmes.common.exception.CorrelationException;
  23 import org.onap.msb.sdk.discovery.entity.MicroServiceFullInfo;
  24 import org.onap.msb.sdk.discovery.entity.MicroServiceInfo;
  25 import org.slf4j.Logger;
  26 import org.slf4j.LoggerFactory;
  27
  28 import javax.ws.rs.client.Entity;
  29 import javax.ws.rs.core.MediaType;
  30 import java.util.concurrent.TimeUnit;
  31
  32 import static org.onap.holmes.common.utils.JerseyClient.PROTOCOL_HTTP;
  33 import static org.onap.holmes.common.utils.JerseyClient.PROTOCOL_HTTPS;
  34
  35 @Service
  36 public class MsbRegister {
  37     private static final Logger log = LoggerFactory.getLogger(MsbRegister.class);
  38
  39     private JerseyClient client = JerseyClient.newInstance();
  40
  41     public MsbRegister() {
  42     }
  43
  44     public void register2Msb(MicroServiceInfo msinfo) throws CorrelationException {
  45         String[] msbAddrInfo = MicroServiceConfig.getMsbIpAndPort();
  46
  47         boolean isHttpsEnabled = StringUtils.isNotBlank(msbAddrInfo[1])
  48                 && msbAddrInfo[1].equals("443");
  49
  50         log.info("Start to register Holmes Service to MSB...");
  51
  52         MicroServiceFullInfo microServiceFullInfo = null;
  53         int retry = 0;
  54         int interval = 5;
  55         while (null == microServiceFullInfo && retry < 20) {
  56             try {
  57                 log.info("Holmes Service Registration. Retry: " + retry++);
  58
  59                 microServiceFullInfo = client
  60                         .header("Accept", MediaType.APPLICATION_JSON)
  61                         .queryParam("createOrUpdate", true)
  62                         .post(String.format("%s://%s:%s/api/microservices/v1/services",
  63                                 isHttpsEnabled ? PROTOCOL_HTTPS : PROTOCOL_HTTP, msbAddrInfo[0], msbAddrInfo[1]),
  64                                 Entity.entity(msinfo, MediaType.APPLICATION_JSON),
  65                                 MicroServiceFullInfo.class);
  66
  67                 if (null == microServiceFullInfo) {
  68                     log.warn(String.format("Failed to register the service to MSB. Sleep %ds and try again.", interval));
  69                     threadSleep(TimeUnit.SECONDS.toSeconds(interval));
  70                     interval += 5;
  71                 } else {
  72                     log.info("Registration succeeded!");
  73                     break;
  74                 }
  75             } catch (Exception e) {
  76                 log.warn("Unexpected exception: " + e.getMessage(), e);
  77             }
  78         }
  79
  80         if (null == microServiceFullInfo) {
  81             throw new CorrelationException("Failed to register the service to MSB!");
  82         }
  83
  84         log.info("Service registration completed.");
  85     }
  86
  87     private void threadSleep(long second) {
  88         log.info("Start sleeping...");
  89         try {
  90             TimeUnit.SECONDS.sleep(second);
  91         } catch (InterruptedException error) {
  92             log.error("thread sleep error message:" + error.getMessage(), error);
  93             Thread.currentThread().interrupt();
  94         }
  95         log.info("Wake up.");
  96     }
  97 }