Obtain resouce-link based on service-instance

Issue-ID: LOG-461
Change-Id: Ieb3fb761be2237808ab9d709b3fadc7e241963de
Signed-off-by: Phillip Leigh <phillip.leigh@amdocs.com>

3 files changed, 77 insertions, 3 deletions

diff --git a/src/main/java/org/onap/pomba/contextbuilder/aai/AAIConfiguration.java b/src/main/java/org/onap/pomba/contextbuilder/aai/AAIConfiguration.java
index 9c6d11d..7befdc9 100644
--- a/src/main/java/org/onap/pomba/contextbuilder/aai/AAIConfiguration.java
+++ b/src/main/java/org/onap/pomba/contextbuilder/aai/AAIConfiguration.java
@@ -104,4 +104,12 @@ public class AAIConfiguration {
         return serviceInstancePath;
     }
 
+    @Value("${aai.searchNodeQuery}")
+    private String searchNodeQuery;
+
+    @Bean(name="aaiPathToSearchNodeQuery")
+    public String getAaiPathToSearchNodeQuery() {
+        return searchNodeQuery.trim();
+    }
+
 }
diff --git a/src/main/java/org/onap/pomba/contextbuilder/aai/service/SpringServiceImpl.java b/src/main/java/org/onap/pomba/contextbuilder/aai/service/SpringServiceImpl.java
index ff09e55..2ec49f3 100644
--- a/src/main/java/org/onap/pomba/contextbuilder/aai/service/SpringServiceImpl.java
+++ b/src/main/java/org/onap/pomba/contextbuilder/aai/service/SpringServiceImpl.java
@@ -39,6 +39,8 @@ public class SpringServiceImpl implements SpringService {
     private String aaiServiceInstancePath;
     @Autowired
     private String aaiBasicAuthorization;
+    @Autowired
+    private String aaiPathToSearchNodeQuery;
 
     public SpringServiceImpl() {
         // needed for instantiation
@@ -55,7 +57,7 @@ public class SpringServiceImpl implements SpringService {
 
         // Retrieve the service instance information from AAI
         try {
-            context= RestUtil.retrieveAAIModelData(aaiClient, aaiBaseUrl, aaiServiceInstancePath, tranId, serviceInstanceId, modelVersionId, modelInvariantId, serviceType, customerId,aaiBasicAuthorization);
+            context= RestUtil.retrieveAAIModelData(aaiClient, aaiBaseUrl, aaiPathToSearchNodeQuery, aaiServiceInstancePath, tranId, serviceInstanceId, modelVersionId, modelInvariantId, serviceType, customerId,aaiBasicAuthorization);
         } catch (AuditException ae) {
             throw ae;
         } catch (Exception e) {
diff --git a/src/main/java/org/onap/pomba/contextbuilder/aai/util/RestUtil.java b/src/main/java/org/onap/pomba/contextbuilder/aai/util/RestUtil.java
index da7d241..905a8bc 100644
--- a/src/main/java/org/onap/pomba/contextbuilder/aai/util/RestUtil.java
+++ b/src/main/java/org/onap/pomba/contextbuilder/aai/util/RestUtil.java
@@ -31,6 +31,7 @@ import javax.ws.rs.core.MediaType;
 import javax.ws.rs.core.MultivaluedMap;
 import javax.ws.rs.core.Response.Status;
 import org.json.JSONArray;
+import org.json.JSONException;
 import org.json.JSONObject;
 import org.onap.aai.restclient.client.OperationResult;
 import org.onap.aai.restclient.client.RestClient;
@@ -48,7 +49,7 @@ import org.onap.pomba.contextbuilder.aai.exception.AuditError;
 import org.onap.pomba.contextbuilder.aai.exception.AuditException;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
-
+import org.springframework.beans.factory.annotation.Autowired;
 
 public class RestUtil {
 
@@ -72,12 +73,17 @@ public class RestUtil {
     // Service Catalog
     private static final String CATALOG_GENERIC_VNF = "generic-vnf";
     private static final String CATALOG_VNFC = "vnfc";
+    private static final String CATALOG_SERVICE_INSTANCE = "service-instance";
+
     // Relationship Json Path
     private static final String RELATIONSHIP_LIST = "relationship-list";
     private static final String RELATIONSHIP = "relationship";
+    private static final String RESULT_DATA = "result-data";
 
     private static final String JSON_ATT_RELATED_TO = "related-to";
     private static final String JSON_ATT_RELATED_LINK = "related-link";
+    private static final String JSON_ATT_RESOURCE_TYPE = "resource-type";
+    private static final String JSON_ATT_RESOURCE_LINK = "resource-link";
 
     private static final String EMPTY_JSON_STRING = "{}";
     private static final String DELIMITER = "$";
@@ -105,6 +111,7 @@ public class RestUtil {
             throw new AuditException(AuditError.INVALID_REQUEST_URL_MISSING_PARAMETER + MODEL_INVARIANT_ID,
                     Status.BAD_REQUEST);
         }
+
         // serviceType
         if (serviceType == null || serviceType.isEmpty()) {
             throw new AuditException(AuditError.INVALID_REQUEST_URL_MISSING_PARAMETER + SERVICE_TYPE, Status.BAD_REQUEST);
@@ -161,7 +168,7 @@ public class RestUtil {
     /*
      * Trigger external API call to AAI to retrieve Service Instance data (i.e. genericVNF and VNFC)
      */
-    public static ModelContext retrieveAAIModelData(RestClient aaiClient, String baseURL, String aaiServiceInstancePath,
+    public static ModelContext retrieveAAIModelData(RestClient aaiClient, String baseURL, String aaiPathToSearchNodeQuery, String aaiServiceInstancePath, 
             String transactionId, String serviceInstanceId, String modelVersionId, String modelInvariantId,
             String serviceType, String customerId, String aaiBasicAuthorization) throws AuditException {
         String serviceInstancePayload = null;
@@ -173,6 +180,8 @@ public class RestUtil {
         Map<String, List<VnfcInstance>> vnfMap = new HashMap<String, List<VnfcInstance>>(); // MAP the vnf-id as the
                                                                                             // key, and list of the vNFC
                                                                                             // pojo object
+        // Obtain resource-link based on resource-type = service-Instance
+        String resourceLink = obtainResouceLinkBasedOnServiceInstanceFromAAI(aaiClient, baseURL, aaiPathToSearchNodeQuery, serviceInstanceId, transactionId, aaiBasicAuthorization);
 
         String url = baseURL
                 + generateServiceInstanceURL(aaiServiceInstancePath, customerId, serviceType, serviceInstanceId);
@@ -442,5 +451,60 @@ public class RestUtil {
         return MessageFormat.format(siPath, customerId, serviceType, serviceInstanceId);
     }
 
+    public static String obtainResouceLinkBasedOnServiceInstanceFromAAI(RestClient aaiClient, String baseURL, String aaiPathToSearchNodeQuery, String serviceInstanceId,
+            String transactionId, String aaiBasicAuthorization) throws AuditException {
+
+        String url = generateGetCustomerInfoUrl(baseURL, aaiPathToSearchNodeQuery, serviceInstanceId);
+        String customerInfoString  = getResource(aaiClient, url, aaiBasicAuthorization, transactionId, MediaType.valueOf(MediaType.APPLICATION_JSON));
+
+        // Handle the case if the service instance is not found in AAI
+        if (isEmptyJson(customerInfoString)) {
+            log.info(LogMessages.NOT_FOUND, "Service Instance" , serviceInstanceId);
+            // Only return the empty Json on the root level. i.e service instance
+            return null;
+        }
+
+        return extractResourceLinkBasedOnResourceType(customerInfoString, CATALOG_SERVICE_INSTANCE);
+    }
+
+    private static String generateGetCustomerInfoUrl (String baseURL, String aaiPathToSearchNodeQuery ,String serviceInstanceId) {
+        return baseURL + aaiPathToSearchNodeQuery + serviceInstanceId;
+    }
+
+    /*
+     * Extract the resource-Link from Json payload. For example
+     * {
+     *     "result-data": [
+     *         {
+     *             "resource-type": "service-instance",
+     *             "resource-link": "/aai/v11/business/customers/customer/DemoCust_651800ed-2a3c-45f5-b920-85c1ed155fc2/service-subscriptions/service-subscription/vFW/service-instances/service-instance/adc3cc2a-c73e-414f-8ddb-367de81300cb"
+     *         }
+     *     ]
+     * }
+     */
+    private static String extractResourceLinkBasedOnResourceType(String payload, String catalog) throws AuditException {
+        String resourceLink = null;
+        log.info("Fetching the resource-link based on resource-type=" + catalog);
+
+        try {
+            JSONArray result_data_list = new JSONObject(payload).getJSONArray(RESULT_DATA);
+            if (result_data_list != null) {
+                for (int i = 0; i < result_data_list.length(); i++) {
+                    JSONObject obj = result_data_list.optJSONObject(i);
+                    if (obj.has(JSON_ATT_RESOURCE_TYPE) && (obj.getString(JSON_ATT_RESOURCE_TYPE).equals(catalog) ))  {
+                        resourceLink = obj.getString(JSON_ATT_RESOURCE_LINK);
+                        log.info(resourceLink);
+                        return resourceLink;
+                    }
+                }
+            }
+        } catch (JSONException e) {
+            log.error(e.getMessage());
+            throw new AuditException(AuditError.JSON_READER_PARSE_ERROR + " " + e.getMessage());
+        }
+
+        log.warn("resource-link CANNOT be found: ", payload );
 
+        return resourceLink;
+    }
 }