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