import java.util.Locale;\r
import java.util.Set;\r
import javax.annotation.PostConstruct;\r
-import javax.annotation.PreDestroy;\r
import javax.inject.Inject;\r
import lombok.extern.slf4j.Slf4j;\r
import org.drools.compiler.kie.builder.impl.InternalKieModule;\r
import org.jvnet.hk2.annotations.Service;\r
+\r
import org.kie.api.KieBase;\r
import org.kie.api.KieServices;\r
import org.kie.api.builder.KieBuilder;\r
import org.kie.api.runtime.KieSession;\r
import org.kie.api.runtime.conf.ClockTypeOption;\r
import org.kie.api.runtime.rule.FactHandle;\r
+\r
+import org.onap.holmes.common.api.entity.AlarmInfo;\r
+\r
import org.onap.holmes.common.api.stat.VesAlarm;\r
import org.onap.holmes.common.dmaap.DmaapService;\r
+import org.onap.holmes.common.exception.AlarmInfoException;\r
+import org.onap.holmes.common.utils.DbDaoUtil;\r
+import org.onap.holmes.engine.db.AlarmInfoDao;\r
import org.onap.holmes.engine.request.DeployRuleRequest;\r
import org.onap.holmes.common.api.entity.CorrelationRule;\r
import org.onap.holmes.common.exception.CorrelationException;\r
@Service\r
public class DroolsEngine {\r
\r
- private final static int ENABLE = 1;\r
+ private static final int ENABLE = 1;\r
+ public static final String UTF_8 = "UTF-8";\r
+ public static final String K_BASE = "KBase";\r
+ private static final String RULES_FILE_NAME = "src/main/resources/rules/rule.drl";\r
private final Set<String> packageNames = new HashSet<String>();\r
+\r
@Inject\r
private RuleMgtWrapper ruleMgtWrapper;\r
\r
+\r
private KieBase kieBase;\r
private KieSession kieSession;\r
private KieContainer kieContainer;\r
private KieResources resources;\r
private KieRepository kieRepository;\r
\r
+ private AlarmInfoDao alarmInfoDao;\r
+ @Inject\r
+ private DbDaoUtil daoUtil;\r
+\r
+\r
@PostConstruct\r
private void init() {\r
+ alarmInfoDao = daoUtil.getJdbiDaoByOnDemand(AlarmInfoDao.class);\r
try {\r
// start engine\r
start();\r
}\r
}\r
\r
- private void start() throws CorrelationException {\r
+ private void start() throws AlarmInfoException {\r
log.info("Drools Engine Initialize Beginning...");\r
\r
initEngineParameter();\r
- initDeployRule();\r
+ alarmSynchronization();\r
+// initDeployRule();\r
\r
- log.info("Business Rule Engine Initialize Successfully.");\r
+ log.info("Alarm synchronization Successfully.");\r
}\r
\r
public void stop() {\r
private void deployRuleFromDB(String ruleContent) throws CorrelationException {\r
avoidDeployBug();\r
StringReader reader = new StringReader(ruleContent);\r
- kfs.write("src/main/resources/rules/rule.drl",\r
- this.resources.newReaderResource(reader,"UTF-8").setResourceType(ResourceType.DRL));\r
+ kfs.write(RULES_FILE_NAME,\r
+ this.resources.newReaderResource(reader, UTF_8).setResourceType(ResourceType.DRL));\r
kieBuilder = ks.newKieBuilder(kfs).buildAll();\r
try {\r
InternalKieModule internalKieModule = (InternalKieModule)kieBuilder.getKieModule();\r
throws CorrelationException {\r
avoidDeployBug();\r
StringReader reader = new StringReader(rule.getContent());\r
- kfs.write("src/main/resources/rules/rule.drl",\r
- this.resources.newReaderResource(reader,"UTF-8").setResourceType(ResourceType.DRL));\r
+ kfs.write(RULES_FILE_NAME,\r
+ this.resources.newReaderResource(reader, UTF_8).setResourceType(ResourceType.DRL));\r
kieBuilder = ks.newKieBuilder(kfs).buildAll();\r
\r
judgeRuleContent(locale, kieBuilder, true);\r
\r
InternalKieModule internalKieModule = (InternalKieModule)kieBuilder.getKieModule();;\r
- String packageName = internalKieModule.getKnowledgePackagesForKieBase("KBase").iterator().next().getName();\r
+ String packageName = internalKieModule.getKnowledgePackagesForKieBase(K_BASE).iterator().next().getName();\r
try {\r
kieContainer.updateToVersion(internalKieModule.getReleaseId());\r
} catch (Exception e) {\r
throws CorrelationException {\r
StringReader reader = new StringReader(content);\r
\r
- kfs.write("src/main/resources/rules/rule.drl",\r
- this.resources.newReaderResource(reader,"UTF-8").setResourceType(ResourceType.DRL));\r
+ kfs.write(RULES_FILE_NAME,\r
+ this.resources.newReaderResource(reader, UTF_8).setResourceType(ResourceType.DRL));\r
\r
kieBuilder = ks.newKieBuilder(kfs).buildAll();\r
\r
if (internalKieModule == null) {\r
throw new CorrelationException("There are errors in the rule!");\r
}\r
- String packageName = internalKieModule.getKnowledgePackagesForKieBase("KBase").iterator().next().getName();\r
+ String packageName = internalKieModule.getKnowledgePackagesForKieBase(K_BASE).iterator().next().getName();\r
\r
if (queryAllPackage().contains(packageName) && judgePackageName) {\r
throw new CorrelationException("The rule " + packageName + " already exists in the drools engine.");\r
}\r
}\r
\r
- public void putRaisedIntoStream(VesAlarm raiseAlarm) {\r
- FactHandle factHandle = this.kieSession.getFactHandle(raiseAlarm);\r
+ public void putRaisedIntoStream(VesAlarm alarm) {\r
+ FactHandle factHandle = this.kieSession.getFactHandle(alarm);\r
if (factHandle != null) {\r
Object obj = this.kieSession.getObject(factHandle);\r
if (obj != null && obj instanceof VesAlarm) {\r
- raiseAlarm.setRootFlag(((VesAlarm) obj).getRootFlag());\r
+ alarm.setRootFlag(((VesAlarm) obj).getRootFlag());\r
}\r
this.kieSession.delete(factHandle);\r
+ \r
+ if (alarm.getAlarmIsCleared() == 1) {\r
+ alarmInfoDao.deleteClearedAlarm(convertVesAlarm2AlarmInfo(alarm));\r
+ }\r
+ } else {\r
+ this.kieSession.insert(alarm);\r
}\r
- this.kieSession.insert(raiseAlarm);\r
+\r
this.kieSession.fireAllRules();\r
+\r
}\r
\r
public List<String> queryAllPackage() {\r
\r
private KieFileSystem createKieFileSystemWithKProject(KieServices ks) {\r
KieModuleModel kieModuleModel = ks.newKieModuleModel();\r
- KieBaseModel kieBaseModel = kieModuleModel.newKieBaseModel("KBase")\r
+ KieBaseModel kieBaseModel = kieModuleModel.newKieBaseModel(K_BASE)\r
.addPackage("rules")\r
.setDefault(true)\r
.setEqualsBehavior(EqualityBehaviorOption.EQUALITY)\r
private void avoidDeployBug() {\r
String tmp = Math.random() + "";\r
String rule = "package justInOrderToAvoidDeployBug" + tmp.substring(2);\r
- kfs.write("src/main/resources/rules/rule.drl", rule);\r
+ kfs.write(RULES_FILE_NAME, rule);\r
kieBuilder = ks.newKieBuilder(kfs).buildAll();\r
InternalKieModule internalKieModule = (InternalKieModule)kieBuilder.getKieModule();\r
- String packageName = internalKieModule.getKnowledgePackagesForKieBase("KBase").iterator().next().getName();\r
+ String packageName = internalKieModule.getKnowledgePackagesForKieBase(K_BASE).iterator().next().getName();\r
kieRepository.addKieModule(internalKieModule);\r
kieContainer.updateToVersion(internalKieModule.getReleaseId());\r
\r
kieBase.removeKiePackage(kiePackage.getName());\r
}\r
\r
+ public void alarmSynchronization() throws AlarmInfoException {\r
+ alarmInfoDao.queryAllAlarm().forEach(alarmInfo -> alarmInfoDao.deleteClearedAlarm(alarmInfo));\r
+ alarmInfoDao.queryAllAlarm().forEach(alarmInfo -> putRaisedIntoStream(convertAlarmInfo2VesAlarm(alarmInfo)));\r
+ }\r
+\r
+ private VesAlarm convertAlarmInfo2VesAlarm(AlarmInfo alarmInfo) {\r
+ VesAlarm vesAlarm = new VesAlarm();\r
+ vesAlarm.setEventId(alarmInfo.getEventId());\r
+ vesAlarm.setEventName(alarmInfo.getEventName());\r
+ vesAlarm.setStartEpochMicrosec(alarmInfo.getStartEpochMicroSec());\r
+ vesAlarm.setSourceId(alarmInfo.getSourceId());\r
+ vesAlarm.setSourceName(alarmInfo.getSourceName());\r
+ vesAlarm.setRootFlag(alarmInfo.getRootFlag());\r
+ vesAlarm.setAlarmIsCleared(alarmInfo.getAlarmIsCleared());\r
+ vesAlarm.setLastEpochMicrosec(alarmInfo.getLastEpochMicroSec());\r
+ return vesAlarm;\r
+ }\r
+\r
+ private AlarmInfo convertVesAlarm2AlarmInfo(VesAlarm vesAlarm){\r
+ AlarmInfo alarmInfo = new AlarmInfo();\r
+ alarmInfo.setEventId(vesAlarm.getEventId());\r
+ alarmInfo.setEventName(vesAlarm.getEventName());\r
+ alarmInfo.setStartEpochMicroSec(vesAlarm.getStartEpochMicrosec());\r
+ alarmInfo.setLastEpochMicroSec(vesAlarm.getLastEpochMicrosec());\r
+ alarmInfo.setSourceId(vesAlarm.getSourceId());\r
+ alarmInfo.setSourceName(vesAlarm.getSourceName());\r
+ alarmInfo.setAlarmIsCleared(vesAlarm.getAlarmIsCleared());\r
+ alarmInfo.setRootFlag(vesAlarm.getRootFlag());\r
+\r
+ return alarmInfo;\r
+ }\r
+\r
}\r
Code Review / holmes / engine-management.git / blobdiff