Fix test cases failing incorrectly
[sdc.git] / catalog-be / src / main / java / org / openecomp / sdc / be / tosca / utils / ForwardingPathToscaUtil.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.be.tosca.utils;
21
22 import fj.data.Either;
23 import java.util.ArrayList;
24 import java.util.Collection;
25 import java.util.Collections;
26 import java.util.HashMap;
27 import java.util.List;
28 import java.util.Map;
29 import java.util.Objects;
30 import java.util.Optional;
31 import lombok.AccessLevel;
32 import lombok.NoArgsConstructor;
33 import org.apache.commons.collections.MapUtils;
34 import org.openecomp.sdc.be.datatypes.elements.ForwardingPathDataDefinition;
35 import org.openecomp.sdc.be.datatypes.elements.ForwardingPathElementDataDefinition;
36 import org.openecomp.sdc.be.model.CapabilityDefinition;
37 import org.openecomp.sdc.be.model.Component;
38 import org.openecomp.sdc.be.model.ComponentInstance;
39 import org.openecomp.sdc.be.model.Service;
40 import org.openecomp.sdc.be.model.jsonjanusgraph.operations.ToscaOperationFacade;
41 import org.openecomp.sdc.be.tosca.CapabilityRequirementConverter;
42 import org.openecomp.sdc.be.tosca.model.ToscaNodeTemplate;
43 import org.openecomp.sdc.be.tosca.model.ToscaTemplateRequirement;
44
45 @NoArgsConstructor(access = AccessLevel.PRIVATE)
46 public class ForwardingPathToscaUtil {
47
48     public static final String FORWARDS_TO_TOSCA_NAME = "org.openecomp.relationships.ForwardsTo";
49     public static final String PROTOCOL = "protocol";
50     public static final String PORTS_RANGE = "target_range";
51     public static final String FORWARDER = "forwarder";
52
53     public static void addForwardingPaths(Service service, Map<String, ToscaNodeTemplate> nodeTemplates,
54                                           CapabilityRequirementConverter capabiltyRequirementConvertor, Map<String, Component> originComponents,
55                                           ToscaOperationFacade toscaOperationFacade) {
56         for (String forwardingPathName : service.getForwardingPaths().keySet()) {
57             ToscaNodeTemplate forwardingPathNodeTemplate = new ToscaNodeTemplate();
58             final ForwardingPathDataDefinition path = service.getForwardingPaths().get(forwardingPathName);
59             forwardingPathNodeTemplate.setType(path.getToscaResourceName());
60             if (Objects.nonNull(path.getDescription())) {
61                 forwardingPathNodeTemplate.setDescription(path.getDescription());
62             }
63             Map<String, Object> props = new HashMap<>();
64             if (Objects.nonNull(path.getDestinationPortNumber())) {
65                 props.put(PORTS_RANGE, Collections.singletonList(path.getDestinationPortNumber()));
66             }
67             if (Objects.nonNull(path.getProtocol())) {
68                 props.put(PROTOCOL, path.getProtocol());
69             }
70             if (MapUtils.isNotEmpty(props)) {
71                 forwardingPathNodeTemplate.setProperties(props);
72             }
73             final List<ForwardingPathElementDataDefinition> pathElements = path.getPathElements().getListToscaDataDefinition();
74             forwardingPathNodeTemplate.setRequirements(
75                 convertPathElementsToRequirements(pathElements, service, capabiltyRequirementConvertor, originComponents, toscaOperationFacade));
76             nodeTemplates.put(path.getName(), forwardingPathNodeTemplate);
77         }
78     }
79
80     private static List<Map<String, ToscaTemplateRequirement>> convertPathElementsToRequirements(
81         List<ForwardingPathElementDataDefinition> pathElements, Service service, CapabilityRequirementConverter capabiltyRequirementConvertor,
82         Map<String, Component> originComponents, ToscaOperationFacade toscaOperationFacade) {
83         List<Map<String, ToscaTemplateRequirement>> toscaRequirements = new ArrayList<>();
84         for (int i = 0; i <= pathElements.size() - 1; i++) {
85             final ForwardingPathElementDataDefinition element = pathElements.get(i);
86             toscaRequirements.add(handleSingleReq(
87                 fetchCPName(service, element.getFromNode(), element.getFromCP(), capabiltyRequirementConvertor, originComponents,
88                     toscaOperationFacade), fetchNodeName(service, element.getFromNode())));
89             if (i == pathElements.size() - 1) {
90                 toscaRequirements.add(handleSingleReq(
91                     fetchCPName(service, element.getToNode(), element.getToCP(), capabiltyRequirementConvertor, originComponents,
92                         toscaOperationFacade), fetchNodeName(service, element.getToNode())));
93             }
94         }
95         return toscaRequirements;
96     }
97
98     private static String fetchNodeName(Service service, String nodeId) {
99         final Optional<ComponentInstance> componentInstanceByName = service.getComponentInstanceByName(nodeId);
100         if (componentInstanceByName.isPresent()) {
101             return componentInstanceByName.get().getName();
102         } else {
103             return "";
104         }
105     }
106
107     private static Map<String, ToscaTemplateRequirement> handleSingleReq(String fromCP, String fromNode) {
108         Map<String, ToscaTemplateRequirement> toscaReqMap = new HashMap<>();
109         ToscaTemplateRequirement firstReq = new ToscaTemplateRequirement();
110         firstReq.setRelationship(FORWARDS_TO_TOSCA_NAME); //todo
111         firstReq.setCapability(fromCP);
112         firstReq.setNode(fromNode);
113         toscaReqMap.put(FORWARDER, firstReq);
114         return toscaReqMap;
115     }
116
117     /**
118      * @todo handle errors.
119      */
120     private static String fetchCPName(Service service, String nodeID, String cpName, CapabilityRequirementConverter capabiltyRequirementConvertor,
121                                       Map<String, Component> originComponents, ToscaOperationFacade toscaOperationFacade) {
122         Optional<ComponentInstance> componentInstance = service.getComponentInstanceByName(nodeID);
123         ComponentInstance componentInstanceVal = componentInstance.get();
124         String name = componentInstanceVal.getNormalizedName();
125         Component component = originComponents.get(componentInstanceVal.getComponentUid());
126         if (componentInstanceVal.getIsProxy()) {
127             component = originComponents.get(componentInstanceVal.getSourceModelUid());
128             if (component == null) {
129                 component = toscaOperationFacade.getToscaFullElement(componentInstanceVal.getSourceModelUid()).left().value();
130             }
131         }
132         CapabilityDefinition capability = componentInstanceVal.getCapabilities().values().stream().flatMap(Collection::stream)
133             .filter(capabilityDefinition -> capabilityDefinition.getName().equals(cpName)).findAny().get();
134         List<String> path = capability.getPath();
135         List<String> reducedPath = new ArrayList<>(path);
136         reducedPath.remove(reducedPath.size() - 1);
137         Either<String, Boolean> stringBooleanEither = capabiltyRequirementConvertor
138             .buildSubstitutedName(originComponents, component, reducedPath, capability.getName(), null);
139         return name + "." + stringBooleanEither.left().value();
140     }
141 }