ac07daad8f66b854077e81fb30cb58ed20687bb4
[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 import org.openecomp.sdc.heat.datatypes.model.Resource;
24
25 public class PortConsolidationData {
26
27     //Key - Service template file name
28     private final Map<String, FilePortConsolidationData> filePortConsolidationData;
29
30     public PortConsolidationData() {
31         this.filePortConsolidationData = new HashMap<>();
32     }
33
34     public Set<String> getAllServiceTemplateFileNames() {
35         return filePortConsolidationData.keySet();
36     }
37
38     public FilePortConsolidationData getFilePortConsolidationData(String serviceTemplateFileName) {
39         return filePortConsolidationData.get(serviceTemplateFileName);
40     }
41
42     public void setFilePortConsolidationData(String serviceTemplateFileName, FilePortConsolidationData
43             filePortConsolidationData) {
44         this.filePortConsolidationData.put(serviceTemplateFileName, filePortConsolidationData);
45     }
46
47     /**
48     * Create port template consolidation data base on given parameters - if it doesn't exist yet.
49     *
50     * @return port template consolidation data
51     */
52     PortTemplateConsolidationData addPortTemplateConsolidationData(String serviceTemplateFileName,
53             String portNodeTemplateId, String portResourceId, String portResourceType) {
54         FilePortConsolidationData consolidationData = addFilePortConsolidationData(serviceTemplateFileName);
55         return consolidationData
56                    .addPortTemplateConsolidationData(portNodeTemplateId, portResourceId, portResourceType);
57     }
58
59     /**
60     * Create subInterface template consolidation data base on given parameters - if it doesn't exist yet.
61     *
62     * @return port template consolidation data by given keys
63     */
64     SubInterfaceTemplateConsolidationData addSubInterfaceTemplateConsolidationData(
65             String serviceTemplateFileName, Resource resource, String subInterfaceNodeTemplateId,
66             String parentPortNodeTemplateId) {
67         FilePortConsolidationData consolidationData = addFilePortConsolidationData(serviceTemplateFileName);
68         return consolidationData.addSubInterfaceTemplateConsolidationData(
69             resource, subInterfaceNodeTemplateId, parentPortNodeTemplateId);
70     }
71
72     private FilePortConsolidationData addFilePortConsolidationData(String serviceTemplateFileName) {
73         FilePortConsolidationData consolidationData = getFilePortConsolidationData(serviceTemplateFileName);
74         if (consolidationData == null) {
75             consolidationData = new FilePortConsolidationData();
76             setFilePortConsolidationData(serviceTemplateFileName, consolidationData);
77         }
78         return consolidationData;
79     }
80 }