9007e209985e0662b7a04abefa9f7ef863163c28
[sdc.git] / catalog-model / src / main / java / org / openecomp / sdc / be / model / jsonjanusgraph / operations / SubstitutionFilterOperation.java
1 /*
2  * ============LICENSE_START=======================================================
3  *  Copyright (C) 2020 Nordix Foundation
4  *  ================================================================================
5  *  Licensed under the Apache License, Version 2.0 (the "License");
6  *  you may not use this file except in compliance with the License.
7  *  You may obtain a copy of the License at
8  *
9  *        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.
15  *
16  *  SPDX-License-Identifier: Apache-2.0
17  *  ============LICENSE_END=========================================================
18  */
19
20 package org.openecomp.sdc.be.model.jsonjanusgraph.operations;
21
22 import com.google.common.collect.ImmutableList;
23 import fj.data.Either;
24 import java.util.List;
25 import org.openecomp.sdc.be.dao.janusgraph.JanusGraphOperationStatus;
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.datatypes.elements.ListDataDefinition;
31 import org.openecomp.sdc.be.datatypes.elements.RequirementSubstitutionFilterPropertyDataDefinition;
32 import org.openecomp.sdc.be.datatypes.elements.SubstitutionFilterDataDefinition;
33 import org.openecomp.sdc.be.datatypes.enums.JsonPresentationFields;
34 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
35 import org.openecomp.sdc.be.model.operations.impl.DaoStatusConverter;
36 import org.openecomp.sdc.common.jsongraph.util.CommonUtility;
37 import org.openecomp.sdc.common.log.enums.EcompErrorSeverity;
38 import org.openecomp.sdc.common.log.enums.EcompLoggerErrorCode;
39 import org.openecomp.sdc.common.log.wrappers.Logger;
40
41 @org.springframework.stereotype.Component("substitution-filter-operations")
42 public class SubstitutionFilterOperation extends BaseOperation {
43
44     private static final Logger LOGGER = Logger.getLogger(SubstitutionFilterOperation.class);
45
46     public Either<SubstitutionFilterDataDefinition, StorageOperationStatus> createSubstitutionFilter(
47         final String componentId, final String componentInstanceId) {
48
49         return addOrUpdateSubstitutionFilter(false, componentId, componentInstanceId,
50             new SubstitutionFilterDataDefinition());
51     }
52
53     public Either<SubstitutionFilterDataDefinition, StorageOperationStatus> deleteConstraint(
54         final String serviceId, final String componentInstanceId,
55         final SubstitutionFilterDataDefinition substitutionFilterDataDefinition, final int propertyIndex) {
56
57         final ListDataDefinition<RequirementSubstitutionFilterPropertyDataDefinition> properties =
58             substitutionFilterDataDefinition.getProperties();
59         properties.getListToscaDataDefinition().remove(propertyIndex);
60         substitutionFilterDataDefinition.setProperties(properties);
61         return addOrUpdateSubstitutionFilter(true, serviceId, componentInstanceId, substitutionFilterDataDefinition);
62     }
63
64     public Either<SubstitutionFilterDataDefinition, StorageOperationStatus> addNewProperty(
65         final String componentId, final String componentInstanceId,
66         final SubstitutionFilterDataDefinition substitutionFilterDataDefinition,
67         final RequirementSubstitutionFilterPropertyDataDefinition requirementSubstitutionFilterPropertyDataDefinition) {
68
69         ListDataDefinition<RequirementSubstitutionFilterPropertyDataDefinition> properties =
70             substitutionFilterDataDefinition.getProperties();
71         if (properties == null) {
72             properties = new ListDataDefinition<>();
73             substitutionFilterDataDefinition.setProperties(properties);
74         }
75         properties.getListToscaDataDefinition().add(requirementSubstitutionFilterPropertyDataDefinition);
76         substitutionFilterDataDefinition.setProperties(properties);
77         return addOrUpdateSubstitutionFilter(true, componentId, componentInstanceId, substitutionFilterDataDefinition);
78     }
79
80     public Either<SubstitutionFilterDataDefinition, StorageOperationStatus> updateSubstitutionFilter(
81         final String serviceId, final String componentInstanceId,
82         final SubstitutionFilterDataDefinition substitutionFilterDataDefinition,
83         final List<RequirementSubstitutionFilterPropertyDataDefinition> requirementSubstitutionFilterPropertyDataDefinitions) {
84
85         final ListDataDefinition<RequirementSubstitutionFilterPropertyDataDefinition> properties =
86             substitutionFilterDataDefinition.getProperties();
87         properties.getListToscaDataDefinition().clear();
88         properties.getListToscaDataDefinition().addAll(requirementSubstitutionFilterPropertyDataDefinitions);
89         substitutionFilterDataDefinition.setProperties(properties);
90         return addOrUpdateSubstitutionFilter(true, serviceId, componentInstanceId,
91             substitutionFilterDataDefinition);
92     }
93
94     private Either<SubstitutionFilterDataDefinition, StorageOperationStatus> addOrUpdateSubstitutionFilter(
95         final boolean isUpdateAction, final String componentId, final String componentInstanceId,
96         final SubstitutionFilterDataDefinition substitutionFilterDataDefinition) {
97
98         StorageOperationStatus statusRes;
99         Either<GraphVertex, JanusGraphOperationStatus> getToscaElementRes;
100
101         getToscaElementRes = janusGraphDao.getVertexById(componentId, JsonParseFlagEnum.NoParse);
102         if (getToscaElementRes.isRight()) {
103             final JanusGraphOperationStatus status = getToscaElementRes.right().value();
104             CommonUtility.addRecordToLog(LOGGER, CommonUtility.LogLevelEnum.DEBUG,
105                 "Failed to get tosca element {} upon adding the properties. Status is {}. ", componentId, status);
106             statusRes = DaoStatusConverter.convertJanusGraphStatusToStorageStatus(status);
107             return Either.right(statusRes);
108         }
109         final GraphVertex serviceVertex = getToscaElementRes.left().value();
110         substitutionFilterDataDefinition.setID(componentInstanceId);
111         statusRes = performUpdateToscaAction(isUpdateAction, serviceVertex,
112             ImmutableList.of(substitutionFilterDataDefinition));
113         if (!statusRes.equals(StorageOperationStatus.OK)) {
114             janusGraphDao.rollback();
115             LOGGER.error(EcompErrorSeverity.ERROR, EcompLoggerErrorCode.BUSINESS_PROCESS_ERROR,
116                 " Failed to perform tosca update for substitution filter in service {} , component instance {}. status is {}",
117                 componentId, componentInstanceId, statusRes);
118             return Either.right(statusRes);
119         }
120         janusGraphDao.commit();
121         return Either.left(substitutionFilterDataDefinition);
122     }
123
124     private StorageOperationStatus performUpdateToscaAction(final boolean isUpdate,
125                                                             final GraphVertex graphVertex,
126                                                             final List<SubstitutionFilterDataDefinition> toscaDataList) {
127         if (isUpdate) {
128             return updateToscaDataOfToscaElement(graphVertex, EdgeLabelEnum.SUBSTITUTION_FILTER_TEMPLATE,
129                 VertexTypeEnum.SUBSTITUTION_FILTER_TEMPLATE, toscaDataList, JsonPresentationFields.UNIQUE_ID);
130         } else {
131             return addToscaDataToToscaElement(graphVertex, EdgeLabelEnum.SUBSTITUTION_FILTER_TEMPLATE,
132                 VertexTypeEnum.SUBSTITUTION_FILTER_TEMPLATE, toscaDataList, JsonPresentationFields.UNIQUE_ID);
133         }
134     }
135
136 }
137
138
139