X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Forg%2Fonap%2Faai%2Fmodelloader%2Frestclient%2FBabelServiceClientImpl.java;h=53ce91779164152aa0d8cc5b038570db4375acaa;hb=8204d232f8cc37e28561ee18b60806c7b3f5f783;hp=540089224ec165dcd5fcc398c60b3f66e60fd57e;hpb=0e9a90774a86475f40d4e946798cf765910a31f6;p=aai%2Fmodel-loader.git

diff --git a/src/main/java/org/onap/aai/modelloader/restclient/BabelServiceClientImpl.java b/src/main/java/org/onap/aai/modelloader/restclient/BabelServiceClientImpl.java
index 5400892..53ce917 100644
--- a/src/main/java/org/onap/aai/modelloader/restclient/BabelServiceClientImpl.java
+++ b/src/main/java/org/onap/aai/modelloader/restclient/BabelServiceClientImpl.java
@@ -69,7 +69,7 @@ public class BabelServiceClientImpl implements BabelServiceClient {
         headers.set(AaiRestClient.HEADER_FROM_APP_ID, AaiRestClient.ML_APP_NAME);
         HttpEntity<BabelRequest> entity = new HttpEntity<>(babelRequest, headers);
 
-        ResponseEntity<List<org.onap.aai.modelloader.babel.BabelArtifact>> artifactResponse = restTemplate.exchange(resourceUrl, HttpMethod.POST, entity, new ParameterizedTypeReference<List<org.onap.aai.modelloader.babel.BabelArtifact>>() {});
+        ResponseEntity<List<BabelArtifact>> artifactResponse = restTemplate.exchange(resourceUrl, HttpMethod.POST, entity, new ParameterizedTypeReference<List<BabelArtifact>>() {});
 
         if (logger.isDebugEnabled()) {
             logger.debug(ModelLoaderMsgs.DISTRIBUTION_EVENT,
@@ -80,20 +80,6 @@ public class BabelServiceClientImpl implements BabelServiceClient {
             throw new BabelServiceClientException(artifactResponse.getBody().toString());
         }
 
-        List<BabelArtifact> babelArtifact = artifactResponse.getBody().stream()
-            .map(this::mapBabelDto)
-            .collect(Collectors.toList());
-
-        return babelArtifact;
-    }
-
-    private BabelArtifact mapBabelDto(org.onap.aai.modelloader.babel.BabelArtifact babelDto) {
-        ArtifactType artifactType = babelDto.getType() == null
-            ? null 
-            : ArtifactType.valueOf(babelDto.getType().name());
-        return new BabelArtifact(
-            babelDto.getName(),
-            artifactType,
-            babelDto.getPayload());
+        return artifactResponse.getBody();
     }
 }