AAI Query optimization for VID

Issue-ID: VID-986
Change-Id: Ia3e012c41df4e9049ce9a1ae56ec83b276e11611
Signed-off-by: PATTANAYAK, SAUMYA SWARUP (sp931a) <sp931a@att.com>

2 files changed, 335 insertions, 0 deletions

diff --git a/vid-automation/src/main/java/org/onap/simulator/presetGenerator/presets/aai/PresetAAIGetServiceInstanceBySubscriberIdAndServiceTypeAndSIID.java b/vid-automation/src/main/java/org/onap/simulator/presetGenerator/presets/aai/PresetAAIGetServiceInstanceBySubscriberIdAndServiceTypeAndSIID.java
new file mode 100644
index 000000000..fe722ae20
--- /dev/null
+++ b/vid-automation/src/main/java/org/onap/simulator/presetGenerator/presets/aai/PresetAAIGetServiceInstanceBySubscriberIdAndServiceTypeAndSIID.java
@@ -0,0 +1,230 @@
+package org.onap.simulator.presetGenerator.presets.aai;
+
+import org.onap.simulator.presetGenerator.presets.BasePresets.BaseAAIPreset;
+import org.springframework.http.HttpMethod;
+
+public class PresetAAIGetServiceInstanceBySubscriberIdAndServiceTypeAndSIID extends BaseAAIPreset {
+    private String subscriberId;
+
+    public String getSubscriberId() {
+        return subscriberId;
+    }
+
+    public String getServiceType() {
+        return serviceType;
+    }
+
+    public String getServiceInstanceId() {
+        return serviceInstanceId;
+    }
+
+    private String serviceType;
+    private String serviceInstanceId;
+
+    public PresetAAIGetServiceInstanceBySubscriberIdAndServiceTypeAndSIID(String subscriberId, String serviceType, String serviceInstanceId) {
+        this.subscriberId = subscriberId;
+        this.serviceType = serviceType;
+        this.serviceInstanceId = serviceInstanceId;
+    }
+
+//    @Override
+//    public Map<String, List> getQueryParams() {
+//        return ImmutableMap.of("depth",  Collections.singletonList("1"));
+//    }
+
+    @Override
+    public HttpMethod getReqMethod() {
+        return HttpMethod.GET;
+    }
+
+    @Override
+    public String getReqPath() {
+        return getRootPath() +
+                "/business/customers/customer/"+getSubscriberId()+
+                "/service-subscriptions/service-subscription/"+getServiceType()+
+                "/service-instances/service-instance/"+getServiceInstanceId();
+    }
+
+    @Override
+    public Object getResponseBody() {
+        return "{\n"
+            + " \"service-instance-id\": \"5d942bc7-3acf-4e35-836a-393619ebde66\","
+            + " \"service-instance-name\": \"dpa2actsf5001v_Port_Mirroring_dpa2a_SVC\","
+            + " \"model-invariant-id\": \"0757d856-a9c6-450d-b494-e1c0a4aab76f\","
+            + " \"model-version-id\": \"a9088517-efe8-4bed-9c54-534462cb08c2\","
+            + " \"resource-version\": \"1500789244673\","
+            + " \"orchestration-status\": \"Active\","
+            + " \"relationship-list\": {"
+            + "  \"relationship\": ["
+            + "    {"
+            + "      \"related-to\": \"generic-vnf\","
+            + "      \"related-link\": \"/aai/v11/network/generic-vnfs/generic-vnf/c015cc0f-0f37-4488-aabf-53795fd93cd3\","
+            + "      \"relationship-data\": ["
+            + "        {"
+            + "          \"relationship-key\": \"generic-vnf.vnf-id\","
+            + "          \"relationship-value\": \"c015cc0f-0f37-4488-aabf-53795fd93cd3\""
+            + "        }"
+            + "      ],"
+            + "      \"related-to-property\": ["
+            + "        {"
+            + "          \"property-key\": \"generic-vnf.vnf-name\","
+            + "          \"property-value\": \"fsd\""
+            + "        }"
+            + "      ]"
+            + "    },"
+            + "    {"
+            + "      \"related-to\": \"generic-vnf\","
+            + "      \"related-link\": \"/aai/v11/network/generic-vnfs/generic-vnf/0846287b-65bf-45a6-88f6-6a1af4149fac\","
+            + "      \"relationship-data\": ["
+            + "        {"
+            + "          \"relationship-key\": \"generic-vnf.vnf-id\","
+            + "          \"relationship-value\": \"0846287b-65bf-45a6-88f6-6a1af4149fac\""
+            + "        }"
+            + "      ],"
+            + "      \"related-to-property\": ["
+            + "        {"
+            + "          \"property-key\": \"generic-vnf.vnf-name\","
+            + "          \"property-value\": \"kjkjk\""
+            + "        }"
+            + "      ]"
+            + "    },"
+            + "    {"
+            + "      \"related-to\": \"generic-vnf\","
+            + "      \"related-link\": \"/aai/v11/network/generic-vnfs/generic-vnf/9908b762-136f-4b1f-8eb4-ef670ef58bb4\","
+            + "      \"relationship-data\": ["
+            + "        {"
+            + "          \"relationship-key\": \"generic-vnf.vnf-id\","
+            + "          \"relationship-value\": \"9908b762-136f-4b1f-8eb4-ef670ef58bb4\""
+            + "        }"
+            + "      ],"
+            + "      \"related-to-property\": ["
+            + "        {"
+            + "          \"property-key\": \"generic-vnf.vnf-name\","
+            + "          \"property-value\": \"uiui\""
+            + "        }"
+            + "      ]"
+            + "    },"
+            + "    {"
+            + "      \"related-to\": \"generic-vnf\","
+            + "      \"related-link\": \"/aai/v11/network/generic-vnfs/generic-vnf/543931f5-e50e-45a2-a69f-ab727e4c7f2f\","
+            + "      \"relationship-data\": ["
+            + "        {"
+            + "          \"relationship-key\": \"generic-vnf.vnf-id\","
+            + "          \"relationship-value\": \"543931f5-e50e-45a2-a69f-ab727e4c7f2f\""
+            + "        }"
+            + "      ],"
+            + "      \"related-to-property\": ["
+            + "        {"
+            + "          \"property-key\": \"generic-vnf.vnf-name\","
+            + "          \"property-value\": \"sdfsdfdsf\""
+            + "        }"
+            + "      ]"
+            + "    },"
+            + "    {"
+            + "      \"related-to\": \"generic-vnf\","
+            + "      \"related-link\": \"/aai/v11/network/generic-vnfs/generic-vnf/25e84884-22d5-44c9-8212-cb459f63e0ba\","
+            + "      \"relationship-data\": ["
+            + "        {"
+            + "          \"relationship-key\": \"generic-vnf.vnf-id\","
+            + "          \"relationship-value\": \"25e84884-22d5-44c9-8212-cb459f63e0ba\""
+            + "        }"
+            + "      ],"
+            + "      \"related-to-property\": ["
+            + "        {"
+            + "          \"property-key\": \"generic-vnf.vnf-name\","
+            + "          \"property-value\": \"sdada\""
+            + "        }"
+            + "      ]"
+            + "    },"
+            + "    {"
+            + "      \"related-to\": \"generic-vnf\","
+            + "      \"related-link\": \"/aai/v11/network/generic-vnfs/generic-vnf/013fb0ba-977b-496c-9faa-7f8e5f083eec\","
+            + "      \"relationship-data\": ["
+            + "        {"
+            + "          \"relationship-key\": \"generic-vnf.vnf-id\","
+            + "          \"relationship-value\": \"013fb0ba-977b-496c-9faa-7f8e5f083eec\""
+            + "        }"
+            + "      ],"
+            + "      \"related-to-property\": ["
+            + "        {"
+            + "          \"property-key\": \"generic-vnf.vnf-name\","
+            + "          \"property-value\": \"gvb\""
+            + "        }"
+            + "      ]"
+            + "    },"
+            + "    {"
+            + "      \"related-to\": \"generic-vnf\","
+            + "      \"related-link\": \"/aai/v11/network/generic-vnfs/generic-vnf/06914296-cb46-4b62-9453-329a706a6cbb\","
+            + "      \"relationship-data\": ["
+            + "        {"
+            + "          \"relationship-key\": \"generic-vnf.vnf-id\","
+            + "          \"relationship-value\": \"06914296-cb46-4b62-9453-329a706a6cbb\""
+            + "        }"
+            + "      ],"
+            + "      \"related-to-property\": ["
+            + "        {"
+            + "          \"property-key\": \"generic-vnf.vnf-name\","
+            + "          \"property-value\": \"lkllll\""
+            + "        }"
+            + "      ]"
+            + "    },"
+            + "    {"
+            + "      \"related-to\": \"generic-vnf\","
+            + "      \"related-link\": \"/aai/v11/network/generic-vnfs/generic-vnf/c55da606-cf38-42c7-bc3c-be8e23b19299\","
+            + "      \"relationship-data\": ["
+            + "        {"
+            + "          \"relationship-key\": \"generic-vnf.vnf-id\","
+            + "          \"relationship-value\": \"c55da606-cf38-42c7-bc3c-be8e23b19299\""
+            + "        }"
+            + "      ],"
+            + "      \"related-to-property\": ["
+            + "        {"
+            + "          \"property-key\": \"generic-vnf.vnf-name\","
+            + "          \"property-value\": \"ss\""
+            + "        }"
+            + "      ]"
+            + "    },"
+            + "    {"
+            + "      \"related-to\": \"generic-vnf\","
+            + "      \"related-link\": \"/aai/v11/network/generic-vnfs/generic-vnf/27cc0914-70be-453e-b552-3df6b1d6cda9\","
+            + "      \"relationship-data\": ["
+            + "        {"
+            + "          \"relationship-key\": \"generic-vnf.vnf-id\","
+            + "          \"relationship-value\": \"27cc0914-70be-453e-b552-3df6b1d6cda9\""
+            + "        }"
+            + "      ],"
+            + "      \"related-to-property\": ["
+            + "        {"
+            + "          \"property-key\": \"generic-vnf.vnf-name\","
+            + "          \"property-value\": \"yh\""
+            + "        }"
+            + "      ]"
+            + "    },"
+            + "    {"
+            + "      \"related-to\": \"logical-link\","
+            + "      \"related-link\": \"/aai/v11/network/logical-links/logical-link/tesai372ve2%3Aae10%7Ctesaaisdgrbclz1a1%3Apo100\","
+            + "      \"relationship-data\": ["
+            + "        {"
+            + "          \"relationship-key\": \"logical-link.link-name\","
+            + "          \"relationship-value\": \"tesai372ve2:ae10|tesaaisdgrbclz1a1:po100\""
+            + "        }"
+            + "      ]"
+            + "    },"
+            + "    {"
+            + "      \"related-to\": \"logical-link\","
+            + "      \"related-link\": \"/aai/v11/network/logical-links/logical-link/SANITY6758cce9%3ALAG1992%7CSANITY6785cce9%3ALAG1961\","
+            + "      \"relationship-data\": ["
+            + "        {"
+            + "          \"relationship-key\": \"logical-link.link-name\","
+            + "          \"relationship-value\": \"SANITY6758cce9:LAG1992|SANITY6785cce9:LAG1961\""
+            + "        }"
+            + "           ]"
+            + "         }"
+            + "       ]"
+            + "     }"
+            + "   }"
+            + " }";
+    }
+
+}
+
diff --git a/vid-automation/src/main/java/org/onap/simulator/presetGenerator/presets/aai/PresetAAIServiceInstanceDSLPut.java b/vid-automation/src/main/java/org/onap/simulator/presetGenerator/presets/aai/PresetAAIServiceInstanceDSLPut.java
new file mode 100644
index 000000000..3ab5a4bbd
--- /dev/null
+++ b/vid-automation/src/main/java/org/onap/simulator/presetGenerator/presets/aai/PresetAAIServiceInstanceDSLPut.java
@@ -0,0 +1,105 @@
+package org.onap.simulator.presetGenerator.presets.aai;
+
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableMap;
+import org.onap.simulator.presetGenerator.presets.BasePresets.BaseAAIPreset;
+import org.springframework.http.HttpMethod;
+
+import java.util.Collections;
+import java.util.List;
+import java.util.Map;
+
+public class PresetAAIServiceInstanceDSLPut extends BaseAAIPreset {
+
+    public PresetAAIServiceInstanceDSLPut(String globalCustomerId, String serviceInstanceIdentifier, String instanceIdentifierType) {
+        this.serviceInstanceIdentifier = serviceInstanceIdentifier;
+        this.globalCustomerId = globalCustomerId;
+        this.instanceIdentifierType = instanceIdentifierType;
+    }
+
+    public String getInstanceIdentifierType() {
+        return instanceIdentifierType;
+    }
+
+    private final String instanceIdentifierType;
+    private final String globalCustomerId;
+
+    public String getGlobalCustomerId() {
+        return globalCustomerId;
+    }
+
+    public String getServiceInstanceIdentifier() {
+        return serviceInstanceIdentifier;
+    }
+
+    private final String serviceInstanceIdentifier;
+
+    @Override
+    public HttpMethod getReqMethod() {
+        return HttpMethod.PUT;
+    }
+
+    @Override
+    public String getReqPath() {
+        return getRootPath() + "/dsl";
+    }
+
+    @Override
+    public Map<String, List> getQueryParams() {
+        return ImmutableMap.of(
+            "format", Collections.singletonList("resource"),
+            "nodesOnly", Collections.singletonList("true"),
+            "depth", Collections.singletonList("0"),
+            "as-tree", Collections.singletonList("true")
+        );
+    }
+
+    @Override
+    public Object getRequestBody() {
+        String requestBody = null;
+        String query = null;
+        if(getInstanceIdentifierType().equals("Service Instance Id")) {
+            query = "customer*('global-customer-id','" + getGlobalCustomerId() + "')>" +
+                "service-subscription>service-instance*('service-instance-id','" + getServiceInstanceIdentifier() + "')";
+            requestBody = "{\"dsl\":\"" + query + "\"}";
+        } else {
+            query = "customer*('global-customer-id','" + getGlobalCustomerId() + "')>" +
+                "service-subscription>service-instance*('service-instance-name','" + getServiceInstanceIdentifier() + "')";
+            requestBody = "{\"dsl\":\"" + query + "\"}";
+        }
+        return requestBody;
+    }
+
+
+    @Override
+    public Object getResponseBody() {
+        return "{\"results\": [\n"
+            + "{\n"
+            + "\"customer\": {\n"
+            + "\"global-customer-id\": \"a9a77d5a-123e-4ca2-9eb9-0b015d2ee0fb\",\n"
+            + "\"subscriber-name\": \"Mobility\",\n"
+            + "\"subscriber-type\": \"INFRA\",\n"
+            + "\"resource-version\": \"1602518417955\",\n"
+            + "\"related-nodes\": [\n"
+            + "{\n"
+            + "\"service-subscription\": {\n"
+            + "\"service-type\": \"VPMS\",\n"
+            + "\"resource-version\": \"1629183620246\",\n"
+            + "\"related-nodes\": [\n"
+            + "{\n"
+            + "\"service-instance\": {\n"
+            + "\"service-instance-id\": \"5d942bc7-3acf-4e35-836a-393619ebde66\",\n"
+            + "\"service-instance-name\": \"dpa2actsf5001v_Port_Mirroring_dpa2a_SVC\",\n"
+            + "\"service-type\": \"PORT-MIRROR\",\n"
+            + "\"service-role\": \"VPROBE\",\n"
+            + "\"environment-context\": \"General_Revenue-Bearing\",\n"
+            + "\"workload-context\": \"Production\",\n"
+            + "\"model-invariant-id\": \"0757d856-a9c6-450d-b494-e1c0a4aab76f\",\n"
+            + "\"model-version-id\": \"a9088517-efe8-4bed-9c54-534462cb08c2\",\n"
+            + "\"resource-version\": \"1615330529236\",\n"
+            + "\"selflink\": \"SOME_SELF_LINK\",\n"
+            + "\"orchestration-status\": \"Active\"\n"
+            + "}}]}}]}}]}";
+
+    }
+}