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
  17 package org.openecomp.sdc.be.model.jsonjanusgraph.operations;
  18
  19 import fj.data.Either;
  20 import org.openecomp.sdc.be.dao.jsongraph.types.EdgeLabelEnum;
  21 import org.openecomp.sdc.be.dao.jsongraph.types.VertexTypeEnum;
  22 import org.openecomp.sdc.be.datatypes.elements.ListRequirementDataDefinition;
  23 import org.openecomp.sdc.be.datatypes.elements.RequirementDataDefinition;
  24 import org.openecomp.sdc.be.datatypes.enums.JsonPresentationFields;
  25 import org.openecomp.sdc.be.model.Component;
  26 import org.openecomp.sdc.be.model.RequirementDefinition;
  27 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
  28 import org.slf4j.Logger;
  29 import org.slf4j.LoggerFactory;
  30
  31 import java.util.ArrayList;
  32 import java.util.Collections;
  33 import java.util.List;
  34
  35 @org.springframework.stereotype.Component("requirement-operation")
  36 public class RequirementOperation extends BaseOperation {
  37     private static final Logger LOGGER = LoggerFactory.getLogger(RequirementOperation.class);
  38
  39     public Either<List<RequirementDefinition>, StorageOperationStatus> addRequirement(
  40             String componentId,
  41             List<RequirementDefinition> requirementDefinitions) {
  42         return addOrUpdateRequirements( componentId, requirementDefinitions, false);
  43     }
  44
  45     public Either<List<RequirementDefinition>, StorageOperationStatus> updateRequirement(
  46             String componentId,
  47             List<RequirementDefinition> requirementDefinitions) {
  48         return addOrUpdateRequirements( componentId, requirementDefinitions, true);
  49     }
  50
  51
  52     private Either<List<RequirementDefinition>, StorageOperationStatus> addOrUpdateRequirements(String componentId,
  53                                                                  List<RequirementDefinition> requirementDefinitions,
  54                                                                                                 boolean isUpdateAction) {
  55
  56         StorageOperationStatus statusRes = performUpdateToscaAction(isUpdateAction,
  57                 componentId, Collections
  58                         .singletonList(convertToListRequirementDataDefinition(requirementDefinitions)));
  59         if (!statusRes.equals(StorageOperationStatus.OK)) {
  60             janusGraphDao.rollback();
  61             LOGGER.error("Failed to find the parent capability of capability type {}."
  62                     + " status is {}", componentId, statusRes);
  63             return Either.right(statusRes);
  64         }
  65         janusGraphDao.commit();
  66         return Either.left(requirementDefinitions);
  67     }
  68
  69     public StorageOperationStatus deleteRequirements(Component component,
  70                                                      String requirementToDelete) {
  71         return deleteToscaDataElements(component.getUniqueId(),
  72                 EdgeLabelEnum.REQUIREMENTS, Collections.singletonList(requirementToDelete));
  73     }
  74
  75     private static ListRequirementDataDefinition convertToListRequirementDataDefinition(
  76             List<RequirementDefinition> requirementDefinitions) {
  77         List<RequirementDataDefinition> requirementDataDefinitions =
  78                 new ArrayList<>(requirementDefinitions);
  79         return new ListRequirementDataDefinition(requirementDataDefinitions);
  80     }
  81
  82     private StorageOperationStatus performUpdateToscaAction(boolean isUpdate,
  83                                                             String componentId, List<ListRequirementDataDefinition> toscaDataList) {
  84         if (isUpdate) {
  85             return updateToscaDataOfToscaElement(componentId, EdgeLabelEnum.REQUIREMENTS,
  86                     VertexTypeEnum.REQUIREMENTS, toscaDataList, JsonPresentationFields.CAPABILITY);
  87         } else {
  88             return addToscaDataToToscaElement(componentId, EdgeLabelEnum.REQUIREMENTS,
  89                     VertexTypeEnum.REQUIREMENTS, toscaDataList, JsonPresentationFields.CAPABILITY);
  90         }
  91     }
  92 }