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.Collection;
  20 import java.util.HashMap;
  21 import java.util.Map;
  22 import java.util.Set;
  23
  24 public class FileComputeConsolidationData {
  25
  26     //key - compute node type name (vm_type)
  27     private final Map<String, TypeComputeConsolidationData> typeComputeConsolidationData;
  28
  29     public FileComputeConsolidationData() {
  30         typeComputeConsolidationData = new HashMap<>();
  31     }
  32
  33     public Set<String> getAllComputeTypes() {
  34         return typeComputeConsolidationData.keySet();
  35     }
  36
  37     public Collection<TypeComputeConsolidationData> getAllTypeComputeConsolidationData() {
  38         return typeComputeConsolidationData.values();
  39     }
  40
  41     public TypeComputeConsolidationData getTypeComputeConsolidationData(String computeType) {
  42         return typeComputeConsolidationData.get(computeType);
  43     }
  44
  45     public void setTypeComputeConsolidationData(String computeType, TypeComputeConsolidationData
  46             typeComputeConsolidationData) {
  47         this.typeComputeConsolidationData.put(computeType, typeComputeConsolidationData);
  48     }
  49
  50
  51     /**
  52     * add compute template consolidation data according to given key if it doesn't exist yet.
  53     *
  54     * @return compute template consolidation data by given keys
  55     */
  56     ComputeTemplateConsolidationData addComputeTemplateConsolidationData(
  57                       String computeType, String computeNodeTemplateId) {
  58
  59         TypeComputeConsolidationData consolidationData = addTypeComputeConsolidationData(computeType);
  60         return consolidationData.addComputeTemplateConsolidationData(computeNodeTemplateId);
  61     }
  62
  63     private TypeComputeConsolidationData addTypeComputeConsolidationData(String computeType) {
  64         TypeComputeConsolidationData consolidationData = getTypeComputeConsolidationData(computeType);
  65         if (consolidationData == null) {
  66             consolidationData = new TypeComputeConsolidationData();
  67             setTypeComputeConsolidationData(computeType, consolidationData);
  68         }
  69         return consolidationData;
  70     }
  71 }