Remove alarms from DB when they are cleared

[holmes/engine-management.git] / engine-d / src / main / java / org / onap / holmes / engine / manager / DroolsEngine.java

diff --git a/engine-d/src/main/java/org/onap/holmes/engine/manager/DroolsEngine.java b/engine-d/src/main/java/org/onap/holmes/engine/manager/DroolsEngine.java
index d790ee7..dcc40fd 100644 (file)
--- a/engine-d/src/main/java/org/onap/holmes/engine/manager/DroolsEngine.java
+++ b/engine-d/src/main/java/org/onap/holmes/engine/manager/DroolsEngine.java
@@ -21,11 +21,11 @@ import java.util.List;
 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
@@ -45,8 +45,14 @@ import org.kie.api.runtime.KieContainer;
 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
@@ -62,6 +68,7 @@ public class DroolsEngine {
     @Inject\r
     private RuleMgtWrapper ruleMgtWrapper;\r
 \r
+\r
     private KieBase kieBase;\r
     private KieSession kieSession;\r
     private KieContainer kieContainer;\r
@@ -71,8 +78,14 @@ public class DroolsEngine {
     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
@@ -82,13 +95,14 @@ public class DroolsEngine {
         }\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
@@ -209,17 +223,24 @@ public class DroolsEngine {
         }\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
@@ -261,4 +282,36 @@ public class DroolsEngine {
         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