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.
  98      *
  99      */
 100     public void addPortToConsolidationData(TranslateTo translateTo, String computeNodeType,
 101                 String computeNodeTemplateId, String portType, String portNodeTemplateId) {
 102         ComputeTemplateConsolidationData consolidationData =
 103                 getComputeTemplateConsolidationData(translateTo, computeNodeType, computeNodeTemplateId);
 104         consolidationData.addPort(portType, portNodeTemplateId);
 105     }
 106
 107     /**
 108      * Add volume to consolidation data.
 109      *
 110      */
 111     public void addVolumeToConsolidationData(TranslateTo translateTo, String computeNodeType,
 112             String computeNodeTemplateId, String requirementId, RequirementAssignment requirementAssignment) {
 113         ComputeTemplateConsolidationData consolidationData =
 114                 getComputeTemplateConsolidationData(translateTo, computeNodeType,
 115                         computeNodeTemplateId);
 116         consolidationData.addVolume(requirementId, requirementAssignment);
 117     }
 118
 119     /**
 120      * Add group id information to consolidation data.
 121      *
 122      * @param translatedGroupId Group id of which compute node is a part
 123      */
 124
 125     public void addGroupIdToConsolidationData(TranslateTo translateTo, String computeNodeType,
 126             String computeNodeTemplateId, String translatedGroupId) {
 127         ComputeTemplateConsolidationData consolidationData =
 128                 getComputeTemplateConsolidationData(translateTo, computeNodeType,
 129                         computeNodeTemplateId);
 130         consolidationData.addGroupId(translatedGroupId);
 131     }
 132
 133     private ComputeTemplateConsolidationData getComputeTemplateConsolidationData(
 134             TranslateTo translateTo, String computeNodeType, String computeNodeTemplateId) {
 135         ServiceTemplate serviceTemplate = translateTo.getServiceTemplate();
 136         String serviceTemplateFileName = ToscaUtil.getServiceTemplateFileName(serviceTemplate);
 137         return getComputeTemplateConsolidationData(serviceTemplateFileName, computeNodeType, computeNodeTemplateId);
 138     }
 139
 140     private ComputeTemplateConsolidationData getComputeTemplateConsolidationData(
 141             String serviceTemplateFileName, String computeNodeType, String computeNodeTemplateId) {
 142
 143         return computeConsolidationData.addComputeTemplateConsolidationData(
 144                         serviceTemplateFileName, computeNodeType, computeNodeTemplateId);
 145
 146     }
 147
 148     private String getNodeType(NodeTemplate computeNodeTemplate, TranslateTo translateTo,
 149                                      String targetResourceId, String nodeTemplateId) {
 150
 151         if (Objects.isNull(computeNodeTemplate)) {
 152             Resource targetResource = translateTo.getHeatOrchestrationTemplate().getResources().get(targetResourceId);
 153             NameExtractor nodeTypeNameExtractor = TranslationContext.getNameExtractorImpl(targetResource.getType());
 154             return nodeTypeNameExtractor.extractNodeTypeName(translateTo.getHeatOrchestrationTemplate()
 155                                         .getResources().get(nodeTemplateId), nodeTemplateId, nodeTemplateId);
 156         }
 157
 158         return computeNodeTemplate.getType();
 159     }
 160
 161     public boolean isNumberOfComputeTypesLegal(String serviceTemplateName) {
 162         return computeConsolidationData.isNumberOfComputeTypesLegal(serviceTemplateName);
 163     }
 164 }