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
107
108
|
package org.openecomp.core.impl;
import org.apache.commons.collections4.MapUtils;
import org.openecomp.sdc.logging.api.Logger;
import org.openecomp.sdc.logging.api.LoggerFactory;
import org.openecomp.sdc.tosca.datatypes.model.Import;
import org.openecomp.sdc.tosca.datatypes.model.NodeType;
import org.openecomp.sdc.tosca.datatypes.model.ServiceTemplate;
import org.openecomp.sdc.translator.services.heattotosca.globaltypes.GlobalTypesGenerator;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.Optional;
public class GlobalSubstitutionServiceTemplate extends ServiceTemplate {
private static final Logger logger = LoggerFactory.getLogger(ServiceTemplate.class);
public static final String GLOBAL_SUBSTITUTION_SERVICE_FILE_NAME =
"GlobalSubstitutionTypesServiceTemplate.yaml";
public static final String TEMPLATE_NAME_PROPERTY = "template_name";
public static final String DEFININTION_VERSION = "tosca_simple_yaml_1_0_0";
public static final String HEAT_INDEX = "openecomp_heat_index";
private static final Map<String, ServiceTemplate> globalServiceTemplates =
GlobalTypesGenerator.getGlobalTypesServiceTemplate();
public GlobalSubstitutionServiceTemplate() {
super();
init();
}
public void appendNodes(Map<String, NodeType> nodes) {
Optional<Map<String, NodeType>> nodeTypesToAdd =
removeExistingGlobalTypes(nodes);
nodeTypesToAdd.ifPresent(nodeTypes -> getNode_types().putAll(nodeTypes));
}
public void init() {
writeDefinitionSection();
writeMetadataSection();
writeImportsSection();
setNode_types(new HashMap<>());
}
private void writeImportsSection() {
List<Map<String, Import>> imports = new ArrayList<>();
Map<String, Import> stringImportMap = new HashMap<>();
imports.add(stringImportMap);
setImports(imports);
Import imprtObj = new Import();
imprtObj.setFile("openecomp-heat/_index.yml");
stringImportMap.put("openecomp_heat_index", imprtObj);
}
private void writeMetadataSection() {
Map<String, String> metadata = new HashMap<>();
metadata.put(TEMPLATE_NAME_PROPERTY, "GlobalSubstitutionTypes");
setMetadata(metadata);
}
private void writeDefinitionSection() {
setTosca_definitions_version(DEFININTION_VERSION);
}
private Optional<Map<String, NodeType>> removeExistingGlobalTypes(Map<String, NodeType> nodes){
Map<String, NodeType> nodeTypesToAdd = new HashMap<>();
ServiceTemplate serviceTemplate = globalServiceTemplates.get("openecomp/nodes.yml");
if(Objects.isNull(serviceTemplate) || MapUtils.isEmpty(serviceTemplate.getNode_types())){
return Optional.of(nodes);
}
Map<String, NodeType> globalNodeTypes = getAllGlobalNodeTypes();
for(Map.Entry<String, NodeType> nodeTypeEntry : nodes.entrySet()){
if(!globalNodeTypes.containsKey(nodeTypeEntry.getKey())){
Optional<NodeType> nodeType =
ToscaConverterUtil
.createObjectFromClass(nodeTypeEntry.getKey(), nodeTypeEntry.getValue(), NodeType.class);
nodeType
.ifPresent(nodeTypeValue -> nodeTypesToAdd.put(nodeTypeEntry.getKey(), nodeTypeValue));
}
}
return Optional.of(nodeTypesToAdd);
}
private Map<String, NodeType> getAllGlobalNodeTypes(){
Map<String, NodeType> globalNodeTypes = new HashMap<>();
for(Map.Entry<String, ServiceTemplate> serviceTemplateEntry : globalServiceTemplates.entrySet()){
if(isNodesServiceTemplate(serviceTemplateEntry.getKey())){
globalNodeTypes.putAll(serviceTemplateEntry.getValue().getNode_types());
}
}
return globalNodeTypes;
}
private boolean isNodesServiceTemplate(String filename) {
return filename.endsWith("nodes.yml") || filename.endsWith("nodes.yaml");
}
}
|
