gerrit.onap Code Review - sdc.git/blob

   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 ComputeConsolidationData {
  24
  25     //Key - Service template file name
  26     private final Map<String, FileComputeConsolidationData> fileComputeConsolidationData;
  27
  28     public ComputeConsolidationData() {
  29         fileComputeConsolidationData = new HashMap<>();
  30     }
  31
  32     public Set<String> getAllServiceTemplateFileNames() {
  33         return fileComputeConsolidationData.keySet();
  34     }
  35
  36     public FileComputeConsolidationData getFileComputeConsolidationData(String serviceTemplateFileName) {
  37         return fileComputeConsolidationData.get(serviceTemplateFileName);
  38     }
  39
  40     public void setFileComputeConsolidationData(String serviceTemplateFileName,
  41               FileComputeConsolidationData fileComputeConsolidationData) {
  42         this.fileComputeConsolidationData.put(serviceTemplateFileName, fileComputeConsolidationData);
  43     }
  44
  45     /**
  46      * add compute template consolidation data entity if it doesn't exist yet
  47      * base on given parameters.
  48      *
  49      * @return compute template consolidation data entity by given keys
  50     */
  51     ComputeTemplateConsolidationData addComputeTemplateConsolidationData(
  52             String serviceTemplateFileName, String computeNodeType, String computeNodeTemplateId) {
  53         FileComputeConsolidationData consolidationData = addFileComputeConsolidationData(serviceTemplateFileName);
  54         return consolidationData.addComputeTemplateConsolidationData(computeNodeType, computeNodeTemplateId);
  55     }
  56
  57     private FileComputeConsolidationData addFileComputeConsolidationData(String serviceTemplateFileName) {
  58         FileComputeConsolidationData consolidationData = getFileComputeConsolidationData(serviceTemplateFileName);
  59         if (consolidationData == null) {
  60             consolidationData = new FileComputeConsolidationData();
  61             setFileComputeConsolidationData(serviceTemplateFileName, consolidationData);
  62         }
  63         return consolidationData;
  64     }
  65 }