holmes-actions/src/main/java/org/onap/holmes/common/dmaap/DmaapService.java

   1 /*
   2  * Copyright 2017 ZTE Corporation.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package org.onap.holmes.common.dmaap;
  17
  18 import java.util.List;
  19 import java.util.Optional;
  20 import lombok.AllArgsConstructor;
  21 import lombok.extern.slf4j.Slf4j;
  22 import org.onap.holmes.common.aai.AaiQuery;
  23 import org.onap.holmes.common.aai.entity.RelationshipList.RelationshipData;
  24 import org.onap.holmes.common.aai.entity.VmEntity;
  25 import org.onap.holmes.common.aai.entity.VnfEntity;
  26 import org.onap.holmes.common.api.stat.VesAlarm;
  27 import org.onap.holmes.common.dmaap.entity.PolicyMsg;
  28 import org.onap.holmes.common.dmaap.entity.PolicyMsg.EVENT_STATUS;
  29 import org.onap.holmes.common.exception.CorrelationException;
  30
  31 @Slf4j
  32 @AllArgsConstructor
  33 public class DmaapService {
  34
  35     private static AaiQuery aaiQuery;
  36     private static Publisher publisher;
  37
  38     public static void publishPolicyMsg(PolicyMsg policyMsg) {
  39         try {
  40             publisher.publish(policyMsg);
  41         } catch (CorrelationException e) {
  42             log.error("Failed to publish policyMsg to dmaap", e.getMessage());
  43         }
  44     }
  45
  46     public static PolicyMsg getPolicyMsg(VesAlarm vesAlarm) {
  47         return Optional.ofNullable(getVmEntity(vesAlarm.getSourceId(), vesAlarm.getSourceName()))
  48                 .map(vmEntity -> getEnrichedPolicyMsg(vmEntity, vesAlarm))
  49                 .orElse(getDefaultPolicyMsg(vesAlarm.getSourceName()));
  50     }
  51
  52     private static String getVserverInstanceId(VnfEntity vnfEntity) {
  53         String vserverInstanceId = "";
  54         if (vnfEntity != null) {
  55             List<RelationshipData> relationshipDataList = vnfEntity.getRelationshipList()
  56                     .getRelationships().stream()
  57                     .filter(relationship -> relationship.getRelatedTo().equals("service-instance"))
  58                     .limit(1).findFirst().get().getRelationshipDataList();
  59
  60             vserverInstanceId = relationshipDataList.stream()
  61                     .filter(relationshipData -> relationshipData.getRelationshipKey().equals("service-instance.service-instance-id"))
  62                     .findFirst()
  63                     .map(relationshipData -> relationshipData.getRelationshipValue()).get();
  64         }
  65         return vserverInstanceId;
  66     }
  67
  68     private static VnfEntity getVnfEntity(String vnfId, String vnfName) {
  69         VnfEntity vnfEntity = null;
  70         try {
  71             vnfEntity = aaiQuery.getAaiVnfData(vnfId, vnfName);
  72         } catch (CorrelationException e) {
  73             log.error("Failed to get vnf data", e.getMessage());
  74         }
  75         return vnfEntity;
  76     }
  77
  78     private static VmEntity getVmEntity(String sourceId, String sourceName) {
  79         VmEntity vmEntity = null;
  80         try {
  81             vmEntity = aaiQuery.getAaiVmData(sourceId, sourceName);
  82         } catch (CorrelationException e) {
  83             log.error("Failed to get vm data", e.getMessage());
  84         }
  85         return vmEntity;
  86     }
  87
  88     private static PolicyMsg getEnrichedPolicyMsg(VmEntity vmEntity, VesAlarm vesAlarm) {
  89         VnfEntity vnfEntity = getVnfEntity(vesAlarm.getEventId(), vesAlarm.getEventName());
  90         String vserverInstatnceId = getVserverInstanceId(vnfEntity);
  91         PolicyMsg policyMsg = new PolicyMsg();
  92         policyMsg.setClosedLoopEventClient("DCAE_INSTANCE_ID.dcae-tca");
  93         policyMsg.setPolicyVersion("1.0.0.5");
  94         policyMsg.setPolicyName("vLoadBalancer");
  95         policyMsg.setPolicyScope(
  96                 "resource=SampleResource,service=SampleService,type=SampleType,closedLoopControlName=SampleClosedLoop");
  97         policyMsg.setTargetType("VM");
  98         policyMsg.setClosedLoopAlarmStart(1484855);
  99         policyMsg.setClosedLoopEventStatus(EVENT_STATUS.ONSET);
 100         policyMsg.setClosedLoopControlName(
 101                 "CL-LB-LOW-TRAFFIC-SIG-d925ed73-8231-4d02-9545-db4e101f88f8");
 102         policyMsg.setVersion("1.0.2");
 103         policyMsg.setTarget("generic-vnf.vnf-id");
 104         policyMsg.setRequestID("8c1b8bd8-06f7-493f-8ed7-daaa4cc481bc");
 105         policyMsg.setFrom("DCAE");
 106         policyMsg.getAai().put("vserver.in-maint", String.valueOf(vmEntity.getInMaint()));
 107         policyMsg.getAai().put("vserver.is-closed-loop-disabled",
 108                 String.valueOf(vmEntity.getClosedLoopDisable()));
 109         policyMsg.getAai().put("vserver.prov-status", vmEntity.getProvStatus());
 110         policyMsg.getAai().put("vserver.resource-version", vmEntity.getResourceVersion());
 111         policyMsg.getAai().put("vserver.vserver-id", vmEntity.getVserverId());
 112         policyMsg.getAai().put("vserver.vserver-name", vmEntity.getVserverName());
 113         policyMsg.getAai().put("vserver.vserver-name2", vmEntity.getVserverName2());
 114         policyMsg.getAai().put("vserver.vserver-selflink", vmEntity.getVserverSelflink());
 115         policyMsg.getAai().put("generic-vnf.vnf-id", vesAlarm.getEventId());
 116         policyMsg.getAai().put("generic-vnf.vnf-name", vesAlarm.getEventName());
 117         policyMsg.getAai().put("generic-vnf.service-instance-id", vserverInstatnceId);
 118         return policyMsg;
 119     }
 120
 121     private static PolicyMsg getDefaultPolicyMsg(String sourceName) {
 122         PolicyMsg policyMsg = new PolicyMsg();
 123         policyMsg.setTarget("vserver.vserver-name");
 124         policyMsg.setTargetType("VM");
 125         policyMsg.getAai().put("vserver.vserver-name", sourceName);
 126         return policyMsg;
 127     }
 128 }