openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/main/java/org/openecomp/sdc/translator/datatypes/heattotosca/unifiedmodel/consolidation/NestedConsolidationDataHandler.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.RequirementAssignment;
  21 import org.onap.sdc.tosca.datatypes.model.ServiceTemplate;
  22 import org.openecomp.core.utilities.file.FileUtils;
  23 import org.openecomp.sdc.common.errors.CoreException;
  24 import org.openecomp.sdc.heat.datatypes.model.HeatOrchestrationTemplate;
  25 import org.openecomp.sdc.tosca.services.ToscaUtil;
  26 import org.openecomp.sdc.translator.datatypes.heattotosca.TranslationContext;
  27 import org.openecomp.sdc.translator.datatypes.heattotosca.to.TranslateTo;
  28 import org.openecomp.sdc.translator.services.heattotosca.errors.DuplicateResourceIdsInDifferentFilesErrorBuilder;
  29
  30 public class NestedConsolidationDataHandler implements ConsolidationDataHandler {
  31
  32     private final NestedConsolidationData nestedConsolidationData;
  33
  34     public NestedConsolidationDataHandler(NestedConsolidationData nestedConsolidationData) {
  35         this.nestedConsolidationData = nestedConsolidationData;
  36     }
  37
  38     @Override
  39     public void addNodesConnectedOut(TranslateTo translateTo, String nodeTemplateId, String requirementId,
  40                                             RequirementAssignment requirementAssignment) {
  41         EntityConsolidationData entityConsolidationData =
  42                 getNestedTemplateConsolidationData(translateTo, translateTo.getHeatFileName(),
  43                         translateTo.getTranslatedId());
  44
  45         if (Objects.nonNull(entityConsolidationData)) {
  46             entityConsolidationData.addNodesConnectedOut(nodeTemplateId, requirementId, requirementAssignment);
  47         }
  48     }
  49
  50     @Override
  51     public void addNodesConnectedIn(TranslateTo translateTo, String sourceNodeTemplateId,
  52                                            String dependentNodeTemplateId, String targetResourceId,
  53                                            String requirementId, RequirementAssignment requirementAssignment) {
  54
  55         EntityConsolidationData entityConsolidationData =
  56                 getNestedTemplateConsolidationData(translateTo, translateTo.getHeatFileName(), dependentNodeTemplateId);
  57
  58         if (Objects.nonNull(entityConsolidationData)) {
  59             entityConsolidationData.addNodesConnectedIn(sourceNodeTemplateId, requirementId, requirementAssignment);
  60         }
  61
  62     }
  63
  64     @Override
  65     public void removeParamNameFromAttrFuncList(ServiceTemplate serviceTemplate,
  66                                                        HeatOrchestrationTemplate heatOrchestrationTemplate,
  67                                                        String paramName,
  68                                                        String contrailSharedResourceId,
  69                                                        String sharedTranslatedResourceId) {
  70
  71         throw new UnsupportedOperationException(
  72                 "API removeParamNameFromAttrFuncList not supported for NestedConsolidationDataHandler");
  73
  74     }
  75
  76     private NestedTemplateConsolidationData getNestedTemplateConsolidationData(TranslateTo translateTo,
  77                                                                                       String nestedHeatFileName,
  78                                                                                       String nestedNodeTemplateId) {
  79         ServiceTemplate serviceTemplate = translateTo.getServiceTemplate();
  80         TranslationContext context = translateTo.getContext();
  81         return getNestedTemplateConsolidationData(serviceTemplate, context, nestedHeatFileName, nestedNodeTemplateId);
  82     }
  83
  84     private NestedTemplateConsolidationData getNestedTemplateConsolidationData(
  85                                 ServiceTemplate serviceTemplate,  TranslationContext context,
  86                                 String nestedHeatFileName, String nestedNodeTemplateId) {
  87
  88
  89         if (isNestedResourceIdOccursInDifferentNestedFiles(context, nestedHeatFileName,
  90                 nestedNodeTemplateId)) {
  91             throw new CoreException(new
  92                                             DuplicateResourceIdsInDifferentFilesErrorBuilder(
  93                                                     nestedNodeTemplateId).build());
  94         }
  95
  96         if (isNodeTemplatePointsToServiceTemplateWithoutNodeTemplates(
  97                 nestedNodeTemplateId, nestedHeatFileName, context)) {
  98             return null;
  99         }
 100
 101         String serviceTemplateFileName = ToscaUtil.getServiceTemplateFileName(serviceTemplate);
 102         FileNestedConsolidationData fileNestedConsolidationData = nestedConsolidationData
 103                                                .getFileNestedConsolidationData(serviceTemplateFileName);
 104
 105         if (fileNestedConsolidationData == null) {
 106             fileNestedConsolidationData = new FileNestedConsolidationData();
 107             nestedConsolidationData.setFileNestedConsolidationData(serviceTemplateFileName,
 108                     fileNestedConsolidationData);
 109         }
 110
 111         NestedTemplateConsolidationData nestedTemplateConsolidationData =
 112                 fileNestedConsolidationData.getNestedTemplateConsolidationData(nestedNodeTemplateId);
 113         if (nestedTemplateConsolidationData == null) {
 114             nestedTemplateConsolidationData = new NestedTemplateConsolidationData();
 115             nestedTemplateConsolidationData.setNodeTemplateId(nestedNodeTemplateId);
 116             fileNestedConsolidationData.setNestedTemplateConsolidationData(nestedNodeTemplateId,
 117                     nestedTemplateConsolidationData);
 118         }
 119
 120         return nestedTemplateConsolidationData;
 121     }
 122
 123     private boolean isNodeTemplatePointsToServiceTemplateWithoutNodeTemplates(String nestedNodeTemplateId,
 124                                                                                      String nestedHeatFileName,
 125                                                                                      TranslationContext context) {
 126
 127         return context.isServiceTemplateWithoutNodeTemplatesSection(
 128                 FileUtils.getFileWithoutExtention(nestedHeatFileName))
 129                        || context.isNodeTemplateIdPointsToStWithoutNodeTemplates(nestedNodeTemplateId);
 130     }
 131
 132     private boolean isNestedResourceIdOccursInDifferentNestedFiles(TranslationContext context,
 133                                                                           String nestedHeatFileName,
 134                                                                           String nestedNodeTemplateId) {
 135         return Objects.nonNull(nestedHeatFileName) && context.getAllTranslatedResourceIdsFromDiffNestedFiles(
 136                 nestedHeatFileName).contains(nestedNodeTemplateId);
 137     }
 138 }