Merge "Merge from ECOMP's repository"

1 files changed, 43 insertions, 0 deletions

diff --git a/vid-app-common/src/main/java/org/onap/vid/asdc/parser/ServiceModelInflator.kt b/vid-app-common/src/main/java/org/onap/vid/asdc/parser/ServiceModelInflator.kt
new file mode 100644
index 000000000..79ab0459a
--- /dev/null
+++ b/vid-app-common/src/main/java/org/onap/vid/asdc/parser/ServiceModelInflator.kt
@@ -0,0 +1,43 @@
+package org.onap.vid.asdc.parser
+
+import org.onap.vid.model.*
+import org.springframework.stereotype.Component
+
+@Component
+class ServiceModelInflator {
+
+    data class Names (val modelCustomizationName: String?, val modelKey: String?)
+
+    fun toNamesByVersionId(model: ServiceModel): Map<String, Names> {
+        return emptyMap<String, Names>()
+                .plus(inflate(model.networks))
+                .plus(inflate(model.vnfs))
+                .plus(inflate(model.vnfGroups))
+    }
+
+    private fun inflate(instances: Map<String, *>): Map<String, Names> {
+        return instances.entries.map { inflate(it.key, it.value) }.fold(emptyMap()) { acc, it -> acc.plus(it) }
+    }
+
+    private fun inflate(modelKey: String, vnf: VNF): Map<String, Names> {
+        return mapOf(vnf.uuid to Names(vnf.modelCustomizationName, modelKey))
+                .plus(inflate(vnf.vfModules))
+                .plus(inflate(vnf.volumeGroups))
+    }
+
+    private fun inflate(modelKey: String, instance: Any?): Map<String, Names> {
+        return when (instance) {
+            is Network -> mapOf(instance.uuid to Names(instance.modelCustomizationName, modelKey))
+            is VfModule -> mapOf(instance.uuid to Names(instance.modelCustomizationName, modelKey))
+            is VolumeGroup -> mapOf(instance.uuid to Names(instance.modelCustomizationName, modelKey))
+            is ResourceGroup -> mapOf(instance.uuid to Names(instance.modelCustomizationName, modelKey))
+            is VNF -> inflate(modelKey, instance)
+
+            else -> {
+                // sink
+                emptyMap()
+            }
+        }
+    }
+
+}