1 files changed, 89 insertions, 0 deletions
diff --git a/vid-app-common/src/main/java/org/onap/vid/aai/util/AAITreeConverter.java b/vid-app-common/src/main/java/org/onap/vid/aai/util/AAITreeConverter.java
new file mode 100644
index 000000000..8fa6f6c6d
--- /dev/null
+++ b/vid-app-common/src/main/java/org/onap/vid/aai/util/AAITreeConverter.java
@@ -0,0 +1,89 @@
+package org.onap.vid.aai.util;
+
+import org.apache.commons.lang3.StringUtils;
+import org.onap.vid.model.aaiTree.*;
+import org.onap.vid.mso.model.ModelInfo;
+import org.onap.vid.services.AAITreeNodeBuilder;
+import org.springframework.stereotype.Component;
+
+import java.util.Objects;
+
+import static java.util.function.Function.identity;
+import static java.util.stream.Collectors.counting;
+import static java.util.stream.Collectors.groupingBy;
+import static org.onap.vid.asdc.parser.ToscaParserImpl2.Constants.A_LA_CARTE;
+
+@Component
+public class AAITreeConverter {
+
+    public static final String VNF_TYPE = "vnf-type";
+    public static final String NETWORK_TYPE = "network-type";
+
+    public static final String IS_BASE_VF_MODULE = "is-base-vf-module";
+
+    public enum ModelType {
+        service,
+        vnf,
+        network,
+        instanceGroup,
+        vfModule
+    }
+
+    public ServiceInstance convertTreeToUIModel(AAITreeNode rootNode, String globalCustomerId, String serviceType, String instantiationType) {
+        ServiceInstance serviceInstance = new ServiceInstance();
+        serviceInstance.setInstanceId(rootNode.getId());
+        serviceInstance.setInstanceName(rootNode.getName());
+        serviceInstance.setOrchStatus(rootNode.getOrchestrationStatus());
+        serviceInstance.setGlobalSubscriberId(globalCustomerId);
+        serviceInstance.setSubscriptionServiceType(serviceType);
+        serviceInstance.setIsALaCarte(StringUtils.equals(instantiationType, A_LA_CARTE));
+
+        serviceInstance.setModelInfo(createModelInfo(rootNode, ModelType.service));
+
+        //set children: vnf, network,group
+        rootNode.getChildren().forEach(child -> {
+            if (child.getType().equals(AAITreeNodeBuilder.GENERIC_VNF)) {
+                serviceInstance.getVnfs().put(child.getUniqueNodeKey(), Vnf.from(child));
+            } else if (child.getType().equals(AAITreeNodeBuilder.NETWORK)) {
+                serviceInstance.getNetworks().put(child.getUniqueNodeKey(), Network.from(child));
+            } else if (child.getType().equals(AAITreeNodeBuilder.INSTANCE_GROUP)) {
+                serviceInstance.getVnfGroups().put(child.getUniqueNodeKey(), VnfGroup.from(child));
+            }
+        });
+
+        serviceInstance.setExistingVNFCounterMap(
+                serviceInstance.getVnfs().entrySet().stream()
+                        .map(k -> k.getValue().getModelInfo().getModelVersionId())
+                        .collect(groupingBy(identity(), counting()))
+        );
+
+        serviceInstance.setExistingNetworksCounterMap(
+                serviceInstance.getNetworks().entrySet().stream()
+                .map(k -> k.getValue().getModelInfo().getModelVersionId())
+                .filter(Objects::nonNull)
+                        .collect(groupingBy(identity(), counting()))
+        );
+
+
+        serviceInstance.setExistingVnfGroupCounterMap(
+                serviceInstance.getVnfGroups().entrySet().stream()
+                .map(k -> k.getValue().getModelInfo().getModelVersionId())
+                .filter(Objects::nonNull)
+                        .collect(groupingBy(identity(), counting()))
+        );
+
+        return serviceInstance;
+    }
+
+    private static ModelInfo createModelInfo(AAITreeNode aaiNode, ModelType modelType) {
+        ModelInfo modelInfo = new ModelInfo();
+        modelInfo.setModelType(modelType.name());
+        modelInfo.setModelName(aaiNode.getModelName());
+        modelInfo.setModelVersion(aaiNode.getModelVersion());
+        modelInfo.setModelVersionId(aaiNode.getModelVersionId());
+        modelInfo.setModelInvariantId(aaiNode.getModelInvariantId());
+        modelInfo.setModelCustomizationId(aaiNode.getModelCustomizationId());
+
+        return modelInfo;
+    }
+}