diff options
Diffstat (limited to 'mod/bpgenerator/src/main')
2 files changed, 30 insertions, 39 deletions
diff --git a/mod/bpgenerator/src/main/java/org/onap/blueprintgenerator/models/blueprint/BpConstants.java b/mod/bpgenerator/src/main/java/org/onap/blueprintgenerator/models/blueprint/BpConstants.java index ac14b47..d0d79a4 100644 --- a/mod/bpgenerator/src/main/java/org/onap/blueprintgenerator/models/blueprint/BpConstants.java +++ b/mod/bpgenerator/src/main/java/org/onap/blueprintgenerator/models/blueprint/BpConstants.java @@ -37,4 +37,7 @@ public final class BpConstants { public static final String TOSCA_DATATYPES_ROOT = "tosca.datatypes.Root"; public static final String TOSCA_NODES_ROOT = "tosca.nodes.Root"; public static final String TOSCA_SIMPLE_YAML = "tosca_simple_yaml_1_0_0"; + + public static final String MEMORY_LIMIT = "128Mi"; + public static final String CPU_LIMIT = "250m"; } diff --git a/mod/bpgenerator/src/main/java/org/onap/blueprintgenerator/models/blueprint/ResourceConfig.java b/mod/bpgenerator/src/main/java/org/onap/blueprintgenerator/models/blueprint/ResourceConfig.java index 030374f..3618be3 100644 --- a/mod/bpgenerator/src/main/java/org/onap/blueprintgenerator/models/blueprint/ResourceConfig.java +++ b/mod/bpgenerator/src/main/java/org/onap/blueprintgenerator/models/blueprint/ResourceConfig.java @@ -22,10 +22,11 @@ package org.onap.blueprintgenerator.models.blueprint; import static org.onap.blueprintgenerator.common.blueprint.BlueprintHelper.createStringInput; +import static org.onap.blueprintgenerator.models.blueprint.BpConstants.CPU_LIMIT; +import static org.onap.blueprintgenerator.models.blueprint.BpConstants.MEMORY_LIMIT; import java.util.LinkedHashMap; import java.util.TreeMap; - import lombok.AllArgsConstructor; import lombok.Builder; import lombok.Getter; @@ -59,69 +60,56 @@ import lombok.Setter; public class ResourceConfig { - /** - * The limits. - */ private TreeMap<String, GetInput> limits; - - /** - * The requests. - */ private TreeMap<String, GetInput> requests; /** * Creates the resource config. * - * @param inps the inps + * @param inputs the inputs * @param name the name * @return the tree map */ public TreeMap<String, LinkedHashMap<String, Object>> createResourceConfig( - TreeMap<String, LinkedHashMap<String, Object>> inps, String name) { - - LinkedHashMap<String, Object> memoryLimit = createStringInput("128Mi"); - LinkedHashMap<String, Object> cpuLimit = createStringInput("250m"); + TreeMap<String, LinkedHashMap<String, Object>> inputs, String name) { String namePrefix = getNamePrefix(name); - //set the limits - TreeMap<String, GetInput> limits = new TreeMap<>(); + limits = createInputs(inputs, namePrefix, "limit"); + requests = createInputs(inputs, namePrefix, "request"); - GetInput cpu = new GetInput(); - cpu.setBpInputName(namePrefix + "cpu_limit"); - limits.put("cpu", cpu); - - GetInput memL = new GetInput(); - memL.setBpInputName(namePrefix + "memory_limit"); - limits.put("memory", memL); - - inps.put(namePrefix + "cpu_limit", cpuLimit); - inps.put(namePrefix + "memory_limit", memoryLimit); - - this.setLimits(limits); + return inputs; + } - //set the requests - TreeMap<String, GetInput> requests = new TreeMap<>(); + private TreeMap<String, GetInput> createInputs(TreeMap<String, LinkedHashMap<String, Object>> inputs, + String namePrefix, + String inputType) { - GetInput cpuR = new GetInput(); - cpuR.setBpInputName(namePrefix + "cpu_request"); - requests.put("cpu", cpuR); + LinkedHashMap<String, Object> memoryLimit = createStringInput(MEMORY_LIMIT); + LinkedHashMap<String, Object> cpuLimit = createStringInput(CPU_LIMIT); - GetInput memR = new GetInput(); - memR.setBpInputName(namePrefix + "memory_request"); - requests.put("memory", memR); + final String cpuKey = namePrefix + "cpu_" + inputType; + final String memoryKey = namePrefix + "memory_" + inputType; + TreeMap<String, GetInput> inps = new TreeMap<>(); - inps.put(namePrefix + "cpu_request", cpuLimit); - inps.put(namePrefix + "memory_request", memoryLimit); + insertInput("cpu", cpuKey, inps); + insertInput("memory", memoryKey, inps); - this.setRequests(requests); + inputs.put(cpuKey, cpuLimit); + inputs.put(memoryKey, memoryLimit); return inps; } + private void insertInput(String type, String name, TreeMap<String, GetInput> inputs) { + GetInput input = new GetInput(); + input.setBpInputName(name); + inputs.put(type, input); + } + private String getNamePrefix(String name) { - return name.isEmpty() ? "" : name + "_"; + return (name == null || name.isEmpty()) ? "" : name + "_"; } } |