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.List;
  20 import java.util.Optional;
  21 import java.util.UUID;
  22 import java.util.concurrent.ConcurrentHashMap;
  23 import javax.inject.Inject;
  24 import lombok.extern.slf4j.Slf4j;
  25 import org.jvnet.hk2.annotations.Service;
  26 import org.onap.holmes.common.aai.AaiQuery;
  27 import org.onap.holmes.common.aai.entity.RelationshipList.Relationship;
  28 import org.onap.holmes.common.aai.entity.RelationshipList.RelationshipData;
  29 import org.onap.holmes.common.aai.entity.VmEntity;
  30 import org.onap.holmes.common.aai.entity.VnfEntity;
  31 import org.onap.holmes.common.api.stat.VesAlarm;
  32 import org.onap.holmes.common.dcae.DcaeConfigurationsCache;
  33 import org.onap.holmes.common.dmaap.entity.PolicyMsg;
  34 import org.onap.holmes.common.dmaap.entity.PolicyMsg.EVENT_STATUS;
  35 import org.onap.holmes.common.exception.CorrelationException;
  36 import org.onap.holmes.common.utils.JacksonUtil;
  37
  38 @Slf4j
  39 @Service
  40 public class DmaapService {
  41
  42     public static final int POLICY_MESSAGE_ABATED = 1;
  43
  44     @Inject
  45     private AaiQuery aaiQuery;
  46
  47     public static ConcurrentHashMap<String, String> loopControlNames = new ConcurrentHashMap<>();
  48
  49     public static ConcurrentHashMap<String, String> alarmUniqueRequestID = new ConcurrentHashMap<>();
  50
  51     public void publishPolicyMsg(PolicyMsg policyMsg, String dmaapConfigKey) {
  52         try {
  53             Publisher publisher = new Publisher();
  54             publisher.setUrl(DcaeConfigurationsCache.getPubSecInfo(dmaapConfigKey).getDmaapInfo().getTopicUrl());
  55             publisher.publish(policyMsg);
  56             log.info("send policyMsg: " + JacksonUtil.beanToJson(policyMsg));
  57         } catch (CorrelationException e) {
  58             log.error("Failed to publish policyMsg to dmaap", e.getMessage());
  59         } catch (JsonProcessingException e) {
  60             log.info("Failed to convert policyMsg to json");
  61         } catch (NullPointerException e) {
  62             log.error("DMaaP configurations does not exist!");
  63         }
  64     }
  65
  66     public PolicyMsg getPolicyMsg(VesAlarm rootAlarm, VesAlarm childAlarm, String packgeName) {
  67         return Optional.ofNullable(getVmEntity(rootAlarm.getSourceId(), rootAlarm.getSourceName()))
  68                 .map(vmEntity -> getEnrichedPolicyMsg(vmEntity, rootAlarm, childAlarm, packgeName))
  69                 .orElse(getDefaultPolicyMsg(rootAlarm.getSourceName()));
  70     }
  71
  72     private String getVserverInstanceId(VnfEntity vnfEntity) {
  73         String vserverInstanceId = "";
  74         if (vnfEntity != null) {
  75             List<Relationship> relationshipList = vnfEntity.getRelationshipList().getRelationships();
  76             Relationship relationship = null;
  77             for(int i = 0; i < relationshipList.size(); i++) {
  78                 if ("service-instance".equals(relationshipList.get(i).getRelatedTo())) {
  79                     relationship = relationshipList.get(i);
  80                     break;
  81                 }
  82             }
  83             if (relationship != null) {
  84                 List<RelationshipData> relationshipDataList = relationship.getRelationshipDataList();
  85                 for(int i = 0; i < relationshipDataList.size(); i++) {
  86                     if ("service-instance.service-instance-id"
  87                             .equals(relationshipDataList.get(i).getRelationshipKey())) {
  88                         vserverInstanceId = relationshipDataList.get(i).getRelationshipValue();
  89                         break;
  90                     }
  91                 }
  92             }
  93         }
  94         return vserverInstanceId;
  95     }
  96
  97     private PolicyMsg getEnrichedPolicyMsg(VmEntity vmEntity, VesAlarm rootAlarm, VesAlarm childAlarm,
  98             String packageName) {
  99         PolicyMsg policyMsg = new PolicyMsg();
 100         if (rootAlarm.getAlarmIsCleared() == POLICY_MESSAGE_ABATED) {
 101             policyMsg.setClosedLoopEventStatus(EVENT_STATUS.ABATED);
 102         } else {
 103             policyMsg.setClosedLoopEventStatus(EVENT_STATUS.ONSET);
 104             enrichVnfInfo(childAlarm, policyMsg);
 105         }
 106         if (alarmUniqueRequestID.containsKey(rootAlarm.getSourceId())) {
 107             policyMsg.setRequestID(alarmUniqueRequestID.get(rootAlarm.getSourceId()));
 108         } else {
 109             String requestID = UUID.randomUUID().toString();
 110             policyMsg.setRequestID(requestID);
 111             alarmUniqueRequestID.put(rootAlarm.getSourceId(), requestID);
 112         }
 113         policyMsg.setClosedLoopControlName(loopControlNames.get(packageName));
 114         policyMsg.setTarget(vmEntity.getVserverName());
 115         policyMsg.getAAI().put("vserver.in-maint", String.valueOf(vmEntity.getInMaint()));
 116         policyMsg.getAAI().put("vserver.is-closed-loop-disabled",
 117                 String.valueOf(vmEntity.getClosedLoopDisable()));
 118         policyMsg.getAAI().put("vserver.prov-status", vmEntity.getProvStatus());
 119         policyMsg.getAAI().put("vserver.resource-version", vmEntity.getResourceVersion());
 120         policyMsg.getAAI().put("vserver.vserver-id", vmEntity.getVserverId());
 121         policyMsg.getAAI().put("vserver.vserver-name", vmEntity.getVserverName());
 122         policyMsg.getAAI().put("vserver.vserver-name2", vmEntity.getVserverName2());
 123         policyMsg.getAAI().put("vserver.vserver-selflink", vmEntity.getVserverSelflink());
 124         return policyMsg;
 125     }
 126
 127     private PolicyMsg getDefaultPolicyMsg(String sourceName) {
 128         PolicyMsg policyMsg = new PolicyMsg();
 129         policyMsg.setTarget("vserver.vserver-name");
 130         policyMsg.setTargetType("VM");
 131         policyMsg.getAAI().put("vserver.vserver-name", sourceName);
 132         return policyMsg;
 133     }
 134
 135     private void enrichVnfInfo(VesAlarm childAlarm, PolicyMsg policyMsg) {
 136         VnfEntity vnfEntity = getVnfEntity(childAlarm.getSourceId(), childAlarm.getSourceName());
 137         String vserverInstatnceId = getVserverInstanceId(vnfEntity);
 138         policyMsg.getAAI().put("generic-vnf.vnf-id", childAlarm.getSourceId());
 139         policyMsg.getAAI().put("generic-vnf.vnf-name", childAlarm.getSourceName());
 140         policyMsg.getAAI().put("generic-vnf.service-instance-id", vserverInstatnceId);
 141     }
 142
 143     private VnfEntity getVnfEntity(String vnfId, String vnfName) {
 144         VnfEntity vnfEntity = null;
 145         try {
 146             vnfEntity = aaiQuery.getAaiVnfData(vnfId, vnfName);
 147         } catch (CorrelationException e) {
 148             log.error("Failed to get vnf data", e.getMessage());
 149         }
 150         return vnfEntity;
 151     }
 152
 153     private VmEntity getVmEntity(String sourceId, String sourceName) {
 154         VmEntity vmEntity = null;
 155         try {
 156             vmEntity = aaiQuery.getAaiVmData(sourceId, sourceName);
 157         } catch (CorrelationException e) {
 158             log.error("Failed to get vm data", e.getMessage());
 159         }
 160         return vmEntity;
 161     }
 162 }