Merge "Added policyNodes, db, tls support"

1 files changed, 13 insertions, 0 deletions

diff --git a/mod/bpgenerator/src/main/java/org/onap/blueprintgenerator/models/onapbp/OnapBlueprint.java b/mod/bpgenerator/src/main/java/org/onap/blueprintgenerator/models/onapbp/OnapBlueprint.java
index 8342020..c0ef8b3 100644
--- a/mod/bpgenerator/src/main/java/org/onap/blueprintgenerator/models/onapbp/OnapBlueprint.java
+++ b/mod/bpgenerator/src/main/java/org/onap/blueprintgenerator/models/onapbp/OnapBlueprint.java
@@ -24,6 +24,8 @@ import java.util.ArrayList;
 import java.util.LinkedHashMap;
 import java.util.TreeMap;
 
+import org.onap.blueprintgenerator.core.PgaasNodeBuilder;
+import org.onap.blueprintgenerator.core.PolicyNodeBuilder;
 import org.onap.blueprintgenerator.models.blueprint.Blueprint;
 import org.onap.blueprintgenerator.models.blueprint.Imports;
 import org.onap.blueprintgenerator.models.blueprint.Node;
@@ -69,9 +71,20 @@ public class OnapBlueprint extends Blueprint{
 		nodeTemplate.put(nodeName, node);
 		this.setNode_templates(nodeTemplate);
 
+		//if present in component spec, populate policyNode information in the blueprint
+		if(cs.getPolicyInfo() != null){
+			PolicyNodeBuilder.addPolicyNodesAndInputs(cs, nodeTemplate, inputs);
+		}
+
+		//if present in component spec, populate pgaasNodes information in the blueprint
+		if(cs.getAuxilary().getDatabases() != null){
+			PgaasNodeBuilder.addPgaasNodesAndInputs(cs, nodeTemplate, inputs);
+		}
+
 		//set the inputs
 		this.setInputs(inputs);
 
+
 		Blueprint bp = new Blueprint();
 		bp.setImports(this.getImports());
 		bp.setInputs(this.getInputs());