From 82d2318187ea739a0b14442cbd82a3a7f6a419a7 Mon Sep 17 00:00:00 2001 From: GuangrongFu Date: Thu, 29 Mar 2018 17:20:07 +0800 Subject: Fixed the Start-up Logic Change-Id: I5646e771e05f5f774162373a2f69a0815acb53e5 Issue-ID: HOLMES-106 Signed-off-by: GuangrongFu --- .../org/onap/holmes/rulemgt/RuleActiveApp.java | 3 + .../java/org/onap/holmes/rulemgt/msb/MsbQuery.java | 54 +++++---- .../onap/holmes/rulemgt/send/Ip4AddingRule.java | 2 +- .../onap/holmes/rulemgt/send/RuleAllocation.java | 135 +++++++++++---------- 4 files changed, 106 insertions(+), 88 deletions(-) (limited to 'rulemgt/src/main/java/org/onap') diff --git a/rulemgt/src/main/java/org/onap/holmes/rulemgt/RuleActiveApp.java b/rulemgt/src/main/java/org/onap/holmes/rulemgt/RuleActiveApp.java index 5f96912..daebff3 100644 --- a/rulemgt/src/main/java/org/onap/holmes/rulemgt/RuleActiveApp.java +++ b/rulemgt/src/main/java/org/onap/holmes/rulemgt/RuleActiveApp.java @@ -31,6 +31,7 @@ import org.onap.holmes.common.exception.CorrelationException; import org.onap.holmes.common.utils.MSBRegisterUtil; import org.onap.holmes.common.utils.transactionid.TransactionIdFilter; import org.onap.holmes.rulemgt.dcae.DcaeConfigurationPolling; +import org.onap.holmes.rulemgt.msb.MsbQuery; import org.onap.holmes.rulemgt.resources.RuleMgtResources; import org.onap.msb.sdk.discovery.entity.MicroServiceInfo; import org.onap.msb.sdk.discovery.entity.Node; @@ -64,6 +65,8 @@ public class RuleActiveApp extends IOCApplication { DcaeConfigurationPolling.POLLING_PERIOD, TimeUnit.MILLISECONDS); environment.servlets().addFilter("customFilter",new TransactionIdFilter()).addMappingForUrlPatterns(EnumSet .allOf(DispatcherType.class),true,"/*"); + + new MsbQuery().startTimer(); } private MicroServiceInfo createMicroServiceInfo() { diff --git a/rulemgt/src/main/java/org/onap/holmes/rulemgt/msb/MsbQuery.java b/rulemgt/src/main/java/org/onap/holmes/rulemgt/msb/MsbQuery.java index ed9b9af..2f440b6 100644 --- a/rulemgt/src/main/java/org/onap/holmes/rulemgt/msb/MsbQuery.java +++ b/rulemgt/src/main/java/org/onap/holmes/rulemgt/msb/MsbQuery.java @@ -1,12 +1,12 @@ /** * Copyright 2017 ZTE Corporation. - * + *

* 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 - * - * http://www.apache.org/licenses/LICENSE-2.0 - * + *

+ * http://www.apache.org/licenses/LICENSE-2.0 + *

* 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. @@ -16,63 +16,71 @@ package org.onap.holmes.rulemgt.msb; import lombok.extern.slf4j.Slf4j; -import org.jvnet.hk2.annotations.Service; +import org.glassfish.hk2.api.ServiceLocator; +import org.onap.holmes.common.dropwizard.ioc.utils.ServiceLocatorHolder; import org.onap.holmes.rulemgt.send.RuleAllocation; import org.onap.holmes.rulemgt.send.Ip4AddingRule; import org.onap.holmes.rulemgt.wrapper.RuleMgtWrapper; -import javax.annotation.PostConstruct; -import javax.inject.Inject; import java.util.List; import java.util.Timer; import java.util.TimerTask; -@Service @Slf4j public class MsbQuery { - @Inject private RuleAllocation ruleAllocation; - @Inject private Ip4AddingRule ip4AddingRule; - @Inject private EngineIpList engineIpList; - @Inject private RuleMgtWrapper ruleMgtWrapper; - private List timerIpList; + private List timerIpList; - @PostConstruct - public void init() { + public MsbQuery() { + ruleAllocation = new RuleAllocation(); - try{ + ServiceLocator locator = ServiceLocatorHolder.getLocator(); + ip4AddingRule = locator.getService(Ip4AddingRule.class); + engineIpList = locator.getService(EngineIpList.class); + ruleMgtWrapper = locator.getService(RuleMgtWrapper.class); + } + + public void startTimer() { + try { timer(); - }catch(Exception e){ - log.error("MSBQuery init timer task failed !" + e.getMessage()); + } catch (Exception e) { + log.error("MSBQuery startTimer timer task failed !" + e.getMessage(), e); + try { + Thread.sleep(30000); + } catch (InterruptedException e1) { + Thread.currentThread().interrupt(); + } } + } - public void timer() throws Exception{ + public void timer() throws Exception { Timer timer = new Timer(); timer.schedule(new TimerTask() { public void run() { try { timerIpList = engineIpList.getServiceCount(); - ip4AddingRule.getIpList(timerIpList); - ruleAllocation.judgeAndAllocateRule(timerIpList); + log.info(String.format("There are %d engine instance(s) running currently.", timerIpList.size())); + ip4AddingRule.setIpList(timerIpList); + ruleAllocation.judgeAndAllocateRule(timerIpList); } catch (Exception e) { - log.error("The timing query engine instance failed " ,e); + log.error("The timing query engine instance failed ", e); } } - }, 5000, 30000); + }, 10000, 30000); } diff --git a/rulemgt/src/main/java/org/onap/holmes/rulemgt/send/Ip4AddingRule.java b/rulemgt/src/main/java/org/onap/holmes/rulemgt/send/Ip4AddingRule.java index f82d3a4..e224bb5 100644 --- a/rulemgt/src/main/java/org/onap/holmes/rulemgt/send/Ip4AddingRule.java +++ b/rulemgt/src/main/java/org/onap/holmes/rulemgt/send/Ip4AddingRule.java @@ -32,7 +32,7 @@ public class Ip4AddingRule { private RuleQueryWrapper ruleQueryWrapper; private List engineService; - public void getIpList(List ipList){ + public void setIpList(List ipList){ engineService = ipList; } diff --git a/rulemgt/src/main/java/org/onap/holmes/rulemgt/send/RuleAllocation.java b/rulemgt/src/main/java/org/onap/holmes/rulemgt/send/RuleAllocation.java index e69be51..75f0a08 100644 --- a/rulemgt/src/main/java/org/onap/holmes/rulemgt/send/RuleAllocation.java +++ b/rulemgt/src/main/java/org/onap/holmes/rulemgt/send/RuleAllocation.java @@ -1,12 +1,12 @@ /** * Copyright 2017 ZTE Corporation. - * + *

* 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 - * - * http://www.apache.org/licenses/LICENSE-2.0 - * + *

+ * http://www.apache.org/licenses/LICENSE-2.0 + *

* 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. @@ -17,8 +17,10 @@ package org.onap.holmes.rulemgt.send; import lombok.extern.slf4j.Slf4j; +import org.glassfish.hk2.api.ServiceLocator; import org.jvnet.hk2.annotations.Service; import org.onap.holmes.common.api.entity.CorrelationRule; +import org.onap.holmes.common.dropwizard.ioc.utils.ServiceLocatorHolder; import org.onap.holmes.common.exception.CorrelationException; import org.onap.holmes.common.utils.DbDaoUtil; import org.onap.holmes.rulemgt.bolt.enginebolt.EngineWrapper; @@ -32,56 +34,62 @@ import javax.inject.Inject; import java.util.*; -@Service @Slf4j public class RuleAllocation { private final static int ENABLE = 1; - - @Inject private RuleMgtWrapper ruleMgtWrapper; - @Inject private RuleQueryWrapper ruleQueryWrapper; - @Inject private EngineWrapper engineWrapper; - @Inject private EngineIpList engineIpList; - @Inject private DbDaoUtil daoUtil; - private CorrelationRuleDao correlationRuleDao; - private int ruleCount; private int serviceCount; private List temIpList = new ArrayList<>(); private List engineService = new ArrayList<>(); private List allRules = new ArrayList<>(); - @PostConstruct - public void initDaoUtilAndEngineIp() throws Exception{ + public RuleAllocation() { + ServiceLocator locator = ServiceLocatorHolder.getLocator(); + ruleMgtWrapper = locator.getService(RuleMgtWrapper.class); + ruleQueryWrapper = locator.getService(RuleQueryWrapper.class); + engineWrapper = locator.getService(EngineWrapper.class); + engineIpList = locator.getService(EngineIpList.class); + daoUtil = locator.getService(DbDaoUtil.class); + + initDaoUtilAndEngineIp(); + } + + private void initDaoUtilAndEngineIp() { correlationRuleDao = daoUtil.getJdbiDaoByOnDemand(CorrelationRuleDao.class); - temIpList = engineIpList.getServiceCount(); + try { + temIpList = engineIpList.getServiceCount(); + + } catch (Exception e) { + log.warn("Failed to get the number of engine instances.", e); + } } - public void judgeAndAllocateRule(List ipList)throws Exception{ - if(ipList != null) { + public void judgeAndAllocateRule(List ipList) throws Exception { + if (ipList != null) { engineService = ipList; serviceCount = ipList.size(); } - if(temIpList.size() < serviceCount){ + if (temIpList.size() < serviceCount) { //extend List deleteRule = calculateRule(temIpList); - List allocateRule = calculateRule(temIpList); - List extendIp = extendCompareIp(engineService,temIpList); - AllocateService(extendIp,allocateRule); - deleteRuleFromFormerEngine(deleteRule,temIpList); + List allocateRule = calculateRule(temIpList); + List extendIp = extendCompareIp(engineService, temIpList); + AllocateService(extendIp, allocateRule); + deleteRuleFromFormerEngine(deleteRule, temIpList); } else if (temIpList.size() > serviceCount) { //destroy List destroyIp = destroyCompareIp(engineService, temIpList); AllocateService(restIp(destroyIp), relocateRuleAfterDestroy(destroyIp)); - } else if(temIpList.size() == serviceCount) { + } else if (temIpList.size() == serviceCount) { temIpList = engineService; return; } @@ -91,18 +99,18 @@ public class RuleAllocation { // When the engine is expanding, the rules that need to be allocated are calculated. - private List calculateRule(List oldIpList) throws Exception{ + private List calculateRule(List oldIpList) throws Exception { allRules = ruleQueryWrapper.queryRuleByEnable(ENABLE); - if(allRules != null) { + if (allRules != null) { ruleCount = allRules.size(); } int count = ruleCount / serviceCount; int remainder = ruleCount % serviceCount; List subRule = new ArrayList<>(); - for(String ip : oldIpList) { + for (String ip : oldIpList) { List rules = ruleQueryWrapper.queryRuleByEngineInstance(ip); - List tem = rules.subList(count + (remainder-- / oldIpList.size()),rules.size()); + List tem = rules.subList(count + (remainder-- / oldIpList.size()), rules.size()); subRule.addAll(tem); } return subRule; @@ -111,24 +119,24 @@ public class RuleAllocation { //Rules that need to be allocated after the engine is destroyed private List relocateRuleAfterDestroy(List destroyIpList) throws CorrelationException { List rules = new ArrayList<>(); - try{ - if(destroyIpList != null){ - for(String ip : destroyIpList) { + try { + if (destroyIpList != null) { + for (String ip : destroyIpList) { rules.addAll(ruleQueryWrapper.queryRuleByEngineInstance(ip)); } } - }catch(CorrelationException e) { - log.error("method relocateRuleAfterDestroy get data from DB failed !" +e.getMessage()); + } catch (CorrelationException e) { + log.error("method relocateRuleAfterDestroy get data from DB failed !", e); } return rules; } //Extended IP - private List extendCompareIp(List newList, List oldList){ + private List extendCompareIp(List newList, List oldList) { List extendIpList = new ArrayList<>(); - for( String ip :newList) { - if(! oldList.contains(ip)) { + for (String ip : newList) { + if (!oldList.contains(ip)) { extendIpList.add(ip); } } @@ -138,8 +146,8 @@ public class RuleAllocation { //Destroyed IP private List destroyCompareIp(List newList, List oldList) { List destroyIpList = new ArrayList<>(); - for(String ip : oldList) { - if(!newList.contains(ip)) { + for (String ip : oldList) { + if (!newList.contains(ip)) { destroyIpList.add(ip); } } @@ -149,52 +157,51 @@ public class RuleAllocation { //Residual IP after destruction private List restIp(List destroyIp) { List restIpList = new ArrayList<>(); - for(String ip : engineService) { - if(!destroyIp.contains(ip)) { + for (String ip : engineService) { + if (!destroyIp.contains(ip)) { restIpList.add(ip); } } return restIpList; } - public void AllocateService(List extendIpList, List subList) throws Exception{ + public void AllocateService(List extendIpList, List subList) throws Exception { List needIpList = getSortIp(extendIpList); - for(int i=0,j=0;j < subList.size();i++,j++ ){ + for (int i = 0, j = 0; j < subList.size(); i++, j++) { int index = i % needIpList.size(); String deployIp = needIpList.get(index); CorrelationRule rule = subList.get(j); rule.setEngineInstance(deployIp); - allocateDeployRule(rule,deployIp); + allocateDeployRule(rule, deployIp); } } //The IP to be allocated is in ascending order, and the least is circulate. - private List getSortIp(List ipList){ - List ipRuleList = new ArrayList<>(); - HashMap hashMap = new HashMap(); + private List getSortIp(List ipList) { + List ipRuleList = new ArrayList<>(); + HashMap hashMap = new HashMap(); - try{ - for(String ip : ipList){ + try { + for (String ip : ipList) { ipRuleList = ruleQueryWrapper.queryRuleByEngineInstance(ip); - if(ipRuleList != null) { + if (ipRuleList != null) { hashMap.put(ip, String.valueOf(ipRuleList.size())); } } - }catch (Exception e){ - log.error("getEngineIp4AddRule failed !" + e.getMessage()); + } catch (Exception e) { + log.error("getEngineIp4AddRule failed !", e); } List> list_Data = new ArrayList>(hashMap.entrySet()); Collections.sort(list_Data, new Comparator>() { - public int compare(Map.Entry o1, Map.Entry o2) - { + public int compare(Map.Entry o1, Map.Entry o2) { return o1.getValue().compareTo(o2.getValue()); } }); List needList = new ArrayList<>(); - for(Map.Entry map: list_Data) { + for (Map.Entry map : list_Data) { String key = map.getKey(); needList.add(key); } @@ -202,25 +209,25 @@ public class RuleAllocation { } private void allocateDeployRule(CorrelationRule rule, String ip) throws CorrelationException { - try{ - ruleMgtWrapper.deployRule2Engine(rule,ip); + try { + ruleMgtWrapper.deployRule2Engine(rule, ip); correlationRuleDao.updateRule(rule); - }catch (CorrelationException e){ + } catch (CorrelationException e) { throw new CorrelationException("allocate Deploy Rule failed", e); } } private void deleteRuleFromFormerEngine(List subRule, List oldList) { - try{ - for(String ip : oldList){ - for(CorrelationRule rule: subRule) { - if(ip.equals(rule.getEngineInstance())) { - engineWrapper.deleteRuleFromEngine(rule.getPackageName(),ip); + try { + for (String ip : oldList) { + for (CorrelationRule rule : subRule) { + if (ip.equals(rule.getEngineInstance())) { + engineWrapper.deleteRuleFromEngine(rule.getPackageName(), ip); } } } - }catch (CorrelationException e) { - log.error("When the engine is extended, deleting rule failed" +e.getMessage()); + } catch (CorrelationException e) { + log.error("When the engine is extended, deleting rule failed", e); } } -- cgit 1.2.3-korg