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 com.fasterxml.jackson.core.JsonProcessingException;
  19 import java.util.Optional;
  20 import java.util.UUID;
  21 import java.util.concurrent.ConcurrentHashMap;
  22 import javax.inject.Inject;
  23 import lombok.extern.slf4j.Slf4j;
  24 import org.jvnet.hk2.annotations.Service;
  25 import org.onap.holmes.common.aai.AaiQuery;
  26 import org.onap.holmes.common.aai.entity.RelationshipList.Relationship;
  27 import org.onap.holmes.common.aai.entity.VmEntity;
  28 import org.onap.holmes.common.aai.entity.VnfEntity;
  29 import org.onap.holmes.common.api.stat.VesAlarm;
  30 import org.onap.holmes.common.dcae.DcaeConfigurationsCache;
  31 import org.onap.holmes.common.dmaap.entity.PolicyMsg;
  32 import org.onap.holmes.common.dmaap.entity.PolicyMsg.EVENT_STATUS;
  33 import org.onap.holmes.common.exception.CorrelationException;
  34 import org.onap.holmes.common.utils.JacksonUtil;
  35
  36 @Slf4j
  37 @Service
  38 public class DmaapService {
  39
  40     @Inject
  41     private AaiQuery aaiQuery;
  42     public static ConcurrentHashMap<String, String> loopControlNames = new ConcurrentHashMap<>();
  43     public static ConcurrentHashMap<String, String> alarmUniqueRequestID = new ConcurrentHashMap<>();
  44
  45     public void publishPolicyMsg(PolicyMsg policyMsg, String dmaapConfigKey) {
  46         try {
  47             Publisher publisher = new Publisher();
  48             publisher.setUrl(DcaeConfigurationsCache.getPubSecInfo(dmaapConfigKey).getDmaapInfo()
  49                     .getTopicUrl());
  50             publisher.publish(policyMsg);
  51             log.info("send policyMsg: " + JacksonUtil.beanToJson(policyMsg));
  52         } catch (CorrelationException e) {
  53             log.error("Failed to publish the control loop event to DMaaP", e);
  54         } catch (JsonProcessingException e) {
  55             log.info("Failed to convert the control loop event to json");
  56         } catch (NullPointerException e) {
  57             log.error("DMaaP configurations do not exist!");
  58         }
  59     }
  60
  61     public PolicyMsg getPolicyMsg(VesAlarm rootAlarm, VesAlarm childAlarm, String packgeName) {
  62         return Optional.ofNullable(getVmEntity(rootAlarm.getSourceId(), rootAlarm.getSourceName()))
  63                 .map(vmEntity -> getEnrichedPolicyMsg(vmEntity, rootAlarm, childAlarm, packgeName))
  64                 .orElse(getDefaultPolicyMsg(rootAlarm, packgeName));
  65     }
  66
  67     private PolicyMsg getEnrichedPolicyMsg(VmEntity vmEntity, VesAlarm rootAlarm, VesAlarm childAlarm,
  68             String packageName) {
  69         PolicyMsg policyMsg = new PolicyMsg();
  70         policyMsg.setRequestID(getUniqueRequestId(rootAlarm));
  71         if (rootAlarm.getAlarmIsCleared() == PolicyMassgeConstant.POLICY_MESSAGE_ONSET) {
  72             enrichVnfInfo(vmEntity, childAlarm, policyMsg);
  73             policyMsg.setClosedLoopEventStatus(EVENT_STATUS.ONSET);
  74             policyMsg.getAai().put("vserver.in-maint", String.valueOf(vmEntity.getInMaint()));
  75             try {
  76                 policyMsg.getAai().put("vserver.is-closed-loop-disabled",
  77                         Boolean.valueOf(vmEntity.getClosedLoopDisable()).booleanValue());
  78             } catch (Exception e) {
  79                 log.error("Failed to parse the field \"is-closed-loop-disabled\". A boolean string (\"true\"/\"false\")"
  80                         + " is expected but the actual value is " + vmEntity.getClosedLoopDisable() + ".", e);
  81             }
  82             policyMsg.getAai().put("vserver.prov-status", vmEntity.getProvStatus());
  83             policyMsg.getAai().put("vserver.resource-version", vmEntity.getResourceVersion());
  84         } else {
  85             policyMsg.setClosedLoopAlarmEnd(rootAlarm.getLastEpochMicrosec());
  86             policyMsg.setClosedLoopEventStatus(EVENT_STATUS.ABATED);
  87         }
  88         policyMsg.setClosedLoopControlName(loopControlNames.get(packageName));
  89         policyMsg.setClosedLoopAlarmStart(rootAlarm.getStartEpochMicrosec());
  90         policyMsg.getAai().put("vserver.vserver-id", vmEntity.getVserverId());
  91         policyMsg.getAai().put("vserver.vserver-name", vmEntity.getVserverName());
  92         policyMsg.getAai().put("vserver.vserver-name2", vmEntity.getVserverName2());
  93         policyMsg.getAai().put("vserver.vserver-selflink", vmEntity.getVserverSelflink());
  94         policyMsg.setTarget("vserver.vserver-name");
  95         return policyMsg;
  96     }
  97
  98     private PolicyMsg getDefaultPolicyMsg(VesAlarm rootAlarm, String packageName) {
  99         PolicyMsg policyMsg = new PolicyMsg();
 100         policyMsg.setRequestID(getUniqueRequestId(rootAlarm));
 101         policyMsg.setClosedLoopControlName(loopControlNames.get(packageName));
 102         policyMsg.setClosedLoopAlarmStart(rootAlarm.getStartEpochMicrosec());
 103         policyMsg.setTarget("vserver.vserver-name");
 104         policyMsg.setTargetType("VM");
 105         policyMsg.getAai().put("vserver.vserver-name", rootAlarm.getSourceName());
 106         if (rootAlarm.getAlarmIsCleared() == PolicyMassgeConstant.POLICY_MESSAGE_ABATED) {
 107             policyMsg.setClosedLoopAlarmEnd(rootAlarm.getLastEpochMicrosec());
 108             policyMsg.setClosedLoopEventStatus(EVENT_STATUS.ABATED);
 109         }
 110         return policyMsg;
 111     }
 112
 113     private String getUniqueRequestId(VesAlarm rootAlarm) {
 114         String alarmUniqueKey = "";
 115         if (rootAlarm.getAlarmIsCleared() == PolicyMassgeConstant.POLICY_MESSAGE_ABATED) {
 116             alarmUniqueKey =
 117                     rootAlarm.getSourceId() + ":" + rootAlarm.getEventName().replace("Cleared", "");
 118         } else {
 119             alarmUniqueKey = rootAlarm.getSourceId() + ":" + rootAlarm.getEventName();
 120         }
 121         if (alarmUniqueRequestID.containsKey(alarmUniqueKey)) {
 122             return alarmUniqueRequestID.get(alarmUniqueKey);
 123         } else {
 124             String requestID = UUID.randomUUID().toString();
 125             alarmUniqueRequestID.put(alarmUniqueKey, requestID);
 126             return requestID;
 127         }
 128     }
 129
 130     private void enrichVnfInfo(VmEntity vmEntity, VesAlarm childAlarm, PolicyMsg policyMsg) {
 131         String vnfId = "";
 132         String vnfName = "";
 133         if (null != childAlarm) {
 134             vnfId = childAlarm.getSourceId();
 135             vnfName = childAlarm.getSourceName();
 136         } else {
 137             Relationship relationship = vmEntity.getRelationshipList()
 138                     .getRelationship(PolicyMassgeConstant.GENERIC_VNF);
 139             if (null != relationship) {
 140                 vnfId = relationship.getRelationshipDataValue(PolicyMassgeConstant.GENERIC_VNF_VNF_ID);
 141                 vnfName = relationship.getRelatedToPropertyValue(PolicyMassgeConstant.GENERIC_VNF_VNF_NAME);
 142             }
 143         }
 144         VnfEntity vnfEntity = getVnfEntity(vnfId, vnfName);
 145         String vserverInstatnceId = getVserverInstanceId(vnfEntity);
 146         policyMsg.getAai().put("generic-vnf.vnf-id", vnfId);
 147         policyMsg.getAai().put("generic-vnf.service-instance-id", vserverInstatnceId);
 148     }
 149
 150
 151     private String getVserverInstanceId(VnfEntity vnfEntity) {
 152         String vserverInstanceId = "";
 153         if (vnfEntity != null) {
 154             Relationship relationship = vnfEntity.getRelationshipList()
 155                     .getRelationship(PolicyMassgeConstant.SERVICE_INSTANCE);
 156             if (relationship == null) {
 157                 return vserverInstanceId;
 158             }
 159             vserverInstanceId = relationship
 160                     .getRelationshipDataValue(PolicyMassgeConstant.SERVICE_INSTANCE_ID);
 161         }
 162         return vserverInstanceId;
 163     }
 164
 165     private VnfEntity getVnfEntity(String vnfId, String vnfName) {
 166         VnfEntity vnfEntity = null;
 167         try {
 168             vnfEntity = aaiQuery.getAaiVnfData(vnfId, vnfName);
 169         } catch (CorrelationException e) {
 170             log.error("Failed to get the VNF data.", e);
 171         }
 172         return vnfEntity;
 173     }
 174
 175     private VmEntity getVmEntity(String sourceId, String sourceName) {
 176         VmEntity vmEntity = null;
 177         try {
 178             vmEntity = aaiQuery.getAaiVmData(sourceId, sourceName);
 179         } catch (CorrelationException e) {
 180             log.error("Failed to get the VM data.", e);
 181         }
 182         return vmEntity;
 183     }
 184 }