src/main/java/org/onap/nbi/ServiceRegisterRunner.java

   1 /**
   2  * Copyright (c) 2018 Orange
   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 package org.onap.nbi;
  17
  18 import com.google.common.base.Strings;
  19 import org.onap.msb.sdk.discovery.entity.MicroServiceFullInfo;
  20 import org.onap.msb.sdk.discovery.entity.MicroServiceInfo;
  21 import org.onap.msb.sdk.discovery.entity.Node;
  22 import org.onap.msb.sdk.httpclient.msb.MSBServiceClient;
  23 import org.slf4j.Logger;
  24 import org.slf4j.LoggerFactory;
  25 import org.springframework.beans.factory.annotation.Value;
  26 import org.springframework.boot.CommandLineRunner;
  27 import org.springframework.stereotype.Component;
  28
  29 import java.net.InetAddress;
  30 import java.util.HashSet;
  31 import java.util.Set;
  32
  33 /**
  34  * Register this NBI instance with MSB discovery when the app is fully started
  35  */
  36 @Component
  37 public class ServiceRegisterRunner implements CommandLineRunner {
  38     private static final Logger logger = LoggerFactory.getLogger(ServiceRegisterRunner.class);
  39
  40     @Value("${msb.enabled}")
  41     private boolean IS_ENABLED;
  42
  43     @Value("${msb.discovery.host}")
  44     private String DISCOVERY_HOST;
  45
  46     @Value("${msb.discovery.port}")
  47     private int DISCOVERY_PORT;
  48
  49     @Value("${msb.discovery.retry}")
  50     private int RETRY;
  51
  52     @Value("${msb.discovery.retry_interval}")
  53     private int RETRY_INTERVAL;
  54
  55     @Value("${msb.service.host}")
  56     private String SERVICE_HOST;
  57
  58     @Value("${server.port}")
  59     private String SERVICE_PORT;
  60
  61     @Value("${msb.service.name}")
  62     private String SERVICE_NAME;
  63
  64     @Value("${nbi.version}")
  65     private String SERVICE_VERSION;
  66
  67     @Value("${server.contextPath}")
  68     private String SERVICE_URL;
  69
  70     @Value("${msb.service.custom_path}")
  71     private String SERVICE_CUSTOM_PATH;
  72
  73     @Value("${msb.service.protocol}")
  74     private String SERVICE_PROTOCOL;
  75
  76     @Value("${msb.service.visual_range}")
  77     private String SERVICE_VISUAL_RANGE;
  78
  79     @Value("${msb.service.enable_ssl}")
  80     private boolean SERVICE_ENABLE_SSL;
  81
  82     @Override
  83     public void run(String... strings) throws Exception {
  84         if (!IS_ENABLED) {
  85             logger.info("Registration with msb discovery is not enabled");
  86             return;
  87         }
  88
  89         MicroServiceInfo msinfo = new MicroServiceInfo();
  90         msinfo.setServiceName(SERVICE_NAME);
  91         msinfo.setVersion(SERVICE_VERSION);
  92         msinfo.setUrl(SERVICE_URL);
  93         msinfo.setProtocol(SERVICE_PROTOCOL);
  94         msinfo.setVisualRange(SERVICE_VISUAL_RANGE);
  95         msinfo.setEnable_ssl(SERVICE_ENABLE_SSL);
  96
  97         if (!Strings.isNullOrEmpty(SERVICE_CUSTOM_PATH)) {
  98             msinfo.setPath(SERVICE_CUSTOM_PATH);
  99         }
 100
 101         Set<Node> nodes = new HashSet<>();
 102         Node thisNode = new Node();
 103         thisNode.setIp(Strings.isNullOrEmpty(SERVICE_HOST) ? InetAddress.getLocalHost().getHostAddress() : SERVICE_HOST);
 104         thisNode.setPort(SERVICE_PORT);
 105         thisNode.setCheckType("HTTP");
 106         thisNode.setCheckUrl(SERVICE_URL + "/status");
 107         nodes.add(thisNode);
 108         msinfo.setNodes(nodes);
 109
 110         logger.info(
 111                 "Registering with msb discovery (" + DISCOVERY_HOST + ":" + DISCOVERY_PORT + "):\n"
 112                         + " - host: [" + thisNode.getIp() + "]\n"
 113                         + " - port: [" + thisNode.getPort() + "]\n"
 114                         + " - name: [" + msinfo.getServiceName() + "]\n"
 115                         + " - version: [" + msinfo.getVersion() + "]\n"
 116                         + " - url: [" + msinfo.getUrl() + "]\n"
 117                         + " - path: [" + msinfo.getPath() + "]\n"
 118                         + " - protocol: [" + msinfo.getProtocol() + "]g\n"
 119                         + " - visualRange: [" + msinfo.getVisualRange() + "]\n"
 120                         + " - enableSSL: [" + SERVICE_ENABLE_SSL + "]\n"
 121         );
 122
 123         int attempt = 0;
 124         while (attempt<RETRY) {
 125             attempt += 1;
 126             try {
 127                 logger.info("Registration with msb discovery (attempt {}/{})", attempt, RETRY);
 128                 MSBServiceClient msbClient = new MSBServiceClient(DISCOVERY_HOST, DISCOVERY_PORT);
 129                 MicroServiceFullInfo microServiceFullInfo = msbClient.registerMicroServiceInfo(msinfo);
 130                 logger.debug("Registration with msb discovery done, microServiceFullInfo = {}", microServiceFullInfo.toString());
 131                 break;
 132             } catch (Exception ex) {
 133                 logger.info("Registration with msb discovery (attempt {}/{}) FAILED. Sleep {}ms", attempt, RETRY, RETRY_INTERVAL);
 134             }
 135             Thread.sleep(RETRY_INTERVAL);
 136         }
 137     }
 138 }