23b1afec93d68a65183148a437ce626e59103af0
[sdc.git] /
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.asdctool.impl.migration.v1707.jsonmodel.relations;
22
23 import fj.data.Either;
24 import org.apache.commons.lang3.tuple.ImmutablePair;
25 import org.openecomp.sdc.be.dao.graph.datatype.GraphEdge;
26 import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
27 import org.openecomp.sdc.be.dao.jsongraph.types.EdgeLabelEnum;
28 import org.openecomp.sdc.be.dao.jsongraph.types.JsonParseFlagEnum;
29 import org.openecomp.sdc.be.dao.jsongraph.types.VertexTypeEnum;
30 import org.openecomp.sdc.be.dao.titan.TitanOperationStatus;
31 import org.openecomp.sdc.be.datatypes.elements.CapabilityDataDefinition;
32 import org.openecomp.sdc.be.datatypes.elements.ListCapabilityDataDefinition;
33 import org.openecomp.sdc.be.datatypes.elements.ListDataDefinition;
34 import org.openecomp.sdc.be.datatypes.elements.MapDataDefinition;
35 import org.openecomp.sdc.be.datatypes.elements.MapListCapabiltyDataDefinition;
36 import org.openecomp.sdc.be.datatypes.enums.NodeTypeEnum;
37 import org.openecomp.sdc.be.model.CapabilityDefinition;
38 import org.openecomp.sdc.be.model.ComponentInstance;
39 import org.openecomp.sdc.be.model.operations.api.ICapabilityOperation;
40 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
41 import org.openecomp.sdc.be.model.operations.impl.ComponentInstanceOperation;
42 import org.openecomp.sdc.be.resources.data.CapabilityData;
43
44 import javax.annotation.Resource;
45 import java.util.ArrayList;
46 import java.util.List;
47 import java.util.Map;
48 import java.util.stream.Collectors;
49
50 public class FulfilledCapabilitiesMigrationService extends FulfilledCapabilityRequirementMigrationService<CapabilityDefinition, CapabilityData> {
51
52     @Resource(name = "capability-operation")
53     private ICapabilityOperation capabilityOperation;
54
55     @Resource(name = "component-instance-operation")
56     private ComponentInstanceOperation componentInstanceOperation;
57
58     @Override
59     Either<CapabilityDefinition, StorageOperationStatus> getToscaDefinition(CapabilityData data) {
60         return capabilityOperation.getCapability(data.getUniqueId());
61     }
62
63     @Override
64     void setPath(CapabilityDefinition def, List<String> path) {
65         def.setPath(path);
66     }
67
68     @Override
69     String getType(CapabilityDefinition def) {
70         return def.getType();
71     }
72
73     @Override
74     Either<List<ImmutablePair<CapabilityData, GraphEdge>>, TitanOperationStatus> getFulfilledCapReqs(ComponentInstance instance, NodeTypeEnum nodeTypeEnum) {
75         return componentInstanceOperation.getFulfilledCapabilities(instance, nodeTypeEnum);
76     }
77
78     @Override
79     ListDataDefinition convertToDefinitionListObject(List<CapabilityDefinition> capReqDefList) {
80         List<CapabilityDataDefinition> capabilityDataDefinitions = new ArrayList<>();
81         capabilityDataDefinitions.addAll(capReqDefList);
82         return new ListCapabilityDataDefinition(capabilityDataDefinitions);
83     }
84
85     @Override
86     MapDataDefinition convertToDefinitionMapObject(Map<String, ListDataDefinition> reqCapForInstance) {
87         Map<String, ListCapabilityDataDefinition> capabilitiesList = castDataDefinitionListToCapabilityList(reqCapForInstance);
88         return new MapListCapabiltyDataDefinition(capabilitiesList);
89     }
90
91     @Override
92     Either<GraphVertex, TitanOperationStatus> getAssociatedDefinitions(GraphVertex component) {
93         return titanDao.getChildVertex(component,  EdgeLabelEnum.FULLFILLED_REQUIREMENTS, JsonParseFlagEnum.NoParse);
94     }
95
96     @Override
97     Either<GraphVertex, StorageOperationStatus> associateToGraph(GraphVertex graphVertex, Map<String, MapDataDefinition> defsByInstance) {
98         return topologyTemplateOperation.assosiateElementToData(graphVertex, VertexTypeEnum.FULLFILLED_CAPABILITIES, EdgeLabelEnum.FULLFILLED_CAPABILITIES, defsByInstance);
99     }
100
101     private Map<String, ListCapabilityDataDefinition> castDataDefinitionListToCapabilityList(Map<String, ListDataDefinition> reqCapForInstance) {
102         return reqCapForInstance.entrySet().stream().collect(Collectors.toMap(Map.Entry::getKey, entry -> (ListCapabilityDataDefinition) entry.getValue()));
103     }
104 }