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.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.RelationshipData;
  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.dmaap.entity.PolicyMsg;
  31 import org.onap.holmes.common.dmaap.entity.PolicyMsg.EVENT_STATUS;
  32 import org.onap.holmes.common.exception.CorrelationException;
  33 import org.onap.holmes.common.utils.JacksonUtil;
  34
  35 @Slf4j
  36 @Service
  37 public class DmaapService {
  38
  39     public static final int POLICY_MESSAGE_ABATED = 1;
  40     @Inject
  41     private AaiQuery aaiQuery;
  42     @Inject
  43     private Publisher publisher;
  44
  45     public static ConcurrentHashMap<String, String> loopControlNames = new ConcurrentHashMap<>();
  46
  47     public void publishPolicyMsg(PolicyMsg policyMsg) {
  48         try {
  49             publisher.publish(policyMsg);
  50             log.info("send policyMsg: " + JacksonUtil.beanToJson(policyMsg));
  51         } catch (CorrelationException e) {
  52             log.error("Failed to publish policyMsg to dmaap", e.getMessage());
  53         } catch (JsonProcessingException e) {
  54             log.info("Failed to convert policyMsg to json");
  55         }
  56     }
  57
  58     public PolicyMsg getPolicyMsg(VesAlarm vesAlarm, String packgeName) {
  59         return Optional.ofNullable(getVmEntity(vesAlarm.getSourceId(), vesAlarm.getSourceName()))
  60                 .map(vmEntity -> getEnrichedPolicyMsg(vmEntity, vesAlarm, packgeName))
  61                 .orElse(getDefaultPolicyMsg(vesAlarm.getSourceName()));
  62     }
  63
  64     private String getVserverInstanceId(VnfEntity vnfEntity) {
  65         String vserverInstanceId = "";
  66         if (vnfEntity != null) {
  67             List<RelationshipData> relationshipDataList = vnfEntity.getRelationshipList()
  68                     .getRelationships().stream()
  69                     .filter(relationship -> relationship.getRelatedTo().equals("service-instance"))
  70                     .limit(1).findFirst().get().getRelationshipDataList();
  71
  72             vserverInstanceId = relationshipDataList.stream()
  73                     .filter(relationshipData -> relationshipData.getRelationshipKey().equals("service-instance.service-instance-id"))
  74                     .findFirst()
  75                     .map(relationshipData -> relationshipData.getRelationshipValue()).get();
  76         }
  77         return vserverInstanceId;
  78     }
  79
  80     private VnfEntity getVnfEntity(String vnfId, String vnfName) {
  81         VnfEntity vnfEntity = null;
  82         try {
  83             vnfEntity = aaiQuery.getAaiVnfData(vnfId, vnfName);
  84         } catch (CorrelationException e) {
  85             log.error("Failed to get vnf data", e.getMessage());
  86         }
  87         return vnfEntity;
  88     }
  89
  90     private VmEntity getVmEntity(String sourceId, String sourceName) {
  91         VmEntity vmEntity = null;
  92         try {
  93             vmEntity = aaiQuery.getAaiVmData(sourceId, sourceName);
  94         } catch (CorrelationException e) {
  95             log.error("Failed to get vm data", e.getMessage());
  96         }
  97         return vmEntity;
  98     }
  99
 100     private PolicyMsg getEnrichedPolicyMsg(VmEntity vmEntity, VesAlarm vesAlarm, String packageName) {
 101         VnfEntity vnfEntity = getVnfEntity(vesAlarm.getEventId(), vesAlarm.getEventName());
 102         String vserverInstatnceId = getVserverInstanceId(vnfEntity);
 103         PolicyMsg policyMsg = new PolicyMsg();
 104         if (vesAlarm.getAlarmIsCleared() == POLICY_MESSAGE_ABATED) {
 105             policyMsg.setClosedLoopEventStatus(EVENT_STATUS.ABATED);
 106         } else {
 107             policyMsg.setClosedLoopEventStatus(EVENT_STATUS.ONSET);
 108         }
 109         policyMsg.setClosedLoopControlName(loopControlNames.get(packageName));
 110         policyMsg.getAai().put("vserver.in-maint", String.valueOf(vmEntity.getInMaint()));
 111         policyMsg.getAai().put("vserver.is-closed-loop-disabled",
 112                 String.valueOf(vmEntity.getClosedLoopDisable()));
 113         policyMsg.getAai().put("vserver.prov-status", vmEntity.getProvStatus());
 114         policyMsg.getAai().put("vserver.resource-version", vmEntity.getResourceVersion());
 115         policyMsg.getAai().put("vserver.vserver-id", vmEntity.getVserverId());
 116         policyMsg.getAai().put("vserver.vserver-name", vmEntity.getVserverName());
 117         policyMsg.getAai().put("vserver.vserver-name2", vmEntity.getVserverName2());
 118         policyMsg.getAai().put("vserver.vserver-selflink", vmEntity.getVserverSelflink());
 119         policyMsg.getAai().put("generic-vnf.vnf-id", vesAlarm.getEventId());
 120         policyMsg.getAai().put("generic-vnf.vnf-name", vesAlarm.getEventName());
 121         policyMsg.getAai().put("generic-vnf.service-instance-id", vserverInstatnceId);
 122         return policyMsg;
 123     }
 124
 125     private PolicyMsg getDefaultPolicyMsg(String sourceName) {
 126         PolicyMsg policyMsg = new PolicyMsg();
 127         policyMsg.setTarget("vserver.vserver-name");
 128         policyMsg.setTargetType("VM");
 129         policyMsg.getAai().put("vserver.vserver-name", sourceName);
 130         return policyMsg;
 131     }
 132 }