rulemgt/src/main/java/org/onap/holmes/rulemgt/RuleActiveApp.java

   1 /**
   2  * Copyright 2017-2018 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.rulemgt;
  18
  19 import io.dropwizard.setup.Environment;
  20 import org.onap.holmes.common.config.MicroServiceConfig;
  21 import org.onap.holmes.common.dropwizard.ioc.bundle.IOCApplication;
  22 import org.onap.holmes.common.exception.CorrelationException;
  23 import org.onap.holmes.common.utils.HttpsUtils;
  24 import org.onap.holmes.common.utils.MSBRegisterUtil;
  25 import org.onap.holmes.common.utils.transactionid.TransactionIdFilter;
  26 import org.onap.holmes.rulemgt.dcae.DcaeConfigurationPolling;
  27 import org.onap.holmes.rulemgt.msb.MsbQuery;
  28 import org.onap.holmes.rulemgt.resources.RuleMgtResources;
  29 import org.onap.msb.sdk.discovery.entity.MicroServiceInfo;
  30 import org.onap.msb.sdk.discovery.entity.Node;
  31 import org.slf4j.Logger;
  32 import org.slf4j.LoggerFactory;
  33
  34 import javax.servlet.DispatcherType;
  35 import java.util.EnumSet;
  36 import java.util.HashSet;
  37 import java.util.Set;
  38 import java.util.concurrent.Executors;
  39 import java.util.concurrent.ScheduledExecutorService;
  40 import java.util.concurrent.TimeUnit;
  41
  42 public class RuleActiveApp extends IOCApplication<RuleAppConfig> {
  43
  44     private Logger log = LoggerFactory.getLogger(RuleActiveApp.class);
  45
  46     public static void main(String[] args) throws Exception {
  47         new RuleActiveApp().run(args);
  48     }
  49
  50     @Override
  51     public void run(RuleAppConfig configuration, Environment environment) throws Exception {
  52         super.run(configuration, environment);
  53
  54         try {
  55             new MSBRegisterUtil().register2Msb(createMicroServiceInfo());
  56         } catch (CorrelationException e) {
  57             log.warn(e.getMessage(), e);
  58         }
  59
  60         if (!"1".equals(System.getenv("TESTING"))) {
  61             ScheduledExecutorService service = Executors.newSingleThreadScheduledExecutor();
  62             service.scheduleAtFixedRate(
  63                     new DcaeConfigurationPolling(MicroServiceConfig.getEnv(MicroServiceConfig.HOSTNAME)), 0,
  64                     DcaeConfigurationPolling.POLLING_PERIOD, TimeUnit.MILLISECONDS);
  65         }
  66
  67         environment.servlets().addFilter("customFilter", new TransactionIdFilter()).addMappingForUrlPatterns(EnumSet
  68                 .allOf(DispatcherType.class), true, "/*");
  69
  70         new MsbQuery().startTimer();
  71     }
  72
  73     private MicroServiceInfo createMicroServiceInfo() {
  74         String msbAddrTemplate = (HttpsUtils.isHttpsEnabled() ? "https" : "http")
  75                 + "://%s:%s/api/holmes-rule-mgmt/v1/healthcheck";
  76         String[] serviceAddrInfo = MicroServiceConfig.getMicroServiceIpAndPort();
  77         MicroServiceInfo msinfo = new MicroServiceInfo();
  78         msinfo.setServiceName("holmes-rule-mgmt");
  79         msinfo.setVersion("v1");
  80         msinfo.setUrl("/api/holmes-rule-mgmt/v1");
  81         msinfo.setPath("/api/holmes-rule-mgmt/v1");
  82         msinfo.setProtocol("REST");
  83         msinfo.setVisualRange("0|1");
  84         msinfo.setLb_policy("round-robin");
  85         msinfo.setEnable_ssl(HttpsUtils.isHttpsEnabled());
  86         Set<Node> nodes = new HashSet<>();
  87         Node node = new Node();
  88         node.setIp(serviceAddrInfo[0]);
  89         node.setPort("9101");
  90         /* Following codes will cause an unregistration from MSB (due to MSB malfunction), comment them for now
  91         node.setCheckType("HTTP");
  92         node.setCheckUrl(String.format(msbAddrTemplate, serviceAddrInfo[0], "9101"));
  93         node.setCheckTimeOut("60s");
  94         node.setCheckInterval("60s");
  95         */
  96         nodes.add(node);
  97         msinfo.setNodes(nodes);
  98         return msinfo;
  99     }
 100 }
 101