openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/main/java/org/openecomp/sdc/translator/services/heattotosca/UnifiedCompositionManager.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * SDC
   4  * ================================================================================
   5  * Copyright (C) 2019 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20 package org.openecomp.sdc.translator.services.heattotosca;
  21
  22 import java.util.Map;
  23 import java.util.Objects;
  24 import java.util.Optional;
  25 import org.onap.sdc.tosca.datatypes.model.NodeTemplate;
  26 import org.onap.sdc.tosca.datatypes.model.ServiceTemplate;
  27 import org.openecomp.sdc.tosca.datatypes.ToscaServiceModel;
  28 import org.openecomp.sdc.tosca.services.DataModelUtil;
  29 import org.openecomp.sdc.tosca.services.ToscaAnalyzerService;
  30 import org.openecomp.sdc.tosca.services.ToscaUtil;
  31 import org.openecomp.sdc.tosca.services.impl.ToscaAnalyzerServiceImpl;
  32 import org.openecomp.sdc.translator.datatypes.heattotosca.TranslationContext;
  33 import org.openecomp.sdc.translator.datatypes.heattotosca.unifiedmodel.consolidation.FileNestedConsolidationData;
  34
  35 public class UnifiedCompositionManager {
  36
  37     private ConsolidationService consolidationService;
  38     private TranslationService translationService = new TranslationService();
  39     private UnifiedCompositionService unifiedCompositionService = new UnifiedCompositionService();
  40
  41     public UnifiedCompositionManager(ConsolidationService consolidationService) {
  42         this.consolidationService = consolidationService;
  43     }
  44
  45     public UnifiedCompositionManager() {
  46     }
  47
  48     /**
  49      * Create unified composition.
  50      *
  51      * @param toscaServiceModel  the tosca service model
  52      * @param translationContext the translation context
  53      * @return the tosca service model
  54      */
  55     public ToscaServiceModel createUnifiedComposition(ToscaServiceModel toscaServiceModel, TranslationContext translationContext) {
  56         Map<String, ServiceTemplate> serviceTemplates = toscaServiceModel.getServiceTemplates();
  57         ServiceTemplate mainServiceTemplate = serviceTemplates.get(toscaServiceModel.getEntryDefinitionServiceTemplate());
  58         createUnifiedComposition(toscaServiceModel, mainServiceTemplate, translationContext);
  59         ToscaServiceModel unifiedToscaServiceModel = HeatToToscaUtil.createToscaServiceModel(mainServiceTemplate, translationContext);
  60         return unifiedToscaServiceModel;
  61     }
  62
  63     private void createUnifiedComposition(ToscaServiceModel toscaServiceModel, ServiceTemplate serviceTemplate,
  64                                           TranslationContext translationContext) {
  65         handleNestedServiceTemplates(toscaServiceModel, serviceTemplate, translationContext);
  66         consolidationService.serviceTemplateConsolidation(serviceTemplate, translationContext);
  67         unifiedCompositionService.updateUnifiedAbstractNodesConnectivity(serviceTemplate, translationContext);
  68         translationContext.addUnifiedHandledServiceTeamplte(serviceTemplate);
  69     }
  70
  71     private void handleNestedServiceTemplates(ToscaServiceModel toscaServiceModel, ServiceTemplate serviceTemplate,
  72                                               TranslationContext translationContext) {
  73         String serviceTemplateFileName = ToscaUtil.getServiceTemplateFileName(serviceTemplate);
  74         FileNestedConsolidationData fileNestedConsolidationData = translationContext.getConsolidationData().getNestedConsolidationData()
  75             .getFileNestedConsolidationData(serviceTemplateFileName);
  76         if (Objects.nonNull(fileNestedConsolidationData)) {
  77             ToscaAnalyzerService toscaAnalyzerService = new ToscaAnalyzerServiceImpl();
  78             for (String substitutedNodeTemplateId : fileNestedConsolidationData.getAllNestedNodeTemplateIds()) {
  79                 NodeTemplate nestedNodeTemplate = DataModelUtil.getNodeTemplate(serviceTemplate, substitutedNodeTemplateId);
  80                 if (Objects.isNull(nestedNodeTemplate) || translationContext
  81                     .isNestedNodeWasHandled(serviceTemplateFileName, substitutedNodeTemplateId)) {
  82                     continue;
  83                 }
  84                 Optional<String> substituteServiceTemplateName = toscaAnalyzerService
  85                     .getSubstituteServiceTemplateName(substitutedNodeTemplateId, nestedNodeTemplate);
  86                 if (substituteServiceTemplateName.isPresent()) {
  87                     ServiceTemplate substitutionServiceTemplate = toscaServiceModel.getServiceTemplates().get(substituteServiceTemplateName.get());
  88                     createUnifiedCompositionForNestedServiceTemplate(toscaServiceModel, serviceTemplate, substitutionServiceTemplate,
  89                         substitutedNodeTemplateId, translationContext);
  90                 }
  91                 translationContext.addNestedNodeAsHandled(serviceTemplateFileName, substitutedNodeTemplateId);
  92             }
  93         }
  94     }
  95
  96     private void createUnifiedCompositionForNestedServiceTemplate(ToscaServiceModel toscaServiceModel, ServiceTemplate serviceTemplate,
  97                                                                   ServiceTemplate substitutionServiceTemplate, String substitutedNodeTemplateId,
  98                                                                   TranslationContext translationContext) {
  99         handleNestedServiceTemplates(toscaServiceModel, substitutionServiceTemplate, translationContext);
 100         consolidationService
 101             .substitutionServiceTemplateConsolidation(substitutedNodeTemplateId, serviceTemplate, substitutionServiceTemplate, translationContext);
 102         unifiedCompositionService.updateUnifiedAbstractNodesConnectivity(substitutionServiceTemplate, translationContext);
 103         translationContext.addUnifiedHandledServiceTeamplte(substitutionServiceTemplate);
 104     }
 105 }