/**
* Copyright 2017 ZTE Corporation.
- *
+ * <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
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
+ * <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.
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;
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<String> temIpList = new ArrayList<>();
private List<String> engineService = new ArrayList<>();
private List<CorrelationRule> 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<String> ipList)throws Exception{
- if(ipList != null) {
+ public void judgeAndAllocateRule(List<String> ipList) throws Exception {
+ if (ipList != null) {
engineService = ipList;
serviceCount = ipList.size();
}
- if(temIpList.size() < serviceCount){
+ if (temIpList.size() < serviceCount) {
//extend
List<CorrelationRule> deleteRule = calculateRule(temIpList);
- List<CorrelationRule> allocateRule = calculateRule(temIpList);
- List<String> extendIp = extendCompareIp(engineService,temIpList);
- AllocateService(extendIp,allocateRule);
- deleteRuleFromFormerEngine(deleteRule,temIpList);
+ List<CorrelationRule> allocateRule = calculateRule(temIpList);
+ List<String> extendIp = extendCompareIp(engineService, temIpList);
+ AllocateService(extendIp, allocateRule);
+ deleteRuleFromFormerEngine(deleteRule, temIpList);
} else if (temIpList.size() > serviceCount) {
//destroy
List<String> destroyIp = destroyCompareIp(engineService, temIpList);
AllocateService(restIp(destroyIp), relocateRuleAfterDestroy(destroyIp));
- } else if(temIpList.size() == serviceCount) {
+ } else if (temIpList.size() == serviceCount) {
temIpList = engineService;
return;
}
// When the engine is expanding, the rules that need to be allocated are calculated.
- private List<CorrelationRule> calculateRule(List<String> oldIpList) throws Exception{
+ private List<CorrelationRule> calculateRule(List<String> 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<CorrelationRule> subRule = new ArrayList<>();
- for(String ip : oldIpList) {
+ for (String ip : oldIpList) {
List<CorrelationRule> rules = ruleQueryWrapper.queryRuleByEngineInstance(ip);
- List<CorrelationRule> tem = rules.subList(count + (remainder-- / oldIpList.size()),rules.size());
+ List<CorrelationRule> tem = rules.subList(count + (remainder-- / oldIpList.size()), rules.size());
subRule.addAll(tem);
}
return subRule;
//Rules that need to be allocated after the engine is destroyed
private List<CorrelationRule> relocateRuleAfterDestroy(List<String> destroyIpList) throws CorrelationException {
List<CorrelationRule> 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<String> extendCompareIp(List<String> newList, List<String> oldList){
+ private List<String> extendCompareIp(List<String> newList, List<String> oldList) {
List<String> extendIpList = new ArrayList<>();
- for( String ip :newList) {
- if(! oldList.contains(ip)) {
+ for (String ip : newList) {
+ if (!oldList.contains(ip)) {
extendIpList.add(ip);
}
}
//Destroyed IP
private List<String> destroyCompareIp(List<String> newList, List<String> oldList) {
List<String> destroyIpList = new ArrayList<>();
- for(String ip : oldList) {
- if(!newList.contains(ip)) {
+ for (String ip : oldList) {
+ if (!newList.contains(ip)) {
destroyIpList.add(ip);
}
}
//Residual IP after destruction
private List<String> restIp(List<String> destroyIp) {
List<String> 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<String> extendIpList, List<CorrelationRule> subList) throws Exception{
+ public void AllocateService(List<String> extendIpList, List<CorrelationRule> subList) throws Exception {
List<String> 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<String > getSortIp(List<String> ipList){
- List<CorrelationRule> ipRuleList = new ArrayList<>();
- HashMap<String,String> hashMap = new HashMap();
+ private List<String> getSortIp(List<String> ipList) {
+ List<CorrelationRule> ipRuleList = new ArrayList<>();
+ HashMap<String, String> 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<Map.Entry<String, String>> list_Data = new ArrayList<Map.Entry<String, String>>(hashMap.entrySet());
Collections.sort(list_Data, new Comparator<Map.Entry<String, String>>() {
- public int compare(Map.Entry<String, String> o1, Map.Entry<String, String> o2)
- {
+ public int compare(Map.Entry<String, String> o1, Map.Entry<String, String> o2) {
return o1.getValue().compareTo(o2.getValue());
}
});
List<String> needList = new ArrayList<>();
- for(Map.Entry<String, String> map: list_Data) {
+ for (Map.Entry<String, String> map : list_Data) {
String key = map.getKey();
needList.add(key);
}
}
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<CorrelationRule> subRule, List<String> 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);
}
}
Code Review / holmes / rule-management.git / blobdiff