summaryrefslogtreecommitdiffstats
path: root/openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/main/java/org/openecomp/sdc/translator/services/heattotosca/ConsolidationEntityType.java
blob: c2de58fd5acc01f46f25a40f8fc52a270b433cf1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
package org.openecomp.sdc.translator.services.heattotosca;

import static org.openecomp.sdc.translator.services.heattotosca.ConsolidationDataUtil.isComputeResource;
import static org.openecomp.sdc.translator.services.heattotosca.ConsolidationDataUtil.isPortResource;
import static org.openecomp.sdc.translator.services.heattotosca.ConsolidationDataUtil.isVolumeResource;

import org.apache.commons.lang3.ObjectUtils;
import org.openecomp.core.utilities.yaml.YamlUtil;
import org.openecomp.sdc.heat.datatypes.manifest.FileData;
import org.openecomp.sdc.heat.datatypes.model.HeatOrchestrationTemplate;
import org.openecomp.sdc.heat.datatypes.model.Resource;
import org.openecomp.sdc.translator.datatypes.heattotosca.TranslationContext;

import java.util.Collection;
import java.util.Objects;
import java.util.Optional;

/**
 * The enum Entity type.
 */
public enum ConsolidationEntityType {
  COMPUTE,
  PORT,
  VOLUME,
  NESTED,
  VFC_NESTED,
  SUB_PORT,
  OTHER;

  private ConsolidationEntityType sourceEntityType;
  private ConsolidationEntityType targetEntityType;

  public ConsolidationEntityType getSourceEntityType() {
    return sourceEntityType;
  }

  public ConsolidationEntityType getTargetEntityType() {
    return targetEntityType;
  }

  /**
   * Sets entity type.
   *
   * @param heatOrchestrationTemplate the heat orchestration template
   * @param sourceResource          the source resource
   * @param targetResource          the target resource
   */
  public void setEntityType(HeatOrchestrationTemplate heatOrchestrationTemplate,
                            Resource sourceResource,
                            Resource targetResource,
                            TranslationContext context) {
    targetEntityType =
        getEntityType(heatOrchestrationTemplate, targetResource, context);
    sourceEntityType =
        getEntityType(heatOrchestrationTemplate, sourceResource, context);
  }

  private ConsolidationEntityType getEntityType(HeatOrchestrationTemplate heatOrchestrationTemplate,
                                                Resource resource, TranslationContext context) {
    if (isComputeResource(resource)) {
      return ConsolidationEntityType.COMPUTE;
    } else if (isPortResource(resource)) {
      return ConsolidationEntityType.PORT;
    } else if (isVolumeResource(resource)) {
      return ConsolidationEntityType.VOLUME;
    } else if (HeatToToscaUtil.isNestedResource(resource)) {
      Optional<String> nestedHeatFileName = HeatToToscaUtil.getNestedHeatFileName(resource);
      if (nestedHeatFileName.isPresent()) {
        HeatOrchestrationTemplate nestedHeatOrchestrationTemplate = new YamlUtil()
            .yamlToObject(context.getFileContent(nestedHeatFileName.get()),
                HeatOrchestrationTemplate.class);
        if (Objects.nonNull(nestedHeatOrchestrationTemplate.getResources())) {
          for (String innerResourceId : nestedHeatOrchestrationTemplate.getResources().keySet()) {
            if (ConsolidationDataUtil
                .isComputeResource(nestedHeatOrchestrationTemplate, innerResourceId)) {
              return ConsolidationEntityType.VFC_NESTED;
            }
          }
        }
      }
      return ConsolidationEntityType.NESTED;
    } else {
      return ConsolidationEntityType.OTHER;
    }
  }
}