2 * Copyright 2017 ZTE Corporation.
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
8 * http://www.apache.org/licenses/LICENSE-2.0
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.
16 package org.onap.holmes.common.dmaap;
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;
40 public class DmaapService {
41 public static final int POLICY_MESSAGE_ABATED = 1;
42 public static final String SERVICE_INSTANCE = "service-instance";
43 public static final String SERVICE_INSTANCE_ID = "service-instance.service-instance-id";
45 private AaiQuery aaiQuery;
46 public static ConcurrentHashMap<String, String> loopControlNames = new ConcurrentHashMap<>();
47 public static ConcurrentHashMap<String, String> alarmUniqueRequestID = new ConcurrentHashMap<>();
49 public void publishPolicyMsg(PolicyMsg policyMsg, String dmaapConfigKey) {
51 Publisher publisher = new Publisher();
52 publisher.setUrl(DcaeConfigurationsCache.getPubSecInfo(dmaapConfigKey).getDmaapInfo().getTopicUrl());
53 publisher.publish(policyMsg);
54 log.info("send policyMsg: " + JacksonUtil.beanToJson(policyMsg));
55 } catch (CorrelationException e) {
56 log.error("Failed to publish policyMsg to dmaap", e.getMessage());
57 } catch (JsonProcessingException e) {
58 log.info("Failed to convert policyMsg to json");
59 } catch (NullPointerException e) {
60 log.error("DMaaP configurations does not exist!");
64 public PolicyMsg getPolicyMsg(VesAlarm rootAlarm, VesAlarm childAlarm, String packgeName) {
65 return Optional.ofNullable(getVmEntity(rootAlarm.getSourceId(), rootAlarm.getSourceName()))
66 .map(vmEntity -> getEnrichedPolicyMsg(vmEntity, rootAlarm, childAlarm, packgeName))
67 .orElse(getDefaultPolicyMsg(rootAlarm.getSourceName()));
70 private String getVserverInstanceId(VnfEntity vnfEntity) {
71 String vserverInstanceId = "";
72 if (vnfEntity != null) {
73 List<Relationship> relationshipList = vnfEntity.getRelationshipList().getRelationships();
74 Relationship relationship = null;
75 for(int i = 0; i < relationshipList.size(); i++) {
76 if (SERVICE_INSTANCE.equals(relationshipList.get(i).getRelatedTo())) {
77 relationship = relationshipList.get(i);
81 if (relationship != null) {
82 List<RelationshipData> relationshipDataList = relationship.getRelationshipDataList();
83 for(int i = 0; i < relationshipDataList.size(); i++) {
84 if (SERVICE_INSTANCE_ID
85 .equals(relationshipDataList.get(i).getRelationshipKey())) {
86 vserverInstanceId = relationshipDataList.get(i).getRelationshipValue();
92 return vserverInstanceId;
95 private PolicyMsg getEnrichedPolicyMsg(VmEntity vmEntity, VesAlarm rootAlarm, VesAlarm childAlarm,
97 PolicyMsg policyMsg = new PolicyMsg();
98 String alarmUniqueKey = "";
99 if (rootAlarm.getAlarmIsCleared() == POLICY_MESSAGE_ABATED) {
100 policyMsg.setClosedLoopEventStatus(EVENT_STATUS.ABATED);
102 rootAlarm.getSourceId() + ":" + rootAlarm.getEventName().replace("Cleared", "");
104 policyMsg.setClosedLoopEventStatus(EVENT_STATUS.ONSET);
105 enrichVnfInfo(childAlarm, policyMsg);
106 alarmUniqueKey = rootAlarm.getSourceId() + ":" + rootAlarm.getEventName();
108 if (alarmUniqueRequestID.containsKey(alarmUniqueKey)) {
109 policyMsg.setRequestID(alarmUniqueRequestID.get(alarmUniqueKey));
111 String requestID = UUID.randomUUID().toString();
112 policyMsg.setRequestID(requestID);
113 alarmUniqueRequestID.put(alarmUniqueKey, requestID);
115 policyMsg.setClosedLoopControlName(loopControlNames.get(packageName));
116 policyMsg.setTarget(vmEntity.getVserverName());
117 policyMsg.getAai().put("vserver.in-maint", String.valueOf(vmEntity.getInMaint()));
118 policyMsg.getAai().put("vserver.is-closed-loop-disabled",
119 String.valueOf(vmEntity.getClosedLoopDisable()));
120 policyMsg.getAai().put("vserver.prov-status", vmEntity.getProvStatus());
121 policyMsg.getAai().put("vserver.resource-version", vmEntity.getResourceVersion());
122 policyMsg.getAai().put("vserver.vserver-id", vmEntity.getVserverId());
123 policyMsg.getAai().put("vserver.vserver-name", vmEntity.getVserverName());
124 policyMsg.getAai().put("vserver.vserver-name2", vmEntity.getVserverName2());
125 policyMsg.getAai().put("vserver.vserver-selflink", vmEntity.getVserverSelflink());
129 private PolicyMsg getDefaultPolicyMsg(String sourceName) {
130 PolicyMsg policyMsg = new PolicyMsg();
131 policyMsg.setTarget("vserver.vserver-name");
132 policyMsg.setTargetType("VM");
133 policyMsg.getAai().put("vserver.vserver-name", sourceName);
137 private void enrichVnfInfo(VesAlarm childAlarm, PolicyMsg policyMsg) {
138 VnfEntity vnfEntity = getVnfEntity(childAlarm.getSourceId(), childAlarm.getSourceName());
139 String vserverInstatnceId = getVserverInstanceId(vnfEntity);
140 policyMsg.getAai().put("generic-vnf.vnf-id", childAlarm.getSourceId());
141 policyMsg.getAai().put("generic-vnf.vnf-name", childAlarm.getSourceName());
142 policyMsg.getAai().put("generic-vnf.service-instance-id", vserverInstatnceId);
145 private VnfEntity getVnfEntity(String vnfId, String vnfName) {
146 VnfEntity vnfEntity = null;
148 vnfEntity = aaiQuery.getAaiVnfData(vnfId, vnfName);
149 } catch (CorrelationException e) {
150 log.error("Failed to get vnf data", e.getMessage());
155 private VmEntity getVmEntity(String sourceId, String sourceName) {
156 VmEntity vmEntity = null;
158 vmEntity = aaiQuery.getAaiVmData(sourceId, sourceName);
159 } catch (CorrelationException e) {
160 log.error("Failed to get vm data", e.getMessage());