1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
package org.openecomp.sdc.translator.services.heattotosca;
import org.onap.sdc.tosca.datatypes.model.NodeTemplate;
import org.onap.sdc.tosca.datatypes.model.ServiceTemplate;
import org.openecomp.sdc.tosca.datatypes.ToscaServiceModel;
import org.openecomp.sdc.tosca.services.DataModelUtil;
import org.openecomp.sdc.tosca.services.ToscaAnalyzerService;
import org.openecomp.sdc.tosca.services.ToscaUtil;
import org.openecomp.sdc.tosca.services.impl.ToscaAnalyzerServiceImpl;
import org.openecomp.sdc.translator.datatypes.heattotosca.TranslationContext;
import org.openecomp.sdc.translator.datatypes.heattotosca.unifiedmodel.consolidation.FileNestedConsolidationData;
import java.util.Map;
import java.util.Objects;
import java.util.Optional;
public class UnifiedCompositionManager {
private ConsolidationService consolidationService;
private TranslationService translationService = new TranslationService();
private UnifiedCompositionService unifiedCompositionService = new UnifiedCompositionService();
public UnifiedCompositionManager(ConsolidationService consolidationService) {
this.consolidationService = consolidationService;
}
public UnifiedCompositionManager() {
}
/**
* Create unified composition.
*
* @param toscaServiceModel the tosca service model
* @param translationContext the translation context
* @return the tosca service model
*/
public ToscaServiceModel createUnifiedComposition(ToscaServiceModel toscaServiceModel,
TranslationContext translationContext) {
Map<String, ServiceTemplate> serviceTemplates = toscaServiceModel.getServiceTemplates();
ServiceTemplate mainServiceTemplate =
serviceTemplates.get(toscaServiceModel.getEntryDefinitionServiceTemplate());
createUnifiedComposition(toscaServiceModel, mainServiceTemplate, translationContext);
ToscaServiceModel unifiedToscaServiceModel =
HeatToToscaUtil.createToscaServiceModel(mainServiceTemplate, translationContext);
return unifiedToscaServiceModel;
}
private void createUnifiedComposition(ToscaServiceModel toscaServiceModel,
ServiceTemplate serviceTemplate,
TranslationContext translationContext) {
handleNestedServiceTemplates(toscaServiceModel, serviceTemplate, translationContext);
consolidationService.serviceTemplateConsolidation(serviceTemplate, translationContext);
unifiedCompositionService
.updateUnifiedAbstractNodesConnectivity(serviceTemplate, translationContext);
translationContext.addUnifiedHandledServiceTeamplte(serviceTemplate);
}
private void handleNestedServiceTemplates(ToscaServiceModel toscaServiceModel,
ServiceTemplate serviceTemplate,
TranslationContext translationContext) {
String serviceTemplateFileName = ToscaUtil.getServiceTemplateFileName(serviceTemplate);
FileNestedConsolidationData fileNestedConsolidationData =
translationContext.getConsolidationData().getNestedConsolidationData()
.getFileNestedConsolidationData(serviceTemplateFileName);
if (Objects.nonNull(fileNestedConsolidationData)) {
ToscaAnalyzerService toscaAnalyzerService = new ToscaAnalyzerServiceImpl();
for (String substitutedNodeTemplateId : fileNestedConsolidationData
.getAllNestedNodeTemplateIds()) {
NodeTemplate nestedNodeTemplate =
DataModelUtil.getNodeTemplate(serviceTemplate, substitutedNodeTemplateId);
if (Objects.isNull(nestedNodeTemplate) || translationContext
.isNestedNodeWasHandled(serviceTemplateFileName, substitutedNodeTemplateId)) {
continue;
}
Optional<String> substituteServiceTemplateName =
toscaAnalyzerService.getSubstituteServiceTemplateName(substitutedNodeTemplateId,
nestedNodeTemplate);
if (substituteServiceTemplateName.isPresent()) {
ServiceTemplate substitutionServiceTemplate =
toscaServiceModel.getServiceTemplates().get(substituteServiceTemplateName.get());
createUnifiedCompositionForNestedServiceTemplate(toscaServiceModel, serviceTemplate,
substitutionServiceTemplate, substitutedNodeTemplateId, translationContext);
}
translationContext.addNestedNodeAsHandled(serviceTemplateFileName,
substitutedNodeTemplateId);
}
}
}
private void createUnifiedCompositionForNestedServiceTemplate(
ToscaServiceModel toscaServiceModel,
ServiceTemplate serviceTemplate,
ServiceTemplate substitutionServiceTemplate,
String substitutedNodeTemplateId,
TranslationContext translationContext) {
handleNestedServiceTemplates(toscaServiceModel, substitutionServiceTemplate,
translationContext);
consolidationService.substitutionServiceTemplateConsolidation(substitutedNodeTemplateId,
serviceTemplate, substitutionServiceTemplate, translationContext);
unifiedCompositionService
.updateUnifiedAbstractNodesConnectivity(substitutionServiceTemplate, translationContext);
translationContext.addUnifiedHandledServiceTeamplte(substitutionServiceTemplate);
}
}
|