engine-d/src/main/java/org/onap/holmes/engine/EngineDActiveApp.java

   1 /**
   2  * Copyright 2017 ZTE Corporation.
   3  *
   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  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  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.holmes.engine;
  17
  18 import io.dropwizard.setup.Environment;
  19 import java.util.EnumSet;
  20 import java.util.HashSet;
  21 import java.util.Set;
  22 import java.util.concurrent.Executors;
  23 import java.util.concurrent.ScheduledExecutorService;
  24 import java.util.concurrent.TimeUnit;
  25 import javax.servlet.DispatcherType;
  26 import lombok.extern.slf4j.Slf4j;
  27 import org.onap.holmes.common.config.MicroServiceConfig;
  28 import org.onap.holmes.common.dropwizard.ioc.bundle.IOCApplication;
  29 import org.onap.holmes.common.exception.CorrelationException;
  30 import org.onap.holmes.common.utils.MSBRegisterUtil;
  31 import org.onap.holmes.common.utils.transactionid.TransactionIdFilter;
  32 import org.onap.holmes.engine.dcae.DcaeConfigurationPolling;
  33 import org.onap.holmes.engine.resources.EngineResources;
  34 import org.onap.msb.sdk.discovery.entity.MicroServiceInfo;
  35 import org.onap.msb.sdk.discovery.entity.Node;
  36
  37 @Slf4j
  38 public class EngineDActiveApp extends IOCApplication<EngineDAppConfig> {
  39
  40     public static void main(String[] args) throws Exception {
  41         new EngineDActiveApp().run(args);
  42     }
  43
  44     public void run(EngineDAppConfig configuration, Environment environment) throws Exception {
  45         super.run(configuration, environment);
  46
  47         environment.jersey().register(new EngineResources());
  48         try {
  49             new MSBRegisterUtil().register2Msb(createMicroServiceInfo());
  50         } catch (CorrelationException e) {
  51             log.warn(e.getMessage(), e);
  52         }
  53
  54         ScheduledExecutorService service = Executors.newSingleThreadScheduledExecutor();
  55         service.scheduleAtFixedRate(
  56                 new DcaeConfigurationPolling(MicroServiceConfig.getEnv(MicroServiceConfig.HOSTNAME)), 0,
  57                 DcaeConfigurationPolling.POLLING_PERIOD, TimeUnit.MILLISECONDS);
  58         environment.servlets().addFilter("logFilter",new TransactionIdFilter()).addMappingForUrlPatterns(EnumSet
  59                 .allOf(DispatcherType.class),true,"/*");
  60     }
  61
  62     private MicroServiceInfo createMicroServiceInfo() {
  63         String[] serviceAddrInfo = MicroServiceConfig.getMicroServiceIpAndPort();
  64         MicroServiceInfo msinfo = new MicroServiceInfo();
  65         msinfo.setServiceName("holmes-engine-mgmt");
  66         msinfo.setVersion("v1");
  67         msinfo.setUrl("/api/holmes-engine-mgmt/v1");
  68         msinfo.setProtocol("REST");
  69         msinfo.setVisualRange("0|1");
  70         msinfo.setEnable_ssl(true);
  71         Set<Node> nodes = new HashSet<>();
  72         Node node = new Node();
  73         node.setIp(serviceAddrInfo[0]);
  74         node.setPort(serviceAddrInfo[1]);
  75         nodes.add(node);
  76         msinfo.setNodes(nodes);
  77         return msinfo;
  78     }
  79 }