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.getSourceName()));
  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, policyMsg));
  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             policyMsg.getAai().put("vserver.is-closed-loop-disabled",
  76                     String.valueOf(vmEntity.getClosedLoopDisable()));
  77             policyMsg.getAai().put("vserver.prov-status", vmEntity.getProvStatus());
  78             policyMsg.getAai().put("vserver.resource-version", vmEntity.getResourceVersion());
  79         } else {
  80             policyMsg.setClosedLoopAlarmEnd(rootAlarm.getLastEpochMicrosec());
  81             policyMsg.setClosedLoopEventStatus(EVENT_STATUS.ABATED);
  82         }
  83         policyMsg.setClosedLoopControlName(loopControlNames.get(packageName));
  84         policyMsg.setClosedLoopAlarmStart(rootAlarm.getStartEpochMicrosec());
  85         policyMsg.getAai().put("vserver.vserver-id", vmEntity.getVserverId());
  86         policyMsg.getAai().put("vserver.vserver-name", vmEntity.getVserverName());
  87         policyMsg.getAai().put("vserver.vserver-name2", vmEntity.getVserverName2());
  88         policyMsg.getAai().put("vserver.vserver-selflink", vmEntity.getVserverSelflink());
  89         policyMsg.setTarget("vserver.vserver-name");
  90         return policyMsg;
  91     }
  92
  93     private PolicyMsg getDefaultPolicyMsg(String sourceName) {
  94         PolicyMsg policyMsg = new PolicyMsg();
  95         policyMsg.setTarget("vserver.vserver-id");
  96         policyMsg.setTargetType("VM");
  97         policyMsg.getAai().put("vserver.vserver-name", sourceName);
  98         return policyMsg;
  99     }
 100
 101     private String getUniqueRequestId(VesAlarm rootAlarm, PolicyMsg policyMsg) {
 102         String alarmUniqueKey = "";
 103         if (rootAlarm.getAlarmIsCleared() == PolicyMassgeConstant.POLICY_MESSAGE_ABATED) {
 104             alarmUniqueKey =
 105                     rootAlarm.getSourceId() + ":" + rootAlarm.getEventName().replace("Cleared", "");
 106         } else {
 107             alarmUniqueKey = rootAlarm.getSourceId() + ":" + rootAlarm.getEventName();
 108         }
 109         if (alarmUniqueRequestID.containsKey(alarmUniqueKey)) {
 110             return alarmUniqueRequestID.get(alarmUniqueKey);
 111         } else {
 112             String requestID = UUID.randomUUID().toString();
 113             alarmUniqueRequestID.put(alarmUniqueKey, requestID);
 114             return requestID;
 115         }
 116     }
 117
 118     private void enrichVnfInfo(VmEntity vmEntity, VesAlarm childAlarm, PolicyMsg policyMsg) {
 119         String vnfId = "";
 120         String vnfName = "";
 121         if (null != childAlarm) {
 122             vnfId = childAlarm.getSourceId();
 123             vnfName = childAlarm.getSourceName();
 124         } else {
 125             Relationship relationship = vmEntity.getRelationshipList()
 126                     .getRelationship(PolicyMassgeConstant.GENERIC_VNF);
 127             if (null != relationship) {
 128                 vnfId = relationship.getRelationshipDataValue(PolicyMassgeConstant.GENERIC_VNF_VNF_ID);
 129                 vnfName = relationship.getRelatedToPropertyValue(PolicyMassgeConstant.GENERIC_VNF_VNF_NAME);
 130             }
 131         }
 132         VnfEntity vnfEntity = getVnfEntity(vnfId, vnfName);
 133         String vserverInstatnceId = getVserverInstanceId(vnfEntity);
 134         policyMsg.getAai().put("generic-vnf.vnf-id", vnfId);
 135         policyMsg.getAai().put("generic-vnf.service-instance-id", vserverInstatnceId);
 136     }
 137
 138
 139     private String getVserverInstanceId(VnfEntity vnfEntity) {
 140         String vserverInstanceId = "";
 141         if (vnfEntity != null) {
 142             Relationship relationship = vnfEntity.getRelationshipList()
 143                     .getRelationship(PolicyMassgeConstant.SERVICE_INSTANCE);
 144             if (relationship == null) {
 145                 return vserverInstanceId;
 146             }
 147             vserverInstanceId = relationship
 148                     .getRelationshipDataValue(PolicyMassgeConstant.SERVICE_INSTANCE_ID);
 149         }
 150         return vserverInstanceId;
 151     }
 152
 153     private VnfEntity getVnfEntity(String vnfId, String vnfName) {
 154         VnfEntity vnfEntity = null;
 155         try {
 156             vnfEntity = aaiQuery.getAaiVnfData(vnfId, vnfName);
 157         } catch (CorrelationException e) {
 158             log.error("Failed to get the VNF data.", e);
 159         }
 160         return vnfEntity;
 161     }
 162
 163     private VmEntity getVmEntity(String sourceId, String sourceName) {
 164         VmEntity vmEntity = null;
 165         try {
 166             vmEntity = aaiQuery.getAaiVmData(sourceId, sourceName);
 167         } catch (CorrelationException e) {
 168             log.error("Failed to get the VM data.", e);
 169         }
 170         return vmEntity;
 171     }
 172 }