afad4a9b0f008bed283b6e450357af78ca02fdea
[sdc.git] /
1 /*
2  * Copyright © 2016-2018 European Support Limited
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 package org.openecomp.sdc.translator.datatypes.heattotosca.unifiedmodel.consolidation;
18
19 import java.util.HashMap;
20 import java.util.Map;
21 import java.util.Set;
22
23 public class NestedConsolidationData {
24
25     //Key - Service template file name
26     private final Map<String, FileNestedConsolidationData> fileNestedConsolidationData;
27
28     public NestedConsolidationData() {
29         this.fileNestedConsolidationData = new HashMap<>();
30     }
31
32     public Set<String> getAllServiceTemplateFileNames() {
33         return fileNestedConsolidationData.keySet();
34     }
35
36     public FileNestedConsolidationData getFileNestedConsolidationData(String serviceTemplateFileName) {
37         return fileNestedConsolidationData.get(serviceTemplateFileName);
38     }
39
40     public void setFileNestedConsolidationData(String serviceTemplateFileName,
41             FileNestedConsolidationData fileNestedConsolidationData) {
42         this.fileNestedConsolidationData.put(serviceTemplateFileName, fileNestedConsolidationData);
43     }
44
45     /**
46     * Create nested template consolidation data base on given key - if it doesn't exist yet.
47     *
48     * @return nested template consolidation data by given keys
49     */
50     NestedTemplateConsolidationData addNestedTemplateConsolidationData(
51             String serviceTemplateFileName, String nestedNodeTemplateId) {
52
53         FileNestedConsolidationData consolidationData = addFileNestedConsolidationData(serviceTemplateFileName);
54         return consolidationData.addNestedTemplateConsolidationData(nestedNodeTemplateId);
55     }
56
57     private FileNestedConsolidationData addFileNestedConsolidationData(String serviceTemplateFileName) {
58         FileNestedConsolidationData consolidationData = getFileNestedConsolidationData(serviceTemplateFileName);
59         if (consolidationData == null) {
60             consolidationData = new FileNestedConsolidationData();
61             setFileNestedConsolidationData(serviceTemplateFileName, consolidationData);
62         }
63         return consolidationData;
64     }
65 }