2 * Copyright 2017 ZTE Corporation.
4 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
5 * in compliance with the License. You may obtain a copy of the License at
7 * http://www.apache.org/licenses/LICENSE-2.0
9 * Unless required by applicable law or agreed to in writing, software distributed under the License
10 * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
11 * or implied. See the License for the specific language governing permissions and limitations under
14 package org.onap.holmes.common.aai;
16 import java.util.HashMap;
18 import javax.inject.Inject;
19 import lombok.extern.slf4j.Slf4j;
20 import org.jvnet.hk2.annotations.Service;
21 import org.onap.holmes.common.aai.config.AaiConfig;
22 import org.onap.holmes.common.aai.entity.VmEntity;
23 import org.onap.holmes.common.aai.entity.VnfEntity;
24 import org.onap.holmes.common.config.MicroServiceConfig;
25 import org.onap.holmes.common.exception.CorrelationException;
26 import org.onap.holmes.common.utils.HttpsUtils;
30 public class AaiQuery {
33 private AaiResponseUtil aaiResponseUtil;
35 public VnfEntity getAaiVnfData(String vnfId, String vnfName) throws CorrelationException {
36 String response = getVnfDataResponse(vnfId, vnfName);
38 return aaiResponseUtil.convertJsonToVnfEntity(response);
39 } catch (Exception e) {
40 throw new CorrelationException("Failed to convert aai vnf response data to vnf entity", e);
44 public VmEntity getAaiVmData(String vserverId, String vserverName) throws CorrelationException {
45 String url = getVmUrl(vserverId, vserverName);
46 String response = getResponse(url);
48 return aaiResponseUtil.convertJsonToVmEntity(response);
49 } catch (Exception e) {
50 throw new CorrelationException("Failed to convert aai vm response data to vm entity", e);
54 private String getVmUrl(String vserverId, String vserverName) throws CorrelationException {
56 String resourceLinkUrl = getVmResourceLinks(vserverId, vserverName);
57 String baseUrl = getBaseUrl("");
58 if (baseUrl.startsWith("http")) {
59 url = baseUrl + getMsbSuffixAddr(resourceLinkUrl);
61 url = baseUrl + resourceLinkUrl;
66 private String getVmResourceLinks(String vserverId, String vserverName) throws CorrelationException {
67 String response = getResourceLinksResponse(vserverId, vserverName);
69 return aaiResponseUtil.convertJsonToVmResourceLink(response).get(0).getResourceLink();
70 } catch (Exception e) {
71 throw new CorrelationException("Failed to get aai resource link", e);
75 private String getResourceLinksResponse(String vserverId, String vserverName) throws CorrelationException {
76 String url = getBaseUrl(getMsbSuffixAddr(AaiConfig.AAI_VM_ADDR) + "vserver-id:EQUALS:" + vserverId);
77 String response = getResponse(url);
78 if ("".equals(response) || "{}".equals(response)) {
79 url = getBaseUrl(AaiConfig.AAI_VM_ADDR + "vserver-name:EQUALS:" + vserverName);
80 response = getResponse(url);
85 private String getVnfDataResponse(String vnfId, String vnfName) throws CorrelationException {
86 String url = getBaseUrl(getMsbSuffixAddr(AaiConfig.AAI_VNF_ADDR)+ "vnf-id=" + vnfId);
87 String response = getResponse(url);
88 if ("".equals(response) || "{}".equals(response)) {
89 url = getBaseUrl(AaiConfig.AAI_VNF_ADDR + "vnf-name=" + vnfName);
90 response = getResponse(url);
95 private String getBaseUrl(String suffixUrl) {
98 url = MicroServiceConfig.getMsbServerAddr() + suffixUrl;
99 } catch (Exception e) {
100 log.info("Failed to get msb address");
102 if (url.equals("")) {
104 url = "https:\\\\" + MicroServiceConfig.getServiceAddrInfoFromCBS("aai_config")
106 } catch (Exception e) {
107 log.info("Failed to get aai address");
113 private String getMsbSuffixAddr(String suffixUrl) {
114 String[] addrSplits = suffixUrl.substring(1).split("/");
115 String ret = addrSplits[1];
116 addrSplits[1] = addrSplits[0] + "-" + addrSplits[2];
118 StringBuffer stringBuffer = new StringBuffer();
119 for (String split : addrSplits) {
120 stringBuffer.append("/" + split);
122 return stringBuffer.toString();
125 private String getResponse(String url) throws CorrelationException {
126 String response = "";
128 response = HttpsUtils.get(url, getHeaders());
129 } catch (Exception e) {
130 throw new CorrelationException("Failed to get data from aai", e);
135 private Map getHeaders() {
136 Map<String, String> headers = new HashMap<>();
137 headers.put("X-TransactionId", AaiConfig.X_TRANSACTION_ID);
138 headers.put("X-FromAppId", AaiConfig.X_FROMAPP_ID);
139 headers.put("Authorization", AaiConfig.getAuthenticationCredentials());
140 headers.put("Accept", "application/json");