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 javax.inject.Inject;
22 import lombok.AllArgsConstructor;
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;
37 public class DmaapService {
40 private AaiQuery aaiQuery;
42 private Publisher publisher;
44 public void publishPolicyMsg(PolicyMsg policyMsg) {
46 publisher.publish(policyMsg);
47 log.info("send policyMsg: " + JacksonUtil.beanToJson(policyMsg));
48 } catch (CorrelationException e) {
49 log.error("Failed to publish policyMsg to dmaap", e.getMessage());
50 } catch (JsonProcessingException e) {
51 log.info("Failed to convert policyMsg to json");
55 public PolicyMsg getPolicyMsg(VesAlarm vesAlarm) {
56 return Optional.ofNullable(getVmEntity(vesAlarm.getSourceId(), vesAlarm.getSourceName()))
57 .map(vmEntity -> getEnrichedPolicyMsg(vmEntity, vesAlarm))
58 .orElse(getDefaultPolicyMsg(vesAlarm.getSourceName()));
61 private String getVserverInstanceId(VnfEntity vnfEntity) {
62 String vserverInstanceId = "";
63 if (vnfEntity != null) {
64 List<RelationshipData> relationshipDataList = vnfEntity.getRelationshipList()
65 .getRelationships().stream()
66 .filter(relationship -> relationship.getRelatedTo().equals("service-instance"))
67 .limit(1).findFirst().get().getRelationshipDataList();
69 vserverInstanceId = relationshipDataList.stream()
70 .filter(relationshipData -> relationshipData.getRelationshipKey().equals("service-instance.service-instance-id"))
72 .map(relationshipData -> relationshipData.getRelationshipValue()).get();
74 return vserverInstanceId;
77 private VnfEntity getVnfEntity(String vnfId, String vnfName) {
78 VnfEntity vnfEntity = null;
80 vnfEntity = aaiQuery.getAaiVnfData(vnfId, vnfName);
81 } catch (CorrelationException e) {
82 log.error("Failed to get vnf data", e.getMessage());
87 private VmEntity getVmEntity(String sourceId, String sourceName) {
88 VmEntity vmEntity = null;
90 vmEntity = aaiQuery.getAaiVmData(sourceId, sourceName);
91 } catch (CorrelationException e) {
92 log.error("Failed to get vm data", e.getMessage());
97 private PolicyMsg getEnrichedPolicyMsg(VmEntity vmEntity, VesAlarm vesAlarm) {
98 VnfEntity vnfEntity = getVnfEntity(vesAlarm.getEventId(), vesAlarm.getEventName());
99 String vserverInstatnceId = getVserverInstanceId(vnfEntity);
100 PolicyMsg policyMsg = new PolicyMsg();
101 policyMsg.setClosedLoopEventClient("DCAE_INSTANCE_ID.dcae-tca");
102 policyMsg.setPolicyVersion("1.0.0.5");
103 policyMsg.setPolicyName("vLoadBalancer");
104 policyMsg.setPolicyScope(
105 "resource=SampleResource,service=SampleService,type=SampleType,closedLoopControlName=SampleClosedLoop");
106 policyMsg.setTargetType("VM");
107 policyMsg.setClosedLoopAlarmStart(1484855);
108 if (vesAlarm.getAlarmIsCleared() == 1) {
109 policyMsg.setClosedLoopEventStatus(EVENT_STATUS.ABATED);
111 policyMsg.setClosedLoopEventStatus(EVENT_STATUS.ONSET);
113 policyMsg.setClosedLoopControlName(
114 "CL-LB-LOW-TRAFFIC-SIG-d925ed73-8231-4d02-9545-db4e101f88f8");
115 policyMsg.setVersion("1.0.2");
116 policyMsg.setTarget("generic-vnf.vnf-id");
117 policyMsg.setRequestID("8c1b8bd8-06f7-493f-8ed7-daaa4cc481bc");
118 policyMsg.setFrom("DCAE");
119 policyMsg.getAai().put("vserver.in-maint", String.valueOf(vmEntity.getInMaint()));
120 policyMsg.getAai().put("vserver.is-closed-loop-disabled",
121 String.valueOf(vmEntity.getClosedLoopDisable()));
122 policyMsg.getAai().put("vserver.prov-status", vmEntity.getProvStatus());
123 policyMsg.getAai().put("vserver.resource-version", vmEntity.getResourceVersion());
124 policyMsg.getAai().put("vserver.vserver-id", vmEntity.getVserverId());
125 policyMsg.getAai().put("vserver.vserver-name", vmEntity.getVserverName());
126 policyMsg.getAai().put("vserver.vserver-name2", vmEntity.getVserverName2());
127 policyMsg.getAai().put("vserver.vserver-selflink", vmEntity.getVserverSelflink());
128 policyMsg.getAai().put("generic-vnf.vnf-id", vesAlarm.getEventId());
129 policyMsg.getAai().put("generic-vnf.vnf-name", vesAlarm.getEventName());
130 policyMsg.getAai().put("generic-vnf.service-instance-id", vserverInstatnceId);
134 private PolicyMsg getDefaultPolicyMsg(String sourceName) {
135 PolicyMsg policyMsg = new PolicyMsg();
136 policyMsg.setTarget("vserver.vserver-name");
137 policyMsg.setTargetType("VM");
138 policyMsg.getAai().put("vserver.vserver-name", sourceName);