From 680b0cc11beb6a8dd6daeda860ddad74cd709d43 Mon Sep 17 00:00:00 2001 From: Shiwei Tian Date: Mon, 25 Sep 2017 10:52:54 +0800 Subject: [PATCH] add push policy message Issue-ID: HOLMES-44 Change-Id: I7280e282627f5fc5bd00c47c19e9803c95c5ed3b Signed-off-by: Shiwei Tian --- .../org/onap/holmes/common/dmaap/DmaapService.java | 128 +++++++++++++++++++++ 1 file changed, 128 insertions(+) create mode 100644 holmes-actions/src/main/java/org/onap/holmes/common/dmaap/DmaapService.java 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 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; + } +} -- 2.16.6