1 files changed, 67 insertions, 0 deletions
diff --git a/catalog-be/src/main/java/org/openecomp/sdc/be/components/merge/utils/ComponentInstanceBuildingBlocks.java b/catalog-be/src/main/java/org/openecomp/sdc/be/components/merge/utils/ComponentInstanceBuildingBlocks.java
new file mode 100644
index 0000000000..9b2bb93ad1
--- /dev/null
+++ b/catalog-be/src/main/java/org/openecomp/sdc/be/components/merge/utils/ComponentInstanceBuildingBlocks.java
@@ -0,0 +1,67 @@
+package org.openecomp.sdc.be.components.merge.utils;
+
+import org.openecomp.sdc.be.datatypes.elements.GroupDataDefinition;
+import org.openecomp.sdc.be.model.ComponentInstance;
+import org.openecomp.sdc.be.model.GroupDefinition;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.stream.Collectors;
+import java.util.stream.Stream;
+
+import static org.openecomp.sdc.be.dao.utils.MapUtil.mergeMaps;
+import static org.openecomp.sdc.be.dao.utils.MapUtil.toMap;
+
+public class ComponentInstanceBuildingBlocks {
+
+    private Map<String, GroupDefinition> groups = new HashMap<>();
+    private Map<String, ComponentInstance> vfcInstances = new HashMap<>();
+
+    private ComponentInstanceBuildingBlocks(){}
+
+    private ComponentInstanceBuildingBlocks(List<GroupDefinition> groups, List<ComponentInstance> vfcInstances) {
+        this.groups = groups == null ? new HashMap<>() : toMap(groups, GroupDataDefinition::getUniqueId);
+        this.vfcInstances = vfcInstances == null ? new HashMap<>() : toMap(vfcInstances, ComponentInstance::getUniqueId);
+    }
+
+    private ComponentInstanceBuildingBlocks(Map<String, GroupDefinition> groups,  Map<String, ComponentInstance> vfcInstances) {
+        this.groups = groups == null ? new HashMap<>() : groups;
+        this.vfcInstances = vfcInstances == null ? new HashMap<>() : vfcInstances;
+    }
+
+    static ComponentInstanceBuildingBlocks of(List<GroupDefinition> groups, List<ComponentInstance> instances) {
+        return new ComponentInstanceBuildingBlocks(groups, instances);
+    }
+
+    static ComponentInstanceBuildingBlocks empty() {
+        return new ComponentInstanceBuildingBlocks();
+    }
+
+
+    @SuppressWarnings("unchecked")
+    static ComponentInstanceBuildingBlocks merge(ComponentInstanceBuildingBlocks first, ComponentInstanceBuildingBlocks second) {
+        return new ComponentInstanceBuildingBlocks(mergeMaps(first.groups, second.groups),
+                                                   mergeMaps(first.vfcInstances, second.vfcInstances));
+    }
+
+    public List<GroupDefinition> getGroups() {
+        return new ArrayList<>(groups.values());
+    }
+
+    public List<ComponentInstance> getVfcInstances() {
+        return new ArrayList<>(vfcInstances.values());
+    }
+
+    public List<CapabilityOwner> getCapabilitiesOwners() {
+        Stream<CapabilityOwner> groupsStream = groups.values().stream()
+                .map(grp -> new CapabilityOwner(grp.getUniqueId(), grp.getInvariantName(), grp.getCapabilities()));
+
+        Stream<CapabilityOwner> instanceStream = vfcInstances.values().stream()
+                .map(instance -> new CapabilityOwner(instance.getUniqueId(), instance.getName(), instance.getCapabilities()));
+
+        return Stream.concat(groupsStream, instanceStream).collect(Collectors.toList());
+    }
+    
+}