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 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;
 
  41 @org.springframework.stereotype.Component("substitution-filter-operations")
 
  42 public class SubstitutionFilterOperation extends BaseOperation {
 
  44     private static final Logger LOGGER = Logger.getLogger(SubstitutionFilterOperation.class);
 
  46     public Either<SubstitutionFilterDataDefinition, StorageOperationStatus> createSubstitutionFilter(
 
  47         final String componentId, final String componentInstanceId) {
 
  49         return addOrUpdateSubstitutionFilter(false, componentId, componentInstanceId,
 
  50             new SubstitutionFilterDataDefinition());
 
  53     public Either<SubstitutionFilterDataDefinition, StorageOperationStatus> deleteConstraint(
 
  54         final String serviceId, final String componentInstanceId,
 
  55         final SubstitutionFilterDataDefinition substitutionFilterDataDefinition, final int propertyIndex) {
 
  57         final ListDataDefinition<RequirementSubstitutionFilterPropertyDataDefinition> properties =
 
  58             substitutionFilterDataDefinition.getProperties();
 
  59         properties.getListToscaDataDefinition().remove(propertyIndex);
 
  60         substitutionFilterDataDefinition.setProperties(properties);
 
  61         return addOrUpdateSubstitutionFilter(true, serviceId, componentInstanceId, substitutionFilterDataDefinition);
 
  64     public Either<SubstitutionFilterDataDefinition, StorageOperationStatus> addNewProperty(
 
  65         final String componentId, final String componentInstanceId,
 
  66         final SubstitutionFilterDataDefinition substitutionFilterDataDefinition,
 
  67         final RequirementSubstitutionFilterPropertyDataDefinition requirementSubstitutionFilterPropertyDataDefinition) {
 
  69         ListDataDefinition<RequirementSubstitutionFilterPropertyDataDefinition> properties =
 
  70             substitutionFilterDataDefinition.getProperties();
 
  71         if (properties == null) {
 
  72             properties = new ListDataDefinition<>();
 
  73             substitutionFilterDataDefinition.setProperties(properties);
 
  75         properties.getListToscaDataDefinition().add(requirementSubstitutionFilterPropertyDataDefinition);
 
  76         substitutionFilterDataDefinition.setProperties(properties);
 
  77         return addOrUpdateSubstitutionFilter(true, componentId, componentInstanceId, substitutionFilterDataDefinition);
 
  80     public Either<SubstitutionFilterDataDefinition, StorageOperationStatus> updateSubstitutionFilter(
 
  81         final String serviceId, final String componentInstanceId,
 
  82         final SubstitutionFilterDataDefinition substitutionFilterDataDefinition,
 
  83         final List<RequirementSubstitutionFilterPropertyDataDefinition> requirementSubstitutionFilterPropertyDataDefinitions) {
 
  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);
 
  94     private Either<SubstitutionFilterDataDefinition, StorageOperationStatus> addOrUpdateSubstitutionFilter(
 
  95         final boolean isUpdateAction, final String componentId, final String componentInstanceId,
 
  96         final SubstitutionFilterDataDefinition substitutionFilterDataDefinition) {
 
  98         StorageOperationStatus statusRes;
 
  99         Either<GraphVertex, JanusGraphOperationStatus> getToscaElementRes;
 
 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);
 
 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);
 
 120         janusGraphDao.commit();
 
 121         return Either.left(substitutionFilterDataDefinition);
 
 124     private StorageOperationStatus performUpdateToscaAction(final boolean isUpdate,
 
 125                                                             final GraphVertex graphVertex,
 
 126                                                             final List<SubstitutionFilterDataDefinition> toscaDataList) {
 
 128             return updateToscaDataOfToscaElement(graphVertex, EdgeLabelEnum.SUBSTITUTION_FILTER_TEMPLATE,
 
 129                 VertexTypeEnum.SUBSTITUTION_FILTER_TEMPLATE, toscaDataList, JsonPresentationFields.UNIQUE_ID);
 
 131             return addToscaDataToToscaElement(graphVertex, EdgeLabelEnum.SUBSTITUTION_FILTER_TEMPLATE,
 
 132                 VertexTypeEnum.SUBSTITUTION_FILTER_TEMPLATE, toscaDataList, JsonPresentationFields.UNIQUE_ID);