summaryrefslogtreecommitdiffstats
path: root/src/main/java/org/onap/nbi/ServiceRegisterRunner.java
blob: e8cca0083dd56732ccd0d5ff5c6945b494998818 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
/**
 * Copyright (c) 2018 Orange
 * <p>
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * <p>
 * http://www.apache.org/licenses/LICENSE-2.0
 * <p>
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.onap.nbi;

import com.google.common.base.Strings;
import org.onap.msb.sdk.discovery.entity.MicroServiceFullInfo;
import org.onap.msb.sdk.discovery.entity.MicroServiceInfo;
import org.onap.msb.sdk.discovery.entity.Node;
import org.onap.msb.sdk.httpclient.msb.MSBServiceClient;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.CommandLineRunner;
import org.springframework.stereotype.Component;

import java.net.InetAddress;
import java.util.HashSet;
import java.util.Set;

/**
 * Register this NBI instance with MSB discovery when the app is fully started
 */
@Component
public class ServiceRegisterRunner implements CommandLineRunner {
    private static final Logger logger = LoggerFactory.getLogger(ServiceRegisterRunner.class);

    @Value("${msb.discovery.host}")
    private String DISCOVERY_HOST;

    @Value("${msb.discovery.port}")
    private int DISCOVERY_PORT;

    @Value("${msb.service.host}")
    private String SERVICE_HOST;

    @Value("${msb.service.port}")
    private String SERVICE_PORT;

    @Value("${msb.service.name}")
    private String SERVICE_NAME;

    @Value("${msb.service.version}")
    private String SERVICE_VERSION;

    @Value("${msb.service.url}")
    private String SERVICE_URL;

    @Value("${msb.service.custom_path}")
    private String SERVICE_CUSTOM_PATH;

    @Value("${msb.service.protocol}")
    private String SERVICE_PROTOCOL;

    @Value("${msb.service.visual_range}")
    private String SERVICE_VISUAL_RANGE;

    @Value("${msb.service.enable_ssl}")
    private boolean SERVICE_ENABLE_SSL;

    @Override
    public void run(String... strings) throws Exception {
        MicroServiceInfo msinfo = new MicroServiceInfo();
        msinfo.setServiceName(SERVICE_NAME);
        msinfo.setVersion(SERVICE_VERSION);
        msinfo.setUrl(SERVICE_URL);
        msinfo.setProtocol(SERVICE_PROTOCOL);
        msinfo.setVisualRange(SERVICE_VISUAL_RANGE);
        msinfo.setEnable_ssl(SERVICE_ENABLE_SSL);

        if (!Strings.isNullOrEmpty(SERVICE_CUSTOM_PATH)) {
            msinfo.setPath(SERVICE_CUSTOM_PATH);
        }

        Set<Node> nodes = new HashSet<>();
        Node thisNode = new Node();
        thisNode.setIp(Strings.isNullOrEmpty(SERVICE_HOST) ? InetAddress.getLocalHost().getHostAddress() : SERVICE_HOST);
        thisNode.setPort(SERVICE_PORT);
        thisNode.setCheckType("HTTP");
        thisNode.setCheckUrl(SERVICE_URL + "/status");
        nodes.add(thisNode);
        msinfo.setNodes(nodes);

        logger.info(
                "Register this service with msb discovery (" + DISCOVERY_HOST + ":" + DISCOVERY_PORT + "):\n"
                        + " - host: [" + thisNode.getIp() + "]\n"
                        + " - port: [" + thisNode.getPort() + "]\n"
                        + " - name: [" + msinfo.getServiceName() + "]\n"
                        + " - version: [" + msinfo.getVersion() + "]\n"
                        + " - url: [" + msinfo.getUrl() + "]\n"
                        + " - path: [" + msinfo.getPath() + "]\n"
                        + " - protocol: [" + msinfo.getProtocol() + "]\n"
                        + " - visualRange: [" + msinfo.getVisualRange() + "]\n"
                        + " - enableSSL: [" + SERVICE_ENABLE_SSL + "]\n"
        );

        MSBServiceClient msbClient = new MSBServiceClient(DISCOVERY_HOST, DISCOVERY_PORT);
        MicroServiceFullInfo microServiceFullInfo = msbClient.registerMicroServiceInfo(msinfo);

        logger.debug("microServiceFullInfo = {}", microServiceFullInfo.toString());
    }
}