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
 
   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.
 
  16  *  SPDX-License-Identifier: Apache-2.0
 
  17  *  ============LICENSE_END=========================================================
 
  20 package org.openecomp.sdc.be.model.jsonjanusgraph.operations;
 
  22 import com.google.common.collect.ImmutableList;
 
  23 import fj.data.Either;
 
  24 import java.util.List;
 
  25 import java.util.Objects;
 
  26 import org.openecomp.sdc.be.dao.janusgraph.JanusGraphOperationStatus;
 
  27 import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
 
  28 import org.openecomp.sdc.be.dao.jsongraph.types.EdgeLabelEnum;
 
  29 import org.openecomp.sdc.be.dao.jsongraph.types.JsonParseFlagEnum;
 
  30 import org.openecomp.sdc.be.dao.jsongraph.types.VertexTypeEnum;
 
  31 import org.openecomp.sdc.be.datatypes.elements.ListDataDefinition;
 
  32 import org.openecomp.sdc.be.datatypes.elements.RequirementSubstitutionFilterPropertyDataDefinition;
 
  33 import org.openecomp.sdc.be.datatypes.elements.SubstitutionFilterDataDefinition;
 
  34 import org.openecomp.sdc.be.datatypes.enums.JsonPresentationFields;
 
  35 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
 
  36 import org.openecomp.sdc.be.model.operations.impl.DaoStatusConverter;
 
  37 import org.openecomp.sdc.common.jsongraph.util.CommonUtility;
 
  38 import org.openecomp.sdc.common.log.enums.EcompErrorSeverity;
 
  39 import org.openecomp.sdc.common.log.enums.EcompLoggerErrorCode;
 
  40 import org.openecomp.sdc.common.log.wrappers.Logger;
 
  42 @org.springframework.stereotype.Component("substitution-filter-operations")
 
  43 public class SubstitutionFilterOperation extends BaseOperation {
 
  45     private static final Logger LOGGER = Logger.getLogger(SubstitutionFilterOperation.class);
 
  47     public Either<SubstitutionFilterDataDefinition, StorageOperationStatus> createSubstitutionFilter(
 
  48             final String componentId) {
 
  50         return addOrUpdateSubstitutionFilter(false, componentId, new SubstitutionFilterDataDefinition());
 
  53     public Either<SubstitutionFilterDataDefinition, StorageOperationStatus> deleteConstraint(
 
  54         final String serviceId, final SubstitutionFilterDataDefinition substitutionFilterDataDefinition,
 
  55         final int propertyIndex) {
 
  57         final ListDataDefinition<RequirementSubstitutionFilterPropertyDataDefinition> properties =
 
  58                 substitutionFilterDataDefinition.getProperties();
 
  59         properties.getListToscaDataDefinition().remove(propertyIndex);
 
  60         substitutionFilterDataDefinition.setProperties(properties);
 
  62         return addOrUpdateSubstitutionFilter(true, serviceId, substitutionFilterDataDefinition);
 
  65     public Either<SubstitutionFilterDataDefinition, StorageOperationStatus> addPropertyFilter(
 
  66         final String componentId, final SubstitutionFilterDataDefinition substitutionFilterDataDefinition,
 
  67         final RequirementSubstitutionFilterPropertyDataDefinition substitutionFilterPropertyDataDefinition) {
 
  69         final SubstitutionFilterDataDefinition substitutionFilterDataDefinition1 =
 
  70             Objects.requireNonNullElseGet(substitutionFilterDataDefinition, SubstitutionFilterDataDefinition::new);
 
  71         final ListDataDefinition<RequirementSubstitutionFilterPropertyDataDefinition> properties =
 
  72             Objects.requireNonNullElseGet(substitutionFilterDataDefinition1.getProperties(), ListDataDefinition::new);
 
  73         properties.getListToscaDataDefinition().add(substitutionFilterPropertyDataDefinition);
 
  74         substitutionFilterDataDefinition1.setProperties(properties);
 
  75         return addOrUpdateSubstitutionFilter(true, componentId, substitutionFilterDataDefinition1);
 
  78     public Either<SubstitutionFilterDataDefinition, StorageOperationStatus> updateProperties(
 
  79         final String componentId, final SubstitutionFilterDataDefinition substitutionFilterDataDefinition,
 
  80         final List<RequirementSubstitutionFilterPropertyDataDefinition> requirementSubstitutionFilterPropertyDataDefinition) {
 
  82         final ListDataDefinition<RequirementSubstitutionFilterPropertyDataDefinition> properties =
 
  83             substitutionFilterDataDefinition.getProperties();
 
  84         properties.getListToscaDataDefinition().clear();
 
  85         properties.getListToscaDataDefinition().addAll(requirementSubstitutionFilterPropertyDataDefinition);
 
  86         substitutionFilterDataDefinition.setProperties(properties);
 
  87         return addOrUpdateSubstitutionFilter(true, componentId, substitutionFilterDataDefinition);
 
  90     private Either<SubstitutionFilterDataDefinition, StorageOperationStatus> addOrUpdateSubstitutionFilter(
 
  91         final boolean isUpdateAction, final String componentId,
 
  92         final SubstitutionFilterDataDefinition substitutionFilterDataDefinition) {
 
  94         final Either<GraphVertex, JanusGraphOperationStatus> toscaElementEither =
 
  95             janusGraphDao.getVertexById(componentId, JsonParseFlagEnum.NoParse);
 
  96         if (toscaElementEither.isRight()) {
 
  97             final JanusGraphOperationStatus status = toscaElementEither.right().value();
 
  98             CommonUtility.addRecordToLog(LOGGER, CommonUtility.LogLevelEnum.DEBUG,
 
  99                 "Failed to get tosca element {} upon adding the properties. Status is {}. ", componentId, status);
 
 100             return Either.right(DaoStatusConverter.convertJanusGraphStatusToStorageStatus(status));
 
 102         final GraphVertex serviceVertex = toscaElementEither.left().value();
 
 103         substitutionFilterDataDefinition.setID(componentId);
 
 104         final StorageOperationStatus operationStatus = performUpdateToscaAction(isUpdateAction, serviceVertex,
 
 105             ImmutableList.of(substitutionFilterDataDefinition));
 
 106         if (!StorageOperationStatus.OK.equals(operationStatus)) {
 
 107             janusGraphDao.rollback();
 
 108             LOGGER.error(EcompErrorSeverity.ERROR, EcompLoggerErrorCode.BUSINESS_PROCESS_ERROR,
 
 109                 " Failed to perform tosca update for substitution filter in service {} , component instance {}. status is {}",
 
 110                 componentId, "componentInstanceId", operationStatus);
 
 111             return Either.right(operationStatus);
 
 113         janusGraphDao.commit();
 
 114         return Either.left(substitutionFilterDataDefinition);
 
 117     private StorageOperationStatus performUpdateToscaAction(final boolean isUpdate,
 
 118                                                             final GraphVertex graphVertex,
 
 119                                                             final List<SubstitutionFilterDataDefinition> toscaDataList) {
 
 121             return updateToscaDataOfToscaElement(graphVertex, EdgeLabelEnum.SUBSTITUTION_FILTER_TEMPLATE,
 
 122                 VertexTypeEnum.SUBSTITUTION_FILTER_TEMPLATE, toscaDataList, JsonPresentationFields.UNIQUE_ID);
 
 124             return addToscaDataToToscaElement(graphVertex, EdgeLabelEnum.SUBSTITUTION_FILTER_TEMPLATE,
 
 125                 VertexTypeEnum.SUBSTITUTION_FILTER_TEMPLATE, toscaDataList, JsonPresentationFields.UNIQUE_ID);