summaryrefslogtreecommitdiffstats
path: root/mod/bpgenerator/src/main/java/org/onap/blueprintgenerator/models/blueprint/ResourceConfig.java
blob: 3618be33077271d9208b7fc1ee09187d3748e690 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
/*============LICENSE_START=======================================================
 org.onap.dcae
 ================================================================================
 Copyright (c) 2019 AT&T Intellectual Property. All rights reserved.
 Copyright (c) 2020 Nokia. All rights reserved.
 ================================================================================
 Licensed under the Apache License, Version 2.0 (the "License");
 you may not use this file except in compliance with the License.
 You may obtain a copy of the License at

      http://www.apache.org/licenses/LICENSE-2.0

 Unless required by applicable law or agreed to in writing, software
 distributed under the License is distributed on an "AS IS" BASIS,
 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License.
 ============LICENSE_END=========================================================

 */

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;
import lombok.NoArgsConstructor;
import lombok.Setter;

//TODO: Auto-generated Javadoc
/* (non-Javadoc)
 * @see java.lang.Object#toString()
 */
@Getter
@Setter

/* (non-Javadoc)
 * @see java.lang.Object#toString()
 */
@Builder

/**
 * Instantiates a new resource config obj.
 */
@NoArgsConstructor

/**
 * Instantiates a new resource config obj.
 *
 * @param limits the limits
 * @param requests the requests
 */
@AllArgsConstructor

public class ResourceConfig {

    private TreeMap<String, GetInput> limits;
    private TreeMap<String, GetInput> requests;


    /**
     * Creates the resource config.
     *
     * @param inputs the inputs
     * @param name the name
     * @return the tree map
     */
    public TreeMap<String, LinkedHashMap<String, Object>> createResourceConfig(
        TreeMap<String, LinkedHashMap<String, Object>> inputs, String name) {

        String namePrefix = getNamePrefix(name);

        limits = createInputs(inputs, namePrefix, "limit");
        requests = createInputs(inputs, namePrefix, "request");

        return inputs;
    }

    private TreeMap<String, GetInput> createInputs(TreeMap<String, LinkedHashMap<String, Object>> inputs,
        String namePrefix,
        String inputType) {

        LinkedHashMap<String, Object> memoryLimit = createStringInput(MEMORY_LIMIT);
        LinkedHashMap<String, Object> cpuLimit = createStringInput(CPU_LIMIT);

        final String cpuKey = namePrefix + "cpu_" + inputType;
        final String memoryKey = namePrefix + "memory_" + inputType;
        TreeMap<String, GetInput> inps = new TreeMap<>();

        insertInput("cpu", cpuKey, inps);
        insertInput("memory", memoryKey, inps);

        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 == null || name.isEmpty()) ? "" : name + "_";
    }
}