2b720af75199459584fc0b238fa2327d0149182a
[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.ListDataDefinition;
32 import org.openecomp.sdc.be.datatypes.elements.ListRequirementDataDefinition;
33 import org.openecomp.sdc.be.datatypes.elements.MapDataDefinition;
34 import org.openecomp.sdc.be.datatypes.elements.MapListRequirementDataDefinition;
35 import org.openecomp.sdc.be.datatypes.elements.RequirementDataDefinition;
36 import org.openecomp.sdc.be.datatypes.enums.NodeTypeEnum;
37 import org.openecomp.sdc.be.model.ComponentInstance;
38 import org.openecomp.sdc.be.model.RequirementDefinition;
39 import org.openecomp.sdc.be.model.operations.api.IComponentInstanceOperation;
40 import org.openecomp.sdc.be.model.operations.api.IRequirementOperation;
41 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
42 import org.openecomp.sdc.be.resources.data.RequirementData;
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 FulfilledRequirementsMigrationService extends FulfilledCapabilityRequirementMigrationService<RequirementDefinition, RequirementData> {
51
52     @Resource(name = "requirement-operation")
53     IRequirementOperation requirementOperation;
54
55     @Resource(name = "component-instance-operation")
56     IComponentInstanceOperation componentInstanceOperation;
57
58     @Override
59     Either<RequirementDefinition, TitanOperationStatus> getToscaDefinition(RequirementData data) {
60         return requirementOperation.getRequirement(data.getUniqueId());
61     }
62
63     @Override
64     void setPath(RequirementDefinition def, List<String> path) {
65         def.setPath(path);
66     }
67
68     @Override
69     String getType(RequirementDefinition def) {
70         return def.getCapability();
71     }
72
73     @Override
74     Either<List<ImmutablePair<RequirementData, GraphEdge>>, TitanOperationStatus> getFulfilledCapReqs(ComponentInstance instance, NodeTypeEnum nodeTypeEnum) {
75         return componentInstanceOperation.getFulfilledRequirements(instance, nodeTypeEnum);
76     }
77
78     @Override
79     ListDataDefinition convertToDefinitionListObject(List<RequirementDefinition> capReqDefList) {
80         List<RequirementDataDefinition> requirementDataDefinitions = new ArrayList<>();
81         requirementDataDefinitions.addAll(capReqDefList);
82         return new ListRequirementDataDefinition(requirementDataDefinitions);
83     }
84
85     @Override
86     MapDataDefinition convertToDefinitionMapObject(Map<String, ListDataDefinition> reqCapForInstance) {
87         Map<String, ListRequirementDataDefinition> reqDefList = castDefinitionListToRequirementList(reqCapForInstance);
88         return new MapListRequirementDataDefinition(reqDefList);
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_REQUIREMENTS, EdgeLabelEnum.FULLFILLED_REQUIREMENTS, defsByInstance);
99     }
100
101     private Map<String, ListRequirementDataDefinition> castDefinitionListToRequirementList(Map<String, ListDataDefinition> reqCapForInstance) {
102         return reqCapForInstance.entrySet().stream().collect(Collectors.toMap(Map.Entry::getKey, entry -> (ListRequirementDataDefinition) entry.getValue()));
103     }
104 }