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.Objects;
  22 import java.util.Set;
  23
  24 public class ComputeConsolidationData {
  25
  26     //Key - Service template file name
  27     private final Map<String, FileComputeConsolidationData> fileComputeConsolidationData;
  28
  29     public ComputeConsolidationData() {
  30         fileComputeConsolidationData = new HashMap<>();
  31     }
  32
  33     public Set<String> getAllServiceTemplateFileNames() {
  34         return fileComputeConsolidationData.keySet();
  35     }
  36
  37     public FileComputeConsolidationData getFileComputeConsolidationData(String serviceTemplateFileName) {
  38         return fileComputeConsolidationData.get(serviceTemplateFileName);
  39     }
  40
  41     public void setFileComputeConsolidationData(String serviceTemplateFileName,
  42               FileComputeConsolidationData fileComputeConsolidationData) {
  43         this.fileComputeConsolidationData.put(serviceTemplateFileName, fileComputeConsolidationData);
  44     }
  45
  46     /**
  47      * add compute template consolidation data entity if it doesn't exist yet
  48      * base on given parameters.
  49      *
  50      * @return compute template consolidation data entity by given keys
  51     */
  52     ComputeTemplateConsolidationData addComputeTemplateConsolidationData(
  53             String serviceTemplateFileName, String computeNodeType, String computeNodeTemplateId) {
  54         FileComputeConsolidationData consolidationData = addFileComputeConsolidationData(serviceTemplateFileName);
  55         return consolidationData.addComputeTemplateConsolidationData(computeNodeType, computeNodeTemplateId);
  56     }
  57
  58     private FileComputeConsolidationData addFileComputeConsolidationData(String serviceTemplateFileName) {
  59         FileComputeConsolidationData consolidationData = getFileComputeConsolidationData(serviceTemplateFileName);
  60         if (consolidationData == null) {
  61             consolidationData = new FileComputeConsolidationData();
  62             setFileComputeConsolidationData(serviceTemplateFileName, consolidationData);
  63         }
  64         return consolidationData;
  65     }
  66
  67     /**
  68      * Is number of compute types legal.
  69      *
  70      * @param serviceTemplateName the service template name
  71      * @return the boolean
  72      */
  73     public boolean isNumberOfComputeTypesLegal(String serviceTemplateName) {
  74         FileComputeConsolidationData fileComputeConsolidationData =
  75                 getFileComputeConsolidationData(serviceTemplateName);
  76         return Objects.nonNull(fileComputeConsolidationData)
  77                        && fileComputeConsolidationData.isNumberOfComputeTypesLegal();
  78     }
  79 }