openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/main/java/org/openecomp/sdc/translator/datatypes/heattotosca/unifiedmodel/consolidation/PortConsolidationData.java

   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 org.openecomp.sdc.heat.datatypes.model.Resource;
  20
  21 import java.util.HashMap;
  22 import java.util.Map;
  23 import java.util.Set;
  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     /**
  73      * Create subInterface template consolidation data base on given parameters - if it doesn't exist yet.
  74      *
  75      * @return port template consolidation data by given keys
  76      */
  77     SubInterfaceTemplateConsolidationData addSubInterfaceTemplateConsolidationData(
  78                 String serviceTemplateFileName, Resource resource,
  79                 String subInterfaceNodeTemplateId, String parentPortNodeTemplateId,
  80                 String parentPortResourceId, String parentPortResourceType) {
  81         FilePortConsolidationData consolidationData = addFilePortConsolidationData(serviceTemplateFileName);
  82         return consolidationData.addSubInterfaceTemplateConsolidationData(
  83                 resource, subInterfaceNodeTemplateId, parentPortNodeTemplateId,
  84                 parentPortResourceId, parentPortResourceType);
  85     }
  86
  87     private FilePortConsolidationData addFilePortConsolidationData(String serviceTemplateFileName) {
  88         FilePortConsolidationData consolidationData = getFilePortConsolidationData(serviceTemplateFileName);
  89         if (consolidationData == null) {
  90             consolidationData = new FilePortConsolidationData();
  91             setFilePortConsolidationData(serviceTemplateFileName, consolidationData);
  92         }
  93         return consolidationData;
  94     }
  95 }