catalog-model/src/main/java/org/openecomp/sdc/be/model/jsonjanusgraph/operations/RequirementOperation.java

   1 /*
   2  * Copyright © 2016-2018 European Support Limited
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  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 package org.openecomp.sdc.be.model.jsonjanusgraph.operations;
  17
  18 import fj.data.Either;
  19 import java.util.ArrayList;
  20 import java.util.Collections;
  21 import java.util.List;
  22 import org.openecomp.sdc.be.dao.jsongraph.types.EdgeLabelEnum;
  23 import org.openecomp.sdc.be.dao.jsongraph.types.VertexTypeEnum;
  24 import org.openecomp.sdc.be.datatypes.elements.ListRequirementDataDefinition;
  25 import org.openecomp.sdc.be.datatypes.elements.RequirementDataDefinition;
  26 import org.openecomp.sdc.be.datatypes.enums.JsonPresentationFields;
  27 import org.openecomp.sdc.be.model.Component;
  28 import org.openecomp.sdc.be.model.RequirementDefinition;
  29 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
  30 import org.slf4j.Logger;
  31 import org.slf4j.LoggerFactory;
  32
  33 @org.springframework.stereotype.Component("requirement-operation")
  34 public class RequirementOperation extends BaseOperation {
  35
  36     private static final Logger LOGGER = LoggerFactory.getLogger(RequirementOperation.class);
  37
  38     private static ListRequirementDataDefinition convertToListRequirementDataDefinition(List<RequirementDefinition> requirementDefinitions) {
  39         List<RequirementDataDefinition> requirementDataDefinitions = new ArrayList<>(requirementDefinitions);
  40         return new ListRequirementDataDefinition(requirementDataDefinitions);
  41     }
  42
  43     public Either<List<RequirementDefinition>, StorageOperationStatus> addRequirement(String componentId,
  44                                                                                       List<RequirementDefinition> requirementDefinitions) {
  45         return addOrUpdateRequirements(componentId, requirementDefinitions, false);
  46     }
  47
  48     public Either<List<RequirementDefinition>, StorageOperationStatus> updateRequirement(String componentId,
  49                                                                                          List<RequirementDefinition> requirementDefinitions) {
  50         return addOrUpdateRequirements(componentId, requirementDefinitions, true);
  51     }
  52
  53     private Either<List<RequirementDefinition>, StorageOperationStatus> addOrUpdateRequirements(String componentId,
  54                                                                                                 List<RequirementDefinition> requirementDefinitions,
  55                                                                                                 boolean isUpdateAction) {
  56         StorageOperationStatus statusRes = performUpdateToscaAction(isUpdateAction, componentId,
  57             Collections.singletonList(convertToListRequirementDataDefinition(requirementDefinitions)));
  58         if (!statusRes.equals(StorageOperationStatus.OK)) {
  59             janusGraphDao.rollback();
  60             LOGGER.error("Failed to find the parent capability of capability type {}." + " status is {}", componentId, statusRes);
  61             return Either.right(statusRes);
  62         }
  63         janusGraphDao.commit();
  64         return Either.left(requirementDefinitions);
  65     }
  66
  67     public StorageOperationStatus deleteRequirements(Component component, String requirementToDelete) {
  68         return deleteToscaDataElements(component.getUniqueId(), EdgeLabelEnum.REQUIREMENTS, Collections.singletonList(requirementToDelete));
  69     }
  70
  71     private StorageOperationStatus performUpdateToscaAction(boolean isUpdate, String componentId, List<ListRequirementDataDefinition> toscaDataList) {
  72         if (isUpdate) {
  73             return updateToscaDataOfToscaElement(componentId, EdgeLabelEnum.REQUIREMENTS, VertexTypeEnum.REQUIREMENTS, toscaDataList,
  74                 JsonPresentationFields.CAPABILITY);
  75         } else {
  76             return addToscaDataToToscaElement(componentId, EdgeLabelEnum.REQUIREMENTS, VertexTypeEnum.REQUIREMENTS, toscaDataList,
  77                 JsonPresentationFields.CAPABILITY);
  78         }
  79     }
  80 }