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.discovery.host}")
  41     private String DISCOVERY_HOST;
  42
  43     @Value("${msb.discovery.port}")
  44     private int DISCOVERY_PORT;
  45
  46     @Value("${msb.service.host}")
  47     private String SERVICE_HOST;
  48
  49     @Value("${msb.service.port}")
  50     private String SERVICE_PORT;
  51
  52     @Value("${msb.service.name}")
  53     private String SERVICE_NAME;
  54
  55     @Value("${msb.service.version}")
  56     private String SERVICE_VERSION;
  57
  58     @Value("${msb.service.url}")
  59     private String SERVICE_URL;
  60
  61     @Value("${msb.service.custom_path}")
  62     private String SERVICE_CUSTOM_PATH;
  63
  64     @Value("${msb.service.protocol}")
  65     private String SERVICE_PROTOCOL;
  66
  67     @Value("${msb.service.visual_range}")
  68     private String SERVICE_VISUAL_RANGE;
  69
  70     @Value("${msb.service.enable_ssl}")
  71     private boolean SERVICE_ENABLE_SSL;
  72
  73     @Override
  74     public void run(String... strings) throws Exception {
  75         MicroServiceInfo msinfo = new MicroServiceInfo();
  76         msinfo.setServiceName(SERVICE_NAME);
  77         msinfo.setVersion(SERVICE_VERSION);
  78         msinfo.setUrl(SERVICE_URL);
  79         msinfo.setProtocol(SERVICE_PROTOCOL);
  80         msinfo.setVisualRange(SERVICE_VISUAL_RANGE);
  81         msinfo.setEnable_ssl(SERVICE_ENABLE_SSL);
  82
  83         if (!Strings.isNullOrEmpty(SERVICE_CUSTOM_PATH)) {
  84             msinfo.setPath(SERVICE_CUSTOM_PATH);
  85         }
  86
  87         Set<Node> nodes = new HashSet<>();
  88         Node thisNode = new Node();
  89         thisNode.setIp(Strings.isNullOrEmpty(SERVICE_HOST) ? InetAddress.getLocalHost().getHostAddress() : SERVICE_HOST);
  90         thisNode.setPort(SERVICE_PORT);
  91         thisNode.setCheckType("HTTP");
  92         thisNode.setCheckUrl(SERVICE_URL + "/status");
  93         nodes.add(thisNode);
  94         msinfo.setNodes(nodes);
  95
  96         logger.info(
  97                 "Register this service with msb discovery (" + DISCOVERY_HOST + ":" + DISCOVERY_PORT + "):\n"
  98                         + " - host: [" + thisNode.getIp() + "]\n"
  99                         + " - port: [" + thisNode.getPort() + "]\n"
 100                         + " - name: [" + msinfo.getServiceName() + "]\n"
 101                         + " - version: [" + msinfo.getVersion() + "]\n"
 102                         + " - url: [" + msinfo.getUrl() + "]\n"
 103                         + " - path: [" + msinfo.getPath() + "]\n"
 104                         + " - protocol: [" + msinfo.getProtocol() + "]\n"
 105                         + " - visualRange: [" + msinfo.getVisualRange() + "]\n"
 106                         + " - enableSSL: [" + SERVICE_ENABLE_SSL + "]\n"
 107         );
 108
 109         MSBServiceClient msbClient = new MSBServiceClient(DISCOVERY_HOST, DISCOVERY_PORT);
 110         MicroServiceFullInfo microServiceFullInfo = msbClient.registerMicroServiceInfo(msinfo);
 111
 112         logger.debug("microServiceFullInfo = {}", microServiceFullInfo.toString());
 113     }
 114 }