add push policy message

Issue-ID: HOLMES-44

Change-Id: I7280e282627f5fc5bd00c47c19e9803c95c5ed3b
Signed-off-by: Shiwei Tian <tian.shiwei@zte.com.cn>

1 files changed, 128 insertions, 0 deletions

diff --git a/holmes-actions/src/main/java/org/onap/holmes/common/dmaap/DmaapService.java b/holmes-actions/src/main/java/org/onap/holmes/common/dmaap/DmaapService.java
new file mode 100644
index 0000000..9c3d19b
--- /dev/null
+++ b/holmes-actions/src/main/java/org/onap/holmes/common/dmaap/DmaapService.java
@@ -0,0 +1,128 @@
+/*
+ * 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
+ *
+ * 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.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.onap.holmes.common.dmaap;
+
+import java.util.List;
+import java.util.Optional;
+import lombok.AllArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.onap.holmes.common.aai.AaiQuery;
+import org.onap.holmes.common.aai.entity.RelationshipList.RelationshipData;
+import org.onap.holmes.common.aai.entity.VmEntity;
+import org.onap.holmes.common.aai.entity.VnfEntity;
+import org.onap.holmes.common.api.stat.VesAlarm;
+import org.onap.holmes.common.dmaap.entity.PolicyMsg;
+import org.onap.holmes.common.dmaap.entity.PolicyMsg.EVENT_STATUS;
+import org.onap.holmes.common.exception.CorrelationException;
+
+@Slf4j
+@AllArgsConstructor
+public class DmaapService {
+
+    private static AaiQuery aaiQuery;
+    private static Publisher publisher;
+
+    public static void publishPolicyMsg(PolicyMsg policyMsg) {
+        try {
+            publisher.publish(policyMsg);
+        } catch (CorrelationException e) {
+            log.error("Failed to publish policyMsg to dmaap", e.getMessage());
+        }
+    }
+
+    public static PolicyMsg getPolicyMsg(VesAlarm vesAlarm) {
+        return Optional.ofNullable(getVmEntity(vesAlarm.getSourceId(), vesAlarm.getSourceName()))
+                .map(vmEntity -> getEnrichedPolicyMsg(vmEntity, vesAlarm))
+                .orElse(getDefaultPolicyMsg(vesAlarm.getSourceName()));
+    }
+
+    private static String getVserverInstanceId(VnfEntity vnfEntity) {
+        String vserverInstanceId = "";
+        if (vnfEntity != null) {
+            List<RelationshipData> relationshipDataList = vnfEntity.getRelationshipList()
+                    .getRelationships().stream()
+                    .filter(relationship -> relationship.getRelatedTo().equals("service-instance"))
+                    .limit(1).findFirst().get().getRelationshipDataList();
+
+            vserverInstanceId = relationshipDataList.stream()
+                    .filter(relationshipData -> relationshipData.getRelationshipKey().equals("service-instance.service-instance-id"))
+                    .findFirst()
+                    .map(relationshipData -> relationshipData.getRelationshipValue()).get();
+        }
+        return vserverInstanceId;
+    }
+
+    private static VnfEntity getVnfEntity(String vnfId, String vnfName) {
+        VnfEntity vnfEntity = null;
+        try {
+            vnfEntity = aaiQuery.getAaiVnfData(vnfId, vnfName);
+        } catch (CorrelationException e) {
+            log.error("Failed to get vnf data", e.getMessage());
+        }
+        return vnfEntity;
+    }
+
+    private static VmEntity getVmEntity(String sourceId, String sourceName) {
+        VmEntity vmEntity = null;
+        try {
+            vmEntity = aaiQuery.getAaiVmData(sourceId, sourceName);
+        } catch (CorrelationException e) {
+            log.error("Failed to get vm data", e.getMessage());
+        }
+        return vmEntity;
+    }
+
+    private static PolicyMsg getEnrichedPolicyMsg(VmEntity vmEntity, VesAlarm vesAlarm) {
+        VnfEntity vnfEntity = getVnfEntity(vesAlarm.getEventId(), vesAlarm.getEventName());
+        String vserverInstatnceId = getVserverInstanceId(vnfEntity);
+        PolicyMsg policyMsg = new PolicyMsg();
+        policyMsg.setClosedLoopEventClient("DCAE_INSTANCE_ID.dcae-tca");
+        policyMsg.setPolicyVersion("1.0.0.5");
+        policyMsg.setPolicyName("vLoadBalancer");
+        policyMsg.setPolicyScope(
+                "resource=SampleResource,service=SampleService,type=SampleType,closedLoopControlName=SampleClosedLoop");
+        policyMsg.setTargetType("VM");
+        policyMsg.setClosedLoopAlarmStart(1484855);
+        policyMsg.setClosedLoopEventStatus(EVENT_STATUS.ONSET);
+        policyMsg.setClosedLoopControlName(
+                "CL-LB-LOW-TRAFFIC-SIG-d925ed73-8231-4d02-9545-db4e101f88f8");
+        policyMsg.setVersion("1.0.2");
+        policyMsg.setTarget("generic-vnf.vnf-id");
+        policyMsg.setRequestID("8c1b8bd8-06f7-493f-8ed7-daaa4cc481bc");
+        policyMsg.setFrom("DCAE");
+        policyMsg.getAai().put("vserver.in-maint", String.valueOf(vmEntity.getInMaint()));
+        policyMsg.getAai().put("vserver.is-closed-loop-disabled",
+                String.valueOf(vmEntity.getClosedLoopDisable()));
+        policyMsg.getAai().put("vserver.prov-status", vmEntity.getProvStatus());
+        policyMsg.getAai().put("vserver.resource-version", vmEntity.getResourceVersion());
+        policyMsg.getAai().put("vserver.vserver-id", vmEntity.getVserverId());
+        policyMsg.getAai().put("vserver.vserver-name", vmEntity.getVserverName());
+        policyMsg.getAai().put("vserver.vserver-name2", vmEntity.getVserverName2());
+        policyMsg.getAai().put("vserver.vserver-selflink", vmEntity.getVserverSelflink());
+        policyMsg.getAai().put("generic-vnf.vnf-id", vesAlarm.getEventId());
+        policyMsg.getAai().put("generic-vnf.vnf-name", vesAlarm.getEventName());
+        policyMsg.getAai().put("generic-vnf.service-instance-id", vserverInstatnceId);
+        return policyMsg;
+    }
+
+    private static PolicyMsg getDefaultPolicyMsg(String sourceName) {
+        PolicyMsg policyMsg = new PolicyMsg();
+        policyMsg.setTarget("vserver.vserver-name");
+        policyMsg.setTargetType("VM");
+        policyMsg.getAai().put("vserver.vserver-name", sourceName);
+        return policyMsg;
+    }
+}