2 * Copyright 2017 ZTE Corporation.
\r
4 * Licensed under the Apache License, Version 2.0 (the "License");
\r
5 * you may not use this file except in compliance with the License.
\r
6 * You may obtain a copy of the License at
\r
8 * http://www.apache.org/licenses/LICENSE-2.0
\r
10 * Unless required by applicable law or agreed to in writing, software
\r
11 * distributed under the License is distributed on an "AS IS" BASIS,
\r
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
\r
13 * See the License for the specific language governing permissions and
\r
14 * limitations under the License.
\r
16 package org.onap.holmes.rulemgt.wrapper;
\r
18 import java.util.ArrayList;
\r
19 import java.util.Date;
\r
20 import java.util.List;
\r
21 import javax.annotation.PostConstruct;
\r
22 import javax.inject.Inject;
\r
23 import javax.inject.Singleton;
\r
24 import lombok.extern.slf4j.Slf4j;
\r
25 import org.jvnet.hk2.annotations.Service;
\r
26 import org.onap.holmes.common.dmaap.DmaapService;
\r
27 import org.onap.holmes.rulemgt.bean.request.CorrelationCheckRule4Engine;
\r
28 import org.onap.holmes.rulemgt.bean.response.RuleResult4API;
\r
29 import org.onap.holmes.rulemgt.constant.RuleMgtConstant;
\r
30 import org.onap.holmes.rulemgt.db.CorrelationRuleDao;
\r
31 import org.onap.holmes.common.api.entity.CorrelationRule;
\r
32 import org.onap.holmes.common.exception.CorrelationException;
\r
33 import org.onap.holmes.common.utils.DbDaoUtil;
\r
34 import org.onap.holmes.rulemgt.bean.request.CorrelationDeployRule4Engine;
\r
35 import org.onap.holmes.rulemgt.bean.request.RuleCreateRequest;
\r
36 import org.onap.holmes.rulemgt.bean.request.RuleDeleteRequest;
\r
37 import org.onap.holmes.rulemgt.bean.request.RuleQueryCondition;
\r
38 import org.onap.holmes.rulemgt.bean.request.RuleUpdateRequest;
\r
39 import org.onap.holmes.rulemgt.bean.response.RuleAddAndUpdateResponse;
\r
40 import org.onap.holmes.rulemgt.bean.response.RuleQueryListResponse;
\r
41 import org.onap.holmes.rulemgt.bolt.enginebolt.EngineWrapper;
\r
42 import org.onap.holmes.rulemgt.db.CorrelationRuleQueryDao;
\r
48 public class RuleMgtWrapper {
\r
51 private CorrelationRuleQueryDao correlationRuleQueryDao;
\r
53 private EngineWrapper engineWarpper;
\r
55 private DbDaoUtil daoUtil;
\r
57 private CorrelationRuleDao correlationRuleDao;
\r
60 public void initDaoUtil() {
\r
61 correlationRuleDao = daoUtil.getJdbiDaoByOnDemand(CorrelationRuleDao.class);
\r
64 public RuleAddAndUpdateResponse addCorrelationRule(String creator, RuleCreateRequest ruleCreateRequest)
\r
65 throws CorrelationException {
\r
66 if (ruleCreateRequest == null) {
\r
67 throw new CorrelationException("The request object can not be empty!");
\r
69 CorrelationRule correlationRule = convertCreateRequest2Rule(creator,
\r
71 checkCorrelation(correlationRule);
\r
72 CorrelationRule ruleTemp = correlationRuleDao.queryRuleByRuleName(correlationRule.getName());
\r
73 if (ruleTemp != null) {
\r
74 throw new CorrelationException("A rule with the same name already exists.");
\r
77 String packageName = deployRule2Engine(correlationRule);
\r
78 DmaapService.loopControlNames.put(packageName, ruleCreateRequest.getLoopControlName());
\r
79 correlationRule.setPackageName(packageName);
\r
80 CorrelationRule result = null;
\r
82 result = correlationRuleDao.saveRule(correlationRule);
\r
83 } catch (CorrelationException e) {
\r
84 engineWarpper.deleteRuleFromEngine(packageName);
\r
85 throw new CorrelationException(e.getMessage());
\r
87 RuleAddAndUpdateResponse ruleAddAndUpdateResponse = new RuleAddAndUpdateResponse();
\r
88 ruleAddAndUpdateResponse.setRuleId(result.getRid());
\r
89 return ruleAddAndUpdateResponse;
\r
92 public RuleAddAndUpdateResponse updateCorrelationRule(String modifier, RuleUpdateRequest ruleUpdateRequest)
\r
93 throws CorrelationException {
\r
94 if (ruleUpdateRequest == null) {
\r
95 throw new CorrelationException("The request object can not be empty!");
\r
97 CorrelationRule oldCorrelationRule = correlationRuleDao.queryRuleByRid(ruleUpdateRequest.getRuleId());
\r
98 if (oldCorrelationRule == null) {
\r
99 throw new CorrelationException("You're trying to update a rule which does not exist in the system.");
\r
101 CorrelationRule newCorrelationRule = convertRuleUpdateRequest2CorrelationRule(modifier,
\r
102 ruleUpdateRequest, oldCorrelationRule.getName());
\r
103 checkCorrelation(newCorrelationRule);
\r
104 RuleAddAndUpdateResponse ruleChangeResponse = new RuleAddAndUpdateResponse();
\r
105 ruleChangeResponse.setRuleId(newCorrelationRule.getRid());
\r
106 if (!haveChange(newCorrelationRule, oldCorrelationRule)) {
\r
107 return ruleChangeResponse;
\r
109 if (oldCorrelationRule.getEnabled() == RuleMgtConstant.STATUS_RULE_OPEN) {
\r
110 engineWarpper.deleteRuleFromEngine(oldCorrelationRule.getPackageName());
\r
112 newCorrelationRule.setPackageName(deployRule2Engine(newCorrelationRule));
\r
113 correlationRuleDao.updateRule(newCorrelationRule);
\r
114 return ruleChangeResponse;
\r
117 private void checkCorrelation(CorrelationRule correlationRule) throws CorrelationException {
\r
118 int enabled = correlationRule.getEnabled();
\r
119 String ruleName = correlationRule.getName() == null ? "" : correlationRule.getName().trim();
\r
120 String content = correlationRule.getContent() == null ? "" : correlationRule.getContent().trim();
\r
121 if ("".equals(content)) {
\r
122 throw new CorrelationException("The contents of the rule can not be empty!");
\r
124 if (enabled != RuleMgtConstant.STATUS_RULE_CLOSE
\r
125 && enabled != RuleMgtConstant.STATUS_RULE_OPEN) {
\r
126 throw new CorrelationException("Invalid rule status. Only 0 (disabled) and 1 (enabled) are allowed.");
\r
128 if ("".equals(ruleName)) {
\r
129 throw new CorrelationException("The name of the rule can not be empty.");
\r
133 private boolean haveChange(CorrelationRule newCorrelationRule, CorrelationRule oldCorrelationRule) {
\r
134 String newContent = newCorrelationRule.getContent();
\r
135 String oldContent = oldCorrelationRule.getContent();
\r
136 int newEnabled = newCorrelationRule.getEnabled();
\r
137 int oldEnabled = oldCorrelationRule.getEnabled();
\r
138 String newDes = newCorrelationRule.getDescription();
\r
139 String oldDes = oldCorrelationRule.getDescription();
\r
140 if (newContent.equals(oldContent) && newEnabled == oldEnabled && newDes.equals(oldDes)) {
\r
146 public void deleteCorrelationRule(RuleDeleteRequest ruleDeleteRequest)
\r
147 throws CorrelationException {
\r
148 if (ruleDeleteRequest == null) {
\r
149 throw new CorrelationException("The request object can not be empty!");
\r
151 CorrelationRule correlationRule = correlationRuleDao.queryRuleByRid(ruleDeleteRequest.getRuleId());
\r
152 if (correlationRule == null) {
\r
153 log.warn("the rule:rule id=" + ruleDeleteRequest.getRuleId() + " does not exist the database.");
\r
154 throw new CorrelationException("You're trying to delete a rule which does not exist in the system.");
\r
156 if (correlationRule.getEnabled() == RuleMgtConstant.STATUS_RULE_OPEN) {
\r
157 engineWarpper.deleteRuleFromEngine(correlationRule.getPackageName());
\r
159 correlationRuleDao.deleteRule(correlationRule);
\r
162 private CorrelationRule convertCreateRequest2Rule(String userName,
\r
163 RuleCreateRequest ruleCreateRequest) throws CorrelationException {
\r
164 String tempContent = ruleCreateRequest.getContent();
\r
165 CorrelationRule correlationRule = new CorrelationRule();
\r
166 String ruleId = "rule_" + System.currentTimeMillis();
\r
167 String description = ruleCreateRequest.getDescription() == null ? "" : ruleCreateRequest.getDescription();
\r
168 correlationRule.setRid(ruleId);
\r
169 if (tempContent != null) {
\r
170 correlationRule.setContent(tempContent.trim());
\r
172 correlationRule.setDescription(description);
\r
173 correlationRule.setCreateTime(new Date());
\r
174 correlationRule.setUpdateTime(new Date());
\r
175 correlationRule.setName(ruleCreateRequest.getRuleName());
\r
176 correlationRule.setEngineID("correlation-d");
\r
177 correlationRule.setEngineType("");
\r
178 correlationRule.setTemplateID(0);
\r
179 correlationRule.setVendor("");
\r
180 correlationRule.setCreator(userName);
\r
181 correlationRule.setModifier(userName);
\r
182 correlationRule.setEnabled(ruleCreateRequest.getEnabled());
\r
183 return correlationRule;
\r
186 private CorrelationRule convertRuleUpdateRequest2CorrelationRule(String modifier,
\r
187 RuleUpdateRequest ruleUpdateRequest, String ruleName) throws CorrelationException {
\r
188 CorrelationRule correlationRule = new CorrelationRule();
\r
189 String description = ruleUpdateRequest.getDescription() == null ? "" : ruleUpdateRequest.getDescription();
\r
190 correlationRule.setRid(ruleUpdateRequest.getRuleId());
\r
191 correlationRule.setContent(ruleUpdateRequest.getContent());
\r
192 correlationRule.setDescription(description);
\r
193 correlationRule.setEnabled(ruleUpdateRequest.getEnabled());
\r
194 correlationRule.setUpdateTime(new Date());
\r
195 correlationRule.setModifier(modifier);
\r
196 correlationRule.setName(ruleName);
\r
197 return correlationRule;
\r
200 private String deployRule2Engine(CorrelationRule correlationRule)
\r
201 throws CorrelationException {
\r
202 if (engineWarpper.checkRuleFromEngine(correlationRules2CheckRule(correlationRule)) && (
\r
203 correlationRule.getEnabled() == RuleMgtConstant.STATUS_RULE_OPEN)) {
\r
204 return engineWarpper.deployEngine(correlationRules2DeployRule(correlationRule));
\r
209 public RuleQueryListResponse getCorrelationRuleByCondition(
\r
210 RuleQueryCondition ruleQueryCondition) throws CorrelationException {
\r
211 List<CorrelationRule> correlationRule = correlationRuleQueryDao
\r
212 .getCorrelationRulesByCondition(ruleQueryCondition);
\r
213 RuleQueryListResponse ruleQueryListResponse = new RuleQueryListResponse();
\r
214 ruleQueryListResponse.setTotalCount(correlationRule.size());
\r
215 ruleQueryListResponse
\r
216 .setCorrelationRules(correlationRules2RuleResult4APIs(correlationRule));
\r
217 return ruleQueryListResponse;
\r
220 private List<RuleResult4API> correlationRules2RuleResult4APIs(
\r
221 List<CorrelationRule> correlationRules) {
\r
222 List<RuleResult4API> ruleResult4APIs = new ArrayList<RuleResult4API>();
\r
223 for (CorrelationRule correlationRule : correlationRules) {
\r
224 RuleResult4API ruleResult4API = new RuleResult4API();
\r
225 String description = correlationRule.getDescription() == null ? "" : correlationRule.getDescription();
\r
226 ruleResult4API.setRuleId(correlationRule.getRid());
\r
227 ruleResult4API.setRuleName(correlationRule.getName());
\r
228 ruleResult4API.setDescription(description);
\r
229 ruleResult4API.setContent(correlationRule.getContent());
\r
230 ruleResult4API.setCreateTime(correlationRule.getCreateTime());
\r
231 ruleResult4API.setCreator(correlationRule.getCreator());
\r
232 ruleResult4API.setUpdateTime(correlationRule.getUpdateTime());
\r
233 ruleResult4API.setModifier(correlationRule.getModifier());
\r
234 ruleResult4API.setEnabled(correlationRule.getEnabled());
\r
235 ruleResult4APIs.add(ruleResult4API);
\r
237 return ruleResult4APIs;
\r
240 private CorrelationDeployRule4Engine correlationRules2DeployRule(
\r
241 CorrelationRule correlationRule) {
\r
242 CorrelationDeployRule4Engine correlationDeployRule4Engine = new CorrelationDeployRule4Engine();
\r
243 correlationDeployRule4Engine.setContent(correlationRule.getContent());
\r
244 correlationDeployRule4Engine.setEngineId(correlationRule.getEngineID());
\r
245 return correlationDeployRule4Engine;
\r
248 private CorrelationCheckRule4Engine correlationRules2CheckRule(
\r
249 CorrelationRule correlationRule) {
\r
250 CorrelationCheckRule4Engine correlationCheckRule4Engine = new CorrelationCheckRule4Engine();
\r
251 correlationCheckRule4Engine.setContent(correlationRule.getContent());
\r
252 return correlationCheckRule4Engine;
\r