diff options
Diffstat (limited to 'openecomp-be/lib/openecomp-heat-lib/src/main/java')
2 files changed, 13 insertions, 1 deletions
diff --git a/openecomp-be/lib/openecomp-heat-lib/src/main/java/org/openecomp/sdc/heat/datatypes/structure/HeatStructureTree.java b/openecomp-be/lib/openecomp-heat-lib/src/main/java/org/openecomp/sdc/heat/datatypes/structure/HeatStructureTree.java index 8d1b5912df..a65db6f2ea 100644 --- a/openecomp-be/lib/openecomp-heat-lib/src/main/java/org/openecomp/sdc/heat/datatypes/structure/HeatStructureTree.java +++ b/openecomp-be/lib/openecomp-heat-lib/src/main/java/org/openecomp/sdc/heat/datatypes/structure/HeatStructureTree.java @@ -50,6 +50,7 @@ public class HeatStructureTree implements Comparable<HeatStructureTree> { private Set<HeatStructureTree> nested; private Set<HeatStructureTree> other; private Set<Artifact> artifacts; + private Set<HeatStructureTree> helm; public HeatStructureTree() { } @@ -165,6 +166,14 @@ public class HeatStructureTree implements Comparable<HeatStructureTree> { this.other.add(other); } + public void addToHelmList(HeatStructureTree helm){ + if (this.helm == null) { + this.helm = new TreeSet<>(); + } + + this.helm.add(helm); + } + /** * Find item in set by name boolean. * diff --git a/openecomp-be/lib/openecomp-heat-lib/src/main/java/org/openecomp/sdc/heat/services/tree/HeatTreeManager.java b/openecomp-be/lib/openecomp-heat-lib/src/main/java/org/openecomp/sdc/heat/services/tree/HeatTreeManager.java index 4ac09fbddd..b441fc8660 100644 --- a/openecomp-be/lib/openecomp-heat-lib/src/main/java/org/openecomp/sdc/heat/services/tree/HeatTreeManager.java +++ b/openecomp-be/lib/openecomp-heat-lib/src/main/java/org/openecomp/sdc/heat/services/tree/HeatTreeManager.java @@ -251,7 +251,10 @@ public class HeatTreeManager { } parentHeatStructureTree.getOther().add(childHeatStructureTree); } - } else { + } else if (FileData.Type.HELM.equals(type)) { + parentHeatStructureTree.addToHelmList(childHeatStructureTree); + } + else { artifact = new Artifact(fileName, type); if (!artifactRef.keySet().contains(fileName)) { artifactRef.put(fileName, artifact); |