add query vnf data and vm data from aai

Issue-ID: HOLMES-44

Change-Id: I627e1b94e22db5c62773ee2623a4ed5d14c2649f
Signed-off-by: Shiwei Tian <tian.shiwei@zte.com.cn>

3 files changed, 319 insertions, 0 deletions

diff --git a/holmes-actions/src/main/java/org/onap/holmes/common/aai/AaiQuery.java b/holmes-actions/src/main/java/org/onap/holmes/common/aai/AaiQuery.java
new file mode 100644
index 0000000..71472ed
--- /dev/null
+++ b/holmes-actions/src/main/java/org/onap/holmes/common/aai/AaiQuery.java
@@ -0,0 +1,91 @@
+/**
+ * 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;
+    }
+}
diff --git a/holmes-actions/src/main/java/org/onap/holmes/common/aai/AaiResponseUtil.java b/holmes-actions/src/main/java/org/onap/holmes/common/aai/AaiResponseUtil.java
new file mode 100644
index 0000000..3869522
--- /dev/null
+++ b/holmes-actions/src/main/java/org/onap/holmes/common/aai/AaiResponseUtil.java
@@ -0,0 +1,192 @@
+/**
+ * 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;
+    }
+}
diff --git a/holmes-actions/src/main/java/org/onap/holmes/common/aai/config/AaiConfig.java b/holmes-actions/src/main/java/org/onap/holmes/common/aai/config/AaiConfig.java
new file mode 100644
index 0000000..e3bacc1
--- /dev/null
+++ b/holmes-actions/src/main/java/org/onap/holmes/common/aai/config/AaiConfig.java
@@ -0,0 +1,36 @@
+/**
+ * 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.config;
+
+public class AaiConfig {
+
+    public static String VNF_ADDR = "/aai/v11/network/generic-vnfs/generic-vnf?";
+
+    public static String VM_ADDR = "/aai/v11/search/nodes-query?search-node-type=vserver&filter=";
+
+    public static String X_TRANSACTION_ID = "9999";
+
+    public static String X_FROMAPP_ID = "jimmy-postman";
+
+    private static String AAI_AUTHENTICATION_USER = "AAI";
+
+    private static String AAI_AUTHENTICATION_PAASWORD = "AAI";
+
+    public static String getAuthenticationCredentials() {
+        String usernameAndPassword = AAI_AUTHENTICATION_USER + ":"
+                + AAI_AUTHENTICATION_PAASWORD;
+        return "Basic " + java.util.Base64.getEncoder().encodeToString(usernameAndPassword.getBytes());
+    }
+
+}