2 * Copyright © 2016-2018 European Support Limited
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
8 * http://www.apache.org/licenses/LICENSE-2.0
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.
17 package org.openecomp.sdc.translator.services.heattotosca;
19 import java.util.ArrayList;
20 import java.util.Collection;
21 import java.util.Collections;
22 import java.util.HashMap;
23 import java.util.List;
25 import java.util.Objects;
26 import java.util.Optional;
28 import org.apache.commons.collections4.MapUtils;
29 import org.onap.sdc.tosca.datatypes.model.NodeTemplate;
30 import org.onap.sdc.tosca.datatypes.model.ServiceTemplate;
31 //import org.openecomp.core.model.types.ServiceTemplate;
32 import org.openecomp.sdc.tosca.services.DataModelUtil;
33 import org.openecomp.sdc.translator.datatypes.heattotosca.TranslationContext;
34 import org.openecomp.sdc.translator.datatypes.heattotosca.unifiedmodel.composition.UnifiedCompositionData;
35 import org.openecomp.sdc.translator.datatypes.heattotosca.unifiedmodel.consolidation.ComputeTemplateConsolidationData;
36 import org.openecomp.sdc.translator.datatypes.heattotosca.unifiedmodel.consolidation.PortTemplateConsolidationData;
37 import org.openecomp.sdc.translator.datatypes.heattotosca.unifiedmodel.consolidation.SubInterfaceTemplateConsolidationData;
39 import com.google.common.collect.ArrayListMultimap;
40 import com.google.common.collect.ListMultimap;
43 * Utility class for consolidation data collection helper methods.
45 public class UnifiedCompositionUtil {
47 private UnifiedCompositionUtil() {
48 //Hiding the implicit public constructor
52 * Gets all ports per port type, which are connected to the computes from the input
53 * computeTemplateConsolidationDataCollection.
55 * @param computeTemplateConsolidationDataCollection collection of compute template
57 * @return set of port ids, per port type
59 static Map<String, List<String>> collectAllPortsFromEachTypesFromComputes(
60 Collection<ComputeTemplateConsolidationData> computeTemplateConsolidationDataCollection) {
61 Map<String, List<String>> portTypeToIds = new HashMap<>();
63 for (ComputeTemplateConsolidationData compute : computeTemplateConsolidationDataCollection) {
64 Map<String, List<String>> ports = compute.getPorts();
65 if (!MapUtils.isEmpty(ports)) {
66 addPortsToMap(portTypeToIds, ports);
73 static ListMultimap<String, SubInterfaceTemplateConsolidationData>
74 collectAllSubInterfacesOfEachTypesFromPorts(Collection<PortTemplateConsolidationData>
75 portTemplateConsolidationDataCollection) {
76 ListMultimap<String, SubInterfaceTemplateConsolidationData> subInterfaceDataByType = ArrayListMultimap.create();
77 for (PortTemplateConsolidationData port : portTemplateConsolidationDataCollection) {
78 port.copyMappedInto(subInterfaceDataByType);
80 return subInterfaceDataByType;
83 private static void addPortsToMap(Map<String, List<String>> portTypeToIds,
84 Map<String, List<String>> ports) {
85 for (Map.Entry<String, List<String>> portTypeToIdEntry : ports.entrySet()) {
86 portTypeToIds.putIfAbsent(portTypeToIdEntry.getKey(), new ArrayList<>());
87 portTypeToIds.get(portTypeToIdEntry.getKey()).addAll(portTypeToIdEntry.getValue());
91 static String getComputeTypeSuffix(ServiceTemplate serviceTemplate,
92 String computeNodeTemplateId) {
93 NodeTemplate computeNodeTemplate = DataModelUtil.getNodeTemplate(serviceTemplate, computeNodeTemplateId);
94 if (Objects.nonNull(computeNodeTemplate)) {
95 return getComputeTypeSuffix(computeNodeTemplate.getType());
100 public static String getNewComputeNodeTemplateId(ServiceTemplate serviceTemplate, String computeNodeTemplateId) {
101 return getComputeTypeSuffix(serviceTemplate, computeNodeTemplateId);
104 static String getComputeTypeSuffix(String computeType) {
105 return DataModelUtil.getNamespaceSuffix(computeType);
108 public static ComputeTemplateConsolidationData getConnectedComputeConsolidationData(
109 List<UnifiedCompositionData> unifiedCompositionDataList,
110 String portNodeTemplateId) {
111 for (UnifiedCompositionData unifiedCompositionData : unifiedCompositionDataList) {
112 if (Objects.isNull(unifiedCompositionData.getComputeTemplateConsolidationData().getPorts())) {
115 Collection<List<String>> portsCollection =
116 unifiedCompositionData.getComputeTemplateConsolidationData().getPorts().values();
117 for (List<String> portIdList : portsCollection) {
118 if (portIdList.contains(portNodeTemplateId)) {
119 return unifiedCompositionData.getComputeTemplateConsolidationData();
126 //The ID should be <vm_type>_<port_type> or <vm_type>_<portNodeTemplateId>
127 public static String getNewPortNodeTemplateId(
128 String portNodeTemplateId,
129 String connectedComputeNodeType,
130 ComputeTemplateConsolidationData computeTemplateConsolidationData) {
132 StringBuilder newPortNodeTemplateId = new StringBuilder();
133 String portType = ConsolidationDataUtil.getPortType(portNodeTemplateId);
134 newPortNodeTemplateId.append(DataModelUtil.getNamespaceSuffix(connectedComputeNodeType));
135 if (computeTemplateConsolidationData.getPorts().get(portType).size() > 1) {
137 newPortNodeTemplateId.append("_").append(portNodeTemplateId);
140 newPortNodeTemplateId.append("_").append(portType);
142 return newPortNodeTemplateId.toString();
145 public static String getNewSubInterfaceNodeTemplateId(ServiceTemplate serviceTemplate,
146 String connectedComputeNodeType,
147 ComputeTemplateConsolidationData computeTemplateConsolidationData,
148 SubInterfaceTemplateConsolidationData
149 subInterfaceTemplateConsolidationData,
150 TranslationContext context) {
151 //The ID should be <vm_type>_<portType/NodetemplateId>_<subInterface_type>
152 // or <vm_type>_<portType/NodetemplateId>_<subInterfaceNodeTemplateId>
153 StringBuilder newSubInterfaceNodeTemplateId = new StringBuilder();
154 newSubInterfaceNodeTemplateId.append(getNewPortNodeTemplateId(subInterfaceTemplateConsolidationData
155 .getParentPortNodeTemplateId(), connectedComputeNodeType, computeTemplateConsolidationData));
156 Optional<PortTemplateConsolidationData> portTemplateConsolidationData =
157 subInterfaceTemplateConsolidationData.getParentPortTemplateConsolidationData(serviceTemplate, context);
158 NodeTemplate subInterfaceNodeTemplate =
159 DataModelUtil.getNodeTemplate(serviceTemplate, subInterfaceTemplateConsolidationData.getNodeTemplateId());
160 if (portTemplateConsolidationData.isPresent()) {
161 String subInterfaceSuffix = getSubInterfaceSuffix(portTemplateConsolidationData.get(),
162 subInterfaceNodeTemplate, subInterfaceTemplateConsolidationData);
163 newSubInterfaceNodeTemplateId.append("_").append(subInterfaceSuffix);
164 return newSubInterfaceNodeTemplateId.toString();
166 return subInterfaceTemplateConsolidationData.getNodeTemplateId();
169 static String getSubInterfaceTypeSuffix(String nodeType) {
170 return DataModelUtil.getNamespaceSuffix(nodeType);
173 public static List<SubInterfaceTemplateConsolidationData> getSubInterfaceTemplateConsolidationDataList(
174 UnifiedCompositionData unifiedCompositionData) {
175 return unifiedCompositionData.getSubInterfaceTemplateConsolidationDataList() == null ? Collections.emptyList() :
176 unifiedCompositionData.getSubInterfaceTemplateConsolidationDataList();
179 private static String getSubInterfaceSuffix(PortTemplateConsolidationData portTemplateConsolidationData,
180 NodeTemplate subInterfaceNodeTemplate,
181 SubInterfaceTemplateConsolidationData
182 subInterfaceTemplateConsolidationData) {
183 if (portTemplateConsolidationData.isSubInterfaceNodeTemplateIdParameter(subInterfaceNodeTemplate.getType())) {
184 //If there are more than one subinterfaces with same type use node template id
185 return subInterfaceTemplateConsolidationData.getNodeTemplateId();
187 //Add sub interface type since we have only one subinterface per type
188 return getSubInterfaceTypeSuffix(subInterfaceNodeTemplate.getType());
191 public static Map<String, List<String>> collectAllPortsOfEachTypeFromComputes(
192 List<UnifiedCompositionData> unifiedCompositionDataList) {
193 Map<String, List<String>> portIdsPerPortType = new HashMap<>();
194 unifiedCompositionDataList
195 .forEach(unifiedCompositionData ->
196 unifiedCompositionData.getComputeTemplateConsolidationData()
197 .collectAllPortsOfEachTypeFromCompute(portIdsPerPortType));
199 return portIdsPerPortType;