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.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(ToscaUtil
  80                     .getServiceTemplateFileName(serviceTemplate), nodeTemplate.getType(), sharedTranslatedResourceId);
  81
  82         if (Objects.nonNull(entityConsolidationData)) {
  83             entityConsolidationData.removeParamNameFromAttrFuncList(paramName);
  84         }
  85     }
  86
  87     /**
  88      * Add compute in consolidation data entity base on given keys.
  89      *
  90      */
  91     public void addConsolidationData(String serviceTemplateFileName,
  92             String computeNodeType, String computeNodeTemplateId) {
  93         getComputeTemplateConsolidationData(serviceTemplateFileName, computeNodeType,  computeNodeTemplateId);
  94     }
  95
  96     /**
  97      * Add port to compute consolidation data entity base on given keys.s
  98      *
  99      */
 100     public void addPortToConsolidationData(TranslateTo translateTo, String computeNodeType,
 101                 String computeNodeTemplateId, String portType, String portNodeTemplateId) {
 102         ComputeTemplateConsolidationData entityConsolidationData =
 103                 getComputeTemplateConsolidationData(translateTo, computeNodeType, computeNodeTemplateId);
 104         entityConsolidationData.addPort(portType, portNodeTemplateId);
 105     }
 106
 107     private ComputeTemplateConsolidationData getComputeTemplateConsolidationData(
 108             TranslateTo translateTo, String computeNodeType, String computeNodeTemplateId) {
 109
 110         ServiceTemplate serviceTemplate = translateTo.getServiceTemplate();
 111         String serviceTemplateFileName = ToscaUtil.getServiceTemplateFileName(serviceTemplate);
 112         return getComputeTemplateConsolidationData(serviceTemplateFileName, computeNodeType, computeNodeTemplateId);
 113
 114     }
 115
 116
 117     private ComputeTemplateConsolidationData getComputeTemplateConsolidationData(
 118             String serviceTemplateFileName, String computeNodeType, String computeNodeTemplateId) {
 119
 120         return computeConsolidationData.addComputeTemplateConsolidationData(
 121                         serviceTemplateFileName, computeNodeType, computeNodeTemplateId);
 122
 123     }
 124
 125     private String getNodeType(NodeTemplate computeNodeTemplate, TranslateTo translateTo,
 126                                      String targetResourceId, String nodeTemplateId) {
 127
 128         if (Objects.isNull(computeNodeTemplate)) {
 129             Resource targetResource = translateTo.getHeatOrchestrationTemplate().getResources().get(targetResourceId);
 130             NameExtractor nodeTypeNameExtractor = TranslationContext.getNameExtractorImpl(targetResource.getType());
 131             return nodeTypeNameExtractor.extractNodeTypeName(translateTo.getHeatOrchestrationTemplate()
 132                                         .getResources().get(nodeTemplateId), nodeTemplateId, nodeTemplateId);
 133         }
 134
 135         return computeNodeTemplate.getType();
 136     }
 137 }