Reformat catalog-model
[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 package org.openecomp.sdc.be.model.jsonjanusgraph.operations;
20
21 import com.google.common.collect.ImmutableList;
22 import fj.data.Either;
23 import java.util.List;
24 import java.util.Objects;
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(final String componentId) {
47         return addOrUpdateSubstitutionFilter(false, componentId, new SubstitutionFilterDataDefinition());
48     }
49
50     public Either<SubstitutionFilterDataDefinition, StorageOperationStatus> deleteConstraint(final String serviceId,
51                                                                                              final SubstitutionFilterDataDefinition substitutionFilterDataDefinition,
52                                                                                              final int propertyIndex) {
53         final ListDataDefinition<RequirementSubstitutionFilterPropertyDataDefinition> properties = substitutionFilterDataDefinition.getProperties();
54         properties.getListToscaDataDefinition().remove(propertyIndex);
55         substitutionFilterDataDefinition.setProperties(properties);
56         return addOrUpdateSubstitutionFilter(true, serviceId, substitutionFilterDataDefinition);
57     }
58
59     public Either<SubstitutionFilterDataDefinition, StorageOperationStatus> addPropertyFilter(final String componentId,
60                                                                                               final SubstitutionFilterDataDefinition substitutionFilterDataDefinition,
61                                                                                               final RequirementSubstitutionFilterPropertyDataDefinition substitutionFilterPropertyDataDefinition) {
62         final SubstitutionFilterDataDefinition substitutionFilterDataDefinition1 = Objects
63             .requireNonNullElseGet(substitutionFilterDataDefinition, SubstitutionFilterDataDefinition::new);
64         final ListDataDefinition<RequirementSubstitutionFilterPropertyDataDefinition> properties = Objects
65             .requireNonNullElseGet(substitutionFilterDataDefinition1.getProperties(), ListDataDefinition::new);
66         properties.getListToscaDataDefinition().add(substitutionFilterPropertyDataDefinition);
67         substitutionFilterDataDefinition1.setProperties(properties);
68         return addOrUpdateSubstitutionFilter(true, componentId, substitutionFilterDataDefinition1);
69     }
70
71     public Either<SubstitutionFilterDataDefinition, StorageOperationStatus> updateProperties(final String componentId,
72                                                                                              final SubstitutionFilterDataDefinition substitutionFilterDataDefinition,
73                                                                                              final List<RequirementSubstitutionFilterPropertyDataDefinition> requirementSubstitutionFilterPropertyDataDefinition) {
74         final ListDataDefinition<RequirementSubstitutionFilterPropertyDataDefinition> properties = substitutionFilterDataDefinition.getProperties();
75         properties.getListToscaDataDefinition().clear();
76         properties.getListToscaDataDefinition().addAll(requirementSubstitutionFilterPropertyDataDefinition);
77         substitutionFilterDataDefinition.setProperties(properties);
78         return addOrUpdateSubstitutionFilter(true, componentId, substitutionFilterDataDefinition);
79     }
80
81     private Either<SubstitutionFilterDataDefinition, StorageOperationStatus> addOrUpdateSubstitutionFilter(final boolean isUpdateAction,
82                                                                                                            final String componentId,
83                                                                                                            final SubstitutionFilterDataDefinition substitutionFilterDataDefinition) {
84         final Either<GraphVertex, JanusGraphOperationStatus> toscaElementEither = janusGraphDao.getVertexById(componentId, JsonParseFlagEnum.NoParse);
85         if (toscaElementEither.isRight()) {
86             final JanusGraphOperationStatus status = toscaElementEither.right().value();
87             CommonUtility
88                 .addRecordToLog(LOGGER, CommonUtility.LogLevelEnum.DEBUG, "Failed to get tosca element {} upon adding the properties. Status is {}. ",
89                     componentId, status);
90             return Either.right(DaoStatusConverter.convertJanusGraphStatusToStorageStatus(status));
91         }
92         final GraphVertex serviceVertex = toscaElementEither.left().value();
93         substitutionFilterDataDefinition.setID(componentId);
94         final StorageOperationStatus operationStatus = performUpdateToscaAction(isUpdateAction, serviceVertex,
95             ImmutableList.of(substitutionFilterDataDefinition));
96         if (!StorageOperationStatus.OK.equals(operationStatus)) {
97             janusGraphDao.rollback();
98             LOGGER.error(EcompErrorSeverity.ERROR, EcompLoggerErrorCode.BUSINESS_PROCESS_ERROR,
99                 " Failed to perform tosca update for substitution filter in service {} , component instance {}. status is {}", componentId,
100                 "componentInstanceId", operationStatus);
101             return Either.right(operationStatus);
102         }
103         janusGraphDao.commit();
104         return Either.left(substitutionFilterDataDefinition);
105     }
106
107     private StorageOperationStatus performUpdateToscaAction(final boolean isUpdate, final GraphVertex graphVertex,
108                                                             final List<SubstitutionFilterDataDefinition> toscaDataList) {
109         if (isUpdate) {
110             return updateToscaDataOfToscaElement(graphVertex, EdgeLabelEnum.SUBSTITUTION_FILTER_TEMPLATE, VertexTypeEnum.SUBSTITUTION_FILTER_TEMPLATE,
111                 toscaDataList, JsonPresentationFields.UNIQUE_ID);
112         } else {
113             return addToscaDataToToscaElement(graphVertex, EdgeLabelEnum.SUBSTITUTION_FILTER_TEMPLATE, VertexTypeEnum.SUBSTITUTION_FILTER_TEMPLATE,
114                 toscaDataList, JsonPresentationFields.UNIQUE_ID);
115         }
116     }
117 }