openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/main/java/org/openecomp/sdc/translator/datatypes/heattotosca/unifiedmodel/consolidation/ComputeConsolidationDataHandler.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.datatypes.heattotosca.unifiedmodel.consolidation;
  18
  19 import java.util.Objects;
  20 import org.onap.sdc.tosca.datatypes.model.NodeTemplate;
  21 import org.onap.sdc.tosca.datatypes.model.RequirementAssignment;
  22 import org.onap.sdc.tosca.datatypes.model.ServiceTemplate;
  23 import org.openecomp.sdc.heat.datatypes.model.HeatOrchestrationTemplate;
  24 import org.openecomp.sdc.heat.datatypes.model.Resource;
  25 import org.openecomp.sdc.tosca.services.DataModelUtil;
  26 import org.openecomp.sdc.tosca.services.ToscaUtil;
  27 import org.openecomp.sdc.translator.datatypes.heattotosca.TranslationContext;
  28 import org.openecomp.sdc.translator.datatypes.heattotosca.to.TranslateTo;
  29 import org.openecomp.sdc.translator.services.heattotosca.NameExtractor;
  30
  31 public class ComputeConsolidationDataHandler implements ConsolidationDataHandler {
  32
  33     private final ComputeConsolidationData computeConsolidationData;
  34
  35     ComputeConsolidationDataHandler(ComputeConsolidationData computeConsolidationData) {
  36         this.computeConsolidationData = computeConsolidationData;
  37     }
  38
  39     @Override
  40     public void addNodesConnectedOut(TranslateTo translateTo, String nodeTemplateId, String requirementId,
  41                                      RequirementAssignment requirementAssignment) {
  42
  43         String translatedSourceNodeId = translateTo.getTranslatedId();
  44         ServiceTemplate serviceTemplate = translateTo.getServiceTemplate();
  45         NodeTemplate computeNodeTemplate = DataModelUtil.getNodeTemplate(serviceTemplate, translatedSourceNodeId);
  46         String nodeType = computeNodeTemplate.getType();
  47
  48         EntityConsolidationData entityConsolidationData =
  49                 getComputeTemplateConsolidationData(translateTo, nodeType, translatedSourceNodeId);
  50
  51         if (Objects.nonNull(entityConsolidationData)) {
  52             entityConsolidationData.addNodesConnectedOut(nodeTemplateId, requirementId, requirementAssignment);
  53         }
  54     }
  55
  56     @Override
  57     public void addNodesConnectedIn(TranslateTo translateTo, String sourceNodeTemplateId,
  58                                     String dependentNodeTemplateId, String targetResourceId, String requirementId,
  59                                     RequirementAssignment requirementAssignment) {
  60
  61         ServiceTemplate serviceTemplate = translateTo.getServiceTemplate();
  62         NodeTemplate nodeTemplate = DataModelUtil.getNodeTemplate(serviceTemplate, dependentNodeTemplateId);
  63         String nodeType = getNodeType(nodeTemplate, translateTo, targetResourceId, dependentNodeTemplateId);
  64         EntityConsolidationData entityConsolidationData =
  65                 getComputeTemplateConsolidationData(translateTo, nodeType, dependentNodeTemplateId);
  66
  67         if (Objects.nonNull(entityConsolidationData)) {
  68             entityConsolidationData.addNodesConnectedIn(sourceNodeTemplateId, requirementId, requirementAssignment);
  69         }
  70     }
  71
  72     @Override
  73     public void removeParamNameFromAttrFuncList(ServiceTemplate serviceTemplate,
  74             HeatOrchestrationTemplate heatOrchestrationTemplate, String paramName, String contrailSharedResourceId,
  75             String sharedTranslatedResourceId) {
  76
  77         NodeTemplate nodeTemplate = DataModelUtil.getNodeTemplate(serviceTemplate, sharedTranslatedResourceId);
  78         EntityConsolidationData entityConsolidationData =
  79                 getComputeTemplateConsolidationData(serviceTemplate, nodeTemplate.getType(),
  80                         sharedTranslatedResourceId);
  81
  82         if (Objects.nonNull(entityConsolidationData)) {
  83             entityConsolidationData.removeParamNameFromAttrFuncList(paramName);
  84         }
  85     }
  86
  87     private ComputeTemplateConsolidationData getComputeTemplateConsolidationData(
  88             TranslateTo translateTo, String computeNodeType, String computeNodeTemplateId) {
  89
  90         ServiceTemplate serviceTemplate = translateTo.getServiceTemplate();
  91         return getComputeTemplateConsolidationData(serviceTemplate, computeNodeType, computeNodeTemplateId);
  92     }
  93
  94     private ComputeTemplateConsolidationData getComputeTemplateConsolidationData(ServiceTemplate serviceTemplate,
  95             String computeNodeType, String computeNodeTemplateId) {
  96
  97         String serviceTemplateFileName = ToscaUtil.getServiceTemplateFileName(serviceTemplate);
  98
  99         FileComputeConsolidationData fileComputeConsolidationData =
 100                 computeConsolidationData.getFileComputeConsolidationData(serviceTemplateFileName);
 101         if (fileComputeConsolidationData == null) {
 102             fileComputeConsolidationData = new FileComputeConsolidationData();
 103             computeConsolidationData.setFileComputeConsolidationData(serviceTemplateFileName,
 104                     fileComputeConsolidationData);
 105         }
 106
 107         TypeComputeConsolidationData typeComputeConsolidationData =
 108                 fileComputeConsolidationData.getTypeComputeConsolidationData(computeNodeType);
 109         if (typeComputeConsolidationData == null) {
 110             typeComputeConsolidationData = new TypeComputeConsolidationData();
 111             fileComputeConsolidationData.setTypeComputeConsolidationData(computeNodeType, typeComputeConsolidationData);
 112         }
 113
 114         ComputeTemplateConsolidationData computeTemplateConsolidationData =
 115                 typeComputeConsolidationData.getComputeTemplateConsolidationData(computeNodeTemplateId);
 116         if (computeTemplateConsolidationData == null) {
 117             computeTemplateConsolidationData = new ComputeTemplateConsolidationData();
 118             computeTemplateConsolidationData.setNodeTemplateId(computeNodeTemplateId);
 119             typeComputeConsolidationData.setComputeTemplateConsolidationData(computeNodeTemplateId,
 120                     computeTemplateConsolidationData);
 121         }
 122
 123         return computeTemplateConsolidationData;
 124     }
 125
 126     private String getNodeType(NodeTemplate computeNodeTemplate, TranslateTo translateTo, String targetResourceId,
 127             String nodeTemplateId) {
 128
 129         if (Objects.isNull(computeNodeTemplate)) {
 130             Resource targetResource = translateTo.getHeatOrchestrationTemplate().getResources().get(targetResourceId);
 131             NameExtractor nodeTypeNameExtractor = TranslationContext.getNameExtractorImpl(targetResource.getType());
 132             return nodeTypeNameExtractor.extractNodeTypeName(translateTo.getHeatOrchestrationTemplate()
 133                                     .getResources().get(nodeTemplateId), nodeTemplateId, nodeTemplateId);
 134         }
 135
 136         return computeNodeTemplate.getType();
 137     }
 138 }