2bc0de68880cf1d4bb31baa79f4943eddf8600f7
[sdc.git] / openecomp-be / lib / openecomp-sdc-translator-lib / openecomp-sdc-translator-core / src / main / java / org / openecomp / sdc / translator / services / heattotosca / impl / resourcetranslation / ResourceConnectionUsingCapabilityHelper.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * SDC
4  * ================================================================================
5  * Copyright (C) 2017 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
21 package org.openecomp.sdc.translator.services.heattotosca.impl.resourcetranslation;
22
23 import org.openecomp.sdc.heat.datatypes.manifest.FileData;
24 import org.openecomp.sdc.heat.datatypes.model.Resource;
25 import org.openecomp.sdc.tosca.datatypes.model.CapabilityDefinition;
26 import org.openecomp.sdc.tosca.datatypes.model.NodeTemplate;
27 import org.openecomp.sdc.tosca.datatypes.model.NodeType;
28 import org.openecomp.sdc.tosca.datatypes.model.RequirementAssignment;
29 import org.openecomp.sdc.tosca.datatypes.model.RequirementDefinition;
30 import org.openecomp.sdc.tosca.datatypes.model.ServiceTemplate;
31 import org.openecomp.sdc.tosca.services.DataModelUtil;
32 import org.openecomp.sdc.translator.datatypes.heattotosca.to.TranslateTo;
33 import org.openecomp.sdc.translator.datatypes.heattotosca.to.TranslatedHeatResource;
34 import org.openecomp.sdc.translator.services.heattotosca.ConsolidationDataUtil;
35 import org.openecomp.sdc.translator.services.heattotosca.HeatToToscaUtil;
36 import org.openecomp.sdc.translator.services.heattotosca.ResourceTranslationFactory;
37
38 import java.util.ArrayList;
39 import java.util.HashMap;
40 import java.util.List;
41 import java.util.Map;
42 import java.util.Optional;
43 import java.util.function.Predicate;
44
45 public abstract class ResourceConnectionUsingCapabilityHelper
46     extends BaseResourceConnection<CapabilityDefinition> {
47   public ResourceConnectionUsingCapabilityHelper(ResourceTranslationBase resourceTranslationBase,
48                                                  TranslateTo translateTo, FileData nestedFileData,
49                                                  NodeTemplate substitutionNodeTemplate,
50                                                  NodeType nodeType) {
51     super(resourceTranslationBase, translateTo, nestedFileData, substitutionNodeTemplate, nodeType);
52   }
53
54   abstract Map.Entry<String, RequirementDefinition> createRequirementDefinition(
55       String capabilityKey);
56
57   @Override
58   String getMappedNodeTranslatedResourceId(ServiceTemplate nestedServiceTemplate,
59                                            Map.Entry<String,
60                                                CapabilityDefinition> connectionPointEntry) {
61     List<String> substitutionMapping =
62         nestedServiceTemplate.getTopology_template().getSubstitution_mappings().getCapabilities()
63             .get(connectionPointEntry.getKey());
64     return substitutionMapping.get(0);
65   }
66
67   @Override
68   Map.Entry<String, CapabilityDefinition> getMappedConnectionPointEntry(
69       ServiceTemplate nestedServiceTemplate,
70       Map.Entry<String, CapabilityDefinition> connectionPointEntry) {
71     List<String> substitutionMapping =
72         nestedServiceTemplate.getTopology_template().getSubstitution_mappings().getCapabilities()
73             .get(connectionPointEntry.getKey());
74     String mappedNodeTranslatedId = substitutionMapping.get(0);
75     String mappedCapabilityId = substitutionMapping.get(1);
76     NodeTemplate mappedNodeTemplate =
77         nestedServiceTemplate.getTopology_template().getNode_templates()
78             .get(mappedNodeTranslatedId);
79     NodeType substituteNodeType =
80         translateTo.getContext().getGlobalSubstitutionServiceTemplate().getNode_types()
81             .get(mappedNodeTemplate.getType());
82     Optional<CapabilityDefinition> capabilityDefinition =
83         DataModelUtil.getCapabilityDefinition(substituteNodeType, mappedCapabilityId);
84     return new Map.Entry<String, CapabilityDefinition>() {
85       @Override
86       public String getKey() {
87         return mappedCapabilityId;
88       }
89
90       @Override
91       public CapabilityDefinition getValue() {
92         return capabilityDefinition.get();
93       }
94
95       @Override
96       public CapabilityDefinition setValue(CapabilityDefinition value) {
97         return null;
98       }
99     };
100   }
101
102   @Override
103   protected List<Map<String, CapabilityDefinition>> getAllConnectionPoints() {
104     List<Map<String, CapabilityDefinition>> exposedRequirementsList = new ArrayList<>();
105     List<Predicate<CapabilityDefinition>> predicates = getPredicatesListForConnectionPoints();
106     Map<String, CapabilityDefinition> capabilities = this.nodeType.getCapabilities();
107     if (capabilities == null) {
108       return exposedRequirementsList;
109     }
110     capabilities.entrySet()
111         .stream()
112         .filter(entry -> predicates
113             .stream()
114             .anyMatch(p -> p.test(entry.getValue())))
115         .forEach(entry -> {
116           Map<String, CapabilityDefinition> exposedRequirementsMap = new HashMap<>();
117           exposedRequirementsMap.put(entry.getKey(), entry.getValue());
118           exposedRequirementsList.add(exposedRequirementsMap);
119         });
120     return exposedRequirementsList;
121   }
122
123   void addRequirementToConnectResource(Map.Entry<String, CapabilityDefinition> connectionPointEntry,
124                                        List<String> supportedSourceNodeTypes, String paramName) {
125     Object paramValue = translateTo.getResource().getProperties().get(paramName);
126     if (paramValue == null) {
127       logger.warn("'" + paramName + "' property is not define in nested resource '"
128           + translateTo.getResourceId() + "' for the nested heat file, therefore, '"
129           + connectionPointEntry.getKey() + "' TOSCA capability will not be connected.");
130       return;
131     }
132
133     Map.Entry<String, RequirementDefinition> requirementDefinition =
134         createRequirementDefinition(connectionPointEntry.getKey());
135
136     Optional<String> sourceResourceId =
137         getConnectionResourceUsingGetResourceFunc(connectionPointEntry, paramName, paramValue,
138             supportedSourceNodeTypes);
139     if (sourceResourceId.isPresent()) {
140       Resource sourceResource = HeatToToscaUtil
141           .getResource(translateTo.getHeatOrchestrationTemplate(), sourceResourceId.get(),
142               translateTo.getHeatFileName());
143       Optional<String> translatedSourceNodeId =
144           ResourceTranslationFactory.getInstance(sourceResource)
145               .translateResource(translateTo.getHeatFileName(), translateTo.getServiceTemplate(),
146                   translateTo.getHeatOrchestrationTemplate(), sourceResource,
147                   sourceResourceId.get(), translateTo.getContext());
148       if (translatedSourceNodeId.isPresent()) {
149         NodeTemplate sourceNodeTemplate = DataModelUtil
150             .getNodeTemplate(translateTo.getServiceTemplate(), translatedSourceNodeId.get());
151         RequirementAssignment requirementAssignment = createRequirementAssignment(
152             requirementDefinition, translateTo.getTranslatedId(), sourceNodeTemplate);
153         ConsolidationDataUtil.updateNodesConnectedData(translateTo, translateTo.getResourceId(),
154             translateTo.getResource(), sourceResource, translatedSourceNodeId.get(),
155             requirementDefinition.getKey(), requirementAssignment);
156       } else {
157         logger.warn(
158             "'" + sourceResource.getType() + "' connection to '" + connectionPointEntry.getKey()
159                 + "' capability of type '" + connectionPointEntry.getValue().getType()
160                 + "' is not supported/invalid, therefore this connection will be ignored in the "
161                 + "TOSCA translation");
162       }
163     } else {
164       Optional<TranslatedHeatResource> sharedSourceTranslatedHeatResource =
165           getConnectionTranslatedHeatResourceUsingGetParamFunc(connectionPointEntry, paramName,
166               supportedSourceNodeTypes);
167       if (sharedSourceTranslatedHeatResource.isPresent()) {
168         NodeTemplate sharedSourceNodeTemplate = DataModelUtil
169             .getNodeTemplate(translateTo.getServiceTemplate(),
170                 sharedSourceTranslatedHeatResource.get().getTranslatedId());
171         RequirementAssignment requirementAssignment = createRequirementAssignment(
172             requirementDefinition, translateTo.getTranslatedId(), sharedSourceNodeTemplate);
173
174         ConsolidationDataUtil.updateNodesConnectedData(translateTo, translateTo.getResourceId(),
175             translateTo.getResource(), sharedSourceTranslatedHeatResource.get().getHeatResource(),
176             sharedSourceTranslatedHeatResource.get().getTranslatedId(),
177             requirementDefinition.getKey(),
178             requirementAssignment);
179       }
180     }
181   }
182
183 }