Fix issue with VNFImage rollback

[aai/model-loader.git] / src / main / java / org / onap / aai / modelloader / entity / catalog / VnfCatalogArtifactHandler.java

diff --git a/src/main/java/org/onap/aai/modelloader/entity/catalog/VnfCatalogArtifactHandler.java b/src/main/java/org/onap/aai/modelloader/entity/catalog/VnfCatalogArtifactHandler.java
index 85be50b..3480c68 100644 (file)
--- a/src/main/java/org/onap/aai/modelloader/entity/catalog/VnfCatalogArtifactHandler.java
+++ b/src/main/java/org/onap/aai/modelloader/entity/catalog/VnfCatalogArtifactHandler.java
@@ -89,7 +89,7 @@ public class VnfCatalogArtifactHandler extends ArtifactHandler {
             // If an empty dataItem is supplied, do nothing.\r
             if (dataItem.isEmpty()) {\r
                 logger.warn(ModelLoaderMsgs.DISTRIBUTION_EVENT, "Empty image data supplied, skipping ingestion.");\r
-                return;\r
+                continue;\r
             }\r
 \r
             String urlParams;\r
@@ -142,18 +142,13 @@ public class VnfCatalogArtifactHandler extends ArtifactHandler {
         String uuid = UUID.randomUUID().toString();\r
         dataItem.put(ATTR_UUID, uuid);\r
 \r
-        String payload = createVnfImagePayload(dataItem);\r
+        String payload = new Gson().toJson(dataItem);\r
         String putUrl = config.getAaiBaseUrl() + config.getAaiVnfImageUrl() + "/vnf-image/" + uuid;\r
         OperationResult putResp =\r
                 restClient.putResource(putUrl, payload, distributionId, MediaType.APPLICATION_JSON_TYPE);\r
         return putResp != null && putResp.getResultCode() == Response.Status.CREATED.getStatusCode();\r
     }\r
 \r
-    private String createVnfImagePayload(Map<String, String> dataItem) {\r
-        dataItem.put(ATTR_UUID, UUID.randomUUID().toString());\r
-        return new Gson().toJson(dataItem);\r
-    }\r
-\r
     private List<Map<String, String>> unmarshallVnfcData(Artifact vnfcArtifact) {\r
         // Unmarshall Babel JSON payload into a List of Maps of JSON attribute name/values.\r
         return new Gson().fromJson(StringEscapeUtils.unescapeJson(vnfcArtifact.getPayload()),\r