openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/main/java/org/openecomp/sdc/translator/services/heattotosca/ConsolidationEntityType.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.services.heattotosca;
  18
  19 import static org.openecomp.sdc.translator.services.heattotosca.ConsolidationDataUtil.isComputeResource;
  20 import static org.openecomp.sdc.translator.services.heattotosca.ConsolidationDataUtil.isPortResource;
  21 import static org.openecomp.sdc.translator.services.heattotosca.ConsolidationDataUtil.isVolumeResource;
  22
  23 import org.openecomp.sdc.heat.datatypes.model.Resource;
  24 import org.openecomp.sdc.translator.datatypes.heattotosca.TranslationContext;
  25
  26 import java.util.Arrays;
  27 import java.util.Collections;
  28 import java.util.HashSet;
  29 import java.util.Set;
  30 import java.util.stream.Collectors;
  31
  32 /**
  33  * The enum Entity type.
  34  */
  35 public enum ConsolidationEntityType {
  36   COMPUTE,
  37   PORT,
  38   VOLUME,
  39   NESTED,
  40   //Simple nested VFC (nested file with one compute) or a complex VFC (nested ST with more than
  41   //one compute)
  42   VFC_NESTED,
  43   SUB_INTERFACE,
  44   OTHER;
  45
  46   private ConsolidationEntityType sourceEntityType;
  47   private ConsolidationEntityType targetEntityType;
  48
  49   public ConsolidationEntityType getSourceEntityType() {
  50     return sourceEntityType;
  51   }
  52
  53   public ConsolidationEntityType getTargetEntityType() {
  54     return targetEntityType;
  55   }
  56
  57
  58   /**
  59    * Sets entity type.
  60    *
  61    * @param sourceResource          the source resource
  62    * @param targetResource          the target resource
  63    */
  64   public void setEntityType(Resource sourceResource,
  65                             Resource targetResource,
  66                             TranslationContext context) {
  67     targetEntityType =
  68         getEntityType(targetResource, context);
  69     sourceEntityType =
  70         getEntityType(sourceResource, context);
  71   }
  72
  73   private static final Set<ConsolidationEntityType> consolidationEntityTypes = initConsolidationEntities();
  74
  75   private static Set<ConsolidationEntityType> initConsolidationEntities() {
  76     Set<ConsolidationEntityType> consolidationEntityTypes = new HashSet<>(Arrays.asList(ConsolidationEntityType
  77         .values()));
  78     return Collections.unmodifiableSet(consolidationEntityTypes.stream()
  79         .filter(entity -> entity != ConsolidationEntityType.OTHER
  80             && entity != ConsolidationEntityType.VOLUME)
  81         .collect(Collectors.toSet()));
  82   }
  83
  84   public static Set<ConsolidationEntityType> getSupportedConsolidationEntities() {
  85     return consolidationEntityTypes;
  86   }
  87
  88   private ConsolidationEntityType getEntityType(Resource resource, TranslationContext context) {
  89     ConsolidationEntityType consolidationEntityType = ConsolidationEntityType.OTHER;
  90     if (isComputeResource(resource)) {
  91       consolidationEntityType = ConsolidationEntityType.COMPUTE;
  92     } else if (isPortResource(resource)) {
  93       consolidationEntityType = ConsolidationEntityType.PORT;
  94     } else if (HeatToToscaUtil.isSubInterfaceResource(resource, context)) {
  95       consolidationEntityType = ConsolidationEntityType.SUB_INTERFACE;
  96     } else if (isVolumeResource(resource)) {
  97       consolidationEntityType = ConsolidationEntityType.VOLUME;
  98     } else if (HeatToToscaUtil.isNestedResource(resource)) {
  99       if (HeatToToscaUtil.isNestedVfcResource(resource, context)) {
 100         consolidationEntityType = ConsolidationEntityType.VFC_NESTED;
 101       } else {
 102         consolidationEntityType = ConsolidationEntityType.NESTED;
 103       }
 104     }
 105     return consolidationEntityType;
 106   }
 107 }