--- /dev/null
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
+ * in compliance with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under the License
+ * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
+ * or implied. See the License for the specific language governing permissions and limitations under
+ * the License.
+ */
+package org.onap.holmes.common.aai;
+
+import javax.ws.rs.client.Client;
+import javax.ws.rs.client.ClientBuilder;
+import javax.ws.rs.client.WebTarget;
+import javax.ws.rs.core.MultivaluedHashMap;
+import org.glassfish.jersey.client.ClientConfig;
+import org.onap.holmes.common.aai.config.AaiConfig;
+import org.onap.holmes.common.aai.entity.VmEntity;
+import org.onap.holmes.common.aai.entity.VnfEntity;
+import org.onap.holmes.common.config.MicroServiceConfig;
+import org.onap.holmes.common.exception.CorrelationException;
+
+public class AaiQuery {
+
+ private AaiResponseUtil aaiResponseUtil;
+
+ public VnfEntity getAaiVnfData(String vnfId, String vnfName) throws CorrelationException {
+ Client client = ClientBuilder.newClient(new ClientConfig());
+ WebTarget webTarget = client
+ .target(MicroServiceConfig.getMsbServerAddr() + AaiConfig.VNF_ADDR + "vnf-id="
+ + vnfId);
+ String response = webTarget.request("application/json").headers(getHeaders()).get()
+ .readEntity(String.class);
+ if (response == null) {
+ webTarget = client
+ .target(MicroServiceConfig.getMsbServerAddr() + AaiConfig.VNF_ADDR + "vnf-name="
+ + vnfName);
+ response = webTarget.request("application/json").headers(getHeaders()).get()
+ .readEntity(String.class);
+ }
+ try {
+ return aaiResponseUtil.convertJsonToVnfEntity(response);
+ } catch (Exception e) {
+ throw new CorrelationException("Failed to convert aai vnf response data to vnf entity", e);
+ }
+ }
+
+ public VmEntity getAaiVmData(String vserverId, String vserverName) throws CorrelationException {
+ Client client = ClientBuilder.newClient(new ClientConfig());
+ String response = client
+ .target(MicroServiceConfig.getMsbServerAddr() + getVmResourceLinks(client,
+ vserverId, vserverName)).request("application/json").headers(getHeaders())
+ .get().readEntity(String.class);
+ try {
+ return aaiResponseUtil.convertJsonToVmEntity(response);
+ } catch (Exception e) {
+ throw new CorrelationException("Failed to convert aai vm response data to vm entity", e);
+ }
+ }
+
+ private String getVmResourceLinks(Client client, String vserverId, String vserverName) throws CorrelationException {
+ WebTarget webTarget = client
+ .target(MicroServiceConfig.getMsbServerAddr() + AaiConfig.VM_ADDR
+ + "vserver-id:EQUALS:" + vserverId);
+ String response = webTarget.request("application/json").headers(getHeaders()).get()
+ .readEntity(String.class);
+ if (response == null) {
+ webTarget = client.target(MicroServiceConfig.getMsbServerAddr() + AaiConfig.VM_ADDR
+ + "vserver-name:EQUALS:" + vserverName);
+ response = webTarget.request("application/json").headers(getHeaders()).get()
+ .readEntity(String.class);
+ }
+ try {
+ return aaiResponseUtil.convertJsonToVmResourceLink(response).get(0).getResourceLink();
+ } catch (Exception e) {
+ throw new CorrelationException("Failed to get aai resource link", e);
+ }
+ }
+
+ private MultivaluedHashMap getHeaders() {
+ MultivaluedHashMap<String, String> headers = new MultivaluedHashMap<>();
+ headers.add("X-TransactionId", AaiConfig.X_TRANSACTION_ID);
+ headers.add("X-FromAppId", AaiConfig.X_FROMAPP_ID);
+ headers.add("Authorization", AaiConfig.getAuthenticationCredentials());
+ return headers;
+ }
+}
--- /dev/null
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.onap.holmes.common.aai;
+
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+import org.jvnet.hk2.annotations.Service;
+import org.onap.holmes.common.aai.entity.RelationshipList;
+import org.onap.holmes.common.aai.entity.RelationshipList.RelatedToProperty;
+import org.onap.holmes.common.aai.entity.RelationshipList.Relationship;
+import org.onap.holmes.common.aai.entity.RelationshipList.RelationshipData;
+import org.onap.holmes.common.aai.entity.VmEntity;
+import org.onap.holmes.common.aai.entity.VmResourceLink;
+import org.onap.holmes.common.aai.entity.VnfEntity;
+
+public class AaiResponseUtil {
+
+ public List<VmResourceLink> convertJsonToVmResourceLink(String responseJson) throws IOException {
+ ObjectMapper mapper = new ObjectMapper();
+ JsonNode jsonNode = mapper.readTree(responseJson);
+
+ List<VmResourceLink> vmResourceLinkList = new ArrayList<>();
+ if (jsonNode.has("result-data")) {
+ JsonNode resultData = jsonNode.get("result-data");
+ vmResourceLinkList = convertResultDataList(resultData);
+ }
+ return vmResourceLinkList;
+ }
+
+ public VmEntity convertJsonToVmEntity(String responseJson) throws IOException {
+ ObjectMapper mapper = new ObjectMapper();
+ JsonNode jsonNode = mapper.readTree(responseJson);
+ if (!jsonNode.iterator().hasNext()) {
+ return null;
+ }
+ VmEntity vmEntity = new VmEntity();
+ vmEntity.setInMaint(getBooleanElementByNode(jsonNode, "in-maint"));
+ vmEntity.setClosedLoopDisable(getBooleanElementByNode(jsonNode,"is-closed-loop-disabled"));
+ vmEntity.setProvStatus(getTextElementByNode(jsonNode, "prov-status"));
+ vmEntity.setResourceVersion(getTextElementByNode(jsonNode,"resource-version"));
+ vmEntity.setVserverId(getTextElementByNode(jsonNode,"vserver-id"));
+ vmEntity.setVserverName(getTextElementByNode(jsonNode,"vserver-name"));
+ vmEntity.setVserverName2(getTextElementByNode(jsonNode,"vserver-name2"));
+ vmEntity.setVserverSelflink(getTextElementByNode(jsonNode,"vserver-selflink"));
+
+ if (jsonNode.has("relationship-list")) {
+ JsonNode relationshipListNode = jsonNode.get("relationship-list");
+ if (relationshipListNode.has("relationship")) {
+ JsonNode relationshipNode = relationshipListNode.get("relationship");
+ vmEntity.getRelationshipList().setRelationships(convertRelationships(relationshipNode));
+ }
+ }
+ if (vmEntity.getRelationshipList().getRelationships() == null) {
+ vmEntity.getRelationshipList().setRelationships(Collections.emptyList());
+ }
+ return vmEntity;
+ }
+
+ public VnfEntity convertJsonToVnfEntity(String responseJson) throws IOException {
+ ObjectMapper mapper = new ObjectMapper();
+ JsonNode jsonNode = mapper.readTree(responseJson);
+
+ if(!jsonNode.elements().hasNext())
+ return null;
+
+ VnfEntity vnfEntity = new VnfEntity();
+ vnfEntity.setInMaint(getBooleanElementByNode(jsonNode, "in-maint"));
+ vnfEntity.setClosedLoopDisabled(getBooleanElementByNode(jsonNode, "is-closed-loop-disabled"));
+ vnfEntity.setOrchestrationStatus(getTextElementByNode(jsonNode, "orchestration-status"));
+ vnfEntity.setProvStatus(getTextElementByNode(jsonNode, "prov-status"));
+ vnfEntity.setResourceVersion(getTextElementByNode(jsonNode,"resource-version"));
+ vnfEntity.setServiceId(getTextElementByNode(jsonNode,"service-id"));
+ vnfEntity.setVnfId(getTextElementByNode(jsonNode,"vnf-id"));
+ vnfEntity.setVnfName(getTextElementByNode(jsonNode,"vnf-name"));
+ vnfEntity.setVnfType(getTextElementByNode(jsonNode,"vnf-type"));
+
+ if (jsonNode.has("relationship-list")) {
+ JsonNode relationshipListNode = jsonNode.get("relationship-list");
+ if (relationshipListNode.has("relationship")) {
+ JsonNode relationshipNode = relationshipListNode.get("relationship");
+ vnfEntity.getRelationshipList().setRelationships(convertRelationships(relationshipNode));
+ }
+ }
+ if (vnfEntity.getRelationshipList().getRelationships() == null) {
+ vnfEntity.getRelationshipList().setRelationships(Collections.emptyList());
+ }
+ return vnfEntity;
+ }
+
+ private List<VmResourceLink> convertResultDataList(JsonNode resultData) {
+ List<VmResourceLink> vmResourceLinkList = new ArrayList<>();
+ if (resultData.isArray()) {
+ resultData.forEach(node ->{
+ if (node.has("resource-link") && node.has("resource-type")) {
+ VmResourceLink vmResourceLink = new VmResourceLink();
+ vmResourceLink.setResourceLink(getTextElementByNode(node, "resource-link"));
+ vmResourceLink.setResourceType(getTextElementByNode(node, "resource-type"));
+ vmResourceLinkList.add(vmResourceLink);
+ }
+ });
+ }
+ return vmResourceLinkList;
+ }
+
+ private List<Relationship> convertRelationships(JsonNode relationshipNode) {
+ List<Relationship> relationshipList = new ArrayList<>();
+ if (relationshipNode.isArray()) {
+ relationshipNode.forEach(node ->{
+ Relationship relationship = new Relationship();
+ relationship.setRelatedLink(getTextElementByNode(node, "related-link"));
+ relationship.setRelatedTo(getTextElementByNode(node, "related-to"));
+ if (node.has("related-to-property")) {
+ JsonNode relatedToPropertyNode = node.get("related-to-property");
+ relationship.setRelatedToPropertyList(
+ convertRelatedToProperty(relatedToPropertyNode));
+ } else {
+ relationship.setRelatedToPropertyList(Collections.emptyList());
+ }
+ if (node.has("relationship-data")) {
+ JsonNode relationshipDataNode = node.get("relationship-data");
+ relationship
+ .setRelationshipDataList(convertRelationshipDate(relationshipDataNode));
+ } else {
+ relationship.setRelationshipDataList(Collections.emptyList());
+ }
+ relationshipList.add(relationship);
+ });
+ }
+ return relationshipList;
+ }
+
+ private List<RelationshipData> convertRelationshipDate(JsonNode relationshipDataNode) {
+ List<RelationshipData> relationshipDataList = new ArrayList<>();
+ if (relationshipDataNode.isArray()) {
+ relationshipDataNode.forEach(node ->{
+ RelationshipData relationshipData = new RelationshipData();
+ relationshipData.setRelationshipKey(
+ getTextElementByNode(node,"relationship-key"));
+ relationshipData.setRelationshipValue(
+ getTextElementByNode(node,"relationship-value"));
+ relationshipDataList.add(relationshipData);
+ });
+ }
+ return relationshipDataList;
+ }
+
+ private List<RelatedToProperty> convertRelatedToProperty(JsonNode relatedToPropertyNode) {
+ List<RelatedToProperty> relatedToPropertyList = new ArrayList<>();
+ if (relatedToPropertyNode.isArray()) {
+ relatedToPropertyNode.forEach(node ->{
+ RelatedToProperty relatedToProperty = new RelatedToProperty();
+ relatedToProperty
+ .setPropertyKey(getTextElementByNode(node, "property-key"));
+ relatedToProperty.setPropertyValue(
+ getTextElementByNode(node, "property-value"));
+ relatedToPropertyList.add(relatedToProperty);
+ });
+ }
+ return relatedToPropertyList;
+ }
+
+ private String getTextElementByNode(JsonNode jsonNode,String name){
+ if(jsonNode.has(name)){
+ return jsonNode.get(name).asText();
+ }
+ return null;
+ }
+
+ private Boolean getBooleanElementByNode(JsonNode jsonNode,String name){
+ if(jsonNode.has(name)){
+ return jsonNode.get(name).asBoolean();
+ }
+ return null;
+ }
+}
Code Review / holmes / common.git / commitdiff