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

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * SDC
   4  * ================================================================================
   5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.openecomp.sdc.be.model.jsonjanusgraph.operations;
  22
  23 import com.google.common.collect.ImmutableList;
  24 import com.google.common.collect.ImmutableSet;
  25 import fj.data.Either;
  26 import java.util.ArrayList;
  27 import java.util.List;
  28 import java.util.Set;
  29
  30 import org.openecomp.sdc.be.dao.janusgraph.JanusGraphOperationStatus;
  31 import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
  32 import org.openecomp.sdc.be.dao.jsongraph.types.EdgeLabelEnum;
  33 import org.openecomp.sdc.be.dao.jsongraph.types.JsonParseFlagEnum;
  34 import org.openecomp.sdc.be.dao.jsongraph.types.VertexTypeEnum;
  35 import org.openecomp.sdc.be.datatypes.elements.CINodeFilterDataDefinition;
  36 import org.openecomp.sdc.be.datatypes.elements.ListDataDefinition;
  37 import org.openecomp.sdc.be.datatypes.elements.RequirementNodeFilterPropertyDataDefinition;
  38 import org.openecomp.sdc.be.datatypes.enums.JsonPresentationFields;
  39 import org.openecomp.sdc.be.model.Service;
  40 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
  41 import org.openecomp.sdc.be.model.operations.impl.DaoStatusConverter;
  42 import org.openecomp.sdc.common.jsongraph.util.CommonUtility;
  43 import org.openecomp.sdc.common.log.wrappers.Logger;
  44
  45 @org.springframework.stereotype.Component("service-filter-operations")
  46 public class NodeFilterOperation extends BaseOperation {
  47
  48     private static Logger logger = Logger.getLogger(NodeFilterOperation.class);
  49
  50     public Either<Set<String>, StorageOperationStatus> deleteNodeFilters(Service service,
  51             Set<String> componentInstanceIds) {
  52         Either<GraphVertex, JanusGraphOperationStatus> getComponentVertex;
  53         Either<GraphVertex, JanusGraphOperationStatus> getNodeFilterVertex;
  54         StorageOperationStatus status;
  55
  56         getComponentVertex = janusGraphDao.getVertexById(service.getUniqueId(), JsonParseFlagEnum.NoParse);
  57         if (getComponentVertex.isRight()) {
  58             return Either.right(
  59                     DaoStatusConverter.convertJanusGraphStatusToStorageStatus(getComponentVertex.right().value()));
  60         }
  61
  62         getNodeFilterVertex =
  63                 janusGraphDao.getChildVertex(getComponentVertex.left().value(), EdgeLabelEnum.NODE_FILTER_TEMPLATE,
  64                         JsonParseFlagEnum.NoParse);
  65         if (getNodeFilterVertex.isLeft()) {
  66             status = deleteToscaDataElements(service.getUniqueId(), EdgeLabelEnum.NODE_FILTER_TEMPLATE,
  67                     new ArrayList<>(componentInstanceIds));
  68             if (status != StorageOperationStatus.OK) {
  69                 return Either.right(status);
  70             }
  71         }
  72
  73         return Either.left(componentInstanceIds);
  74     }
  75
  76
  77     public Either<String, StorageOperationStatus> deleteNodeFilter(Service service, String componentInstanceId) {
  78         final Either<Set<String>, StorageOperationStatus> listStorageOperationStatusEither =
  79                 deleteNodeFilters(service, ImmutableSet.of(componentInstanceId));
  80         if (listStorageOperationStatusEither.isRight()) {
  81             return Either.right(listStorageOperationStatusEither.right().value());
  82         }
  83         return Either.left(componentInstanceId);
  84     }
  85
  86
  87     public Either<CINodeFilterDataDefinition, StorageOperationStatus> createNodeFilter(String serviceId,
  88             String componentInstanceId) {
  89         CINodeFilterDataDefinition nodeFilterDataDefinition = new CINodeFilterDataDefinition();
  90         return addOrUpdateNodeFilter(false, serviceId, componentInstanceId, nodeFilterDataDefinition);
  91     }
  92
  93     public Either<CINodeFilterDataDefinition, StorageOperationStatus> deleteConstraint(String serviceId,
  94             String componentInstanceId, CINodeFilterDataDefinition nodeFilterDataDefinition, int propertyIndex) {
  95         ListDataDefinition<RequirementNodeFilterPropertyDataDefinition> properties =
  96                 nodeFilterDataDefinition.getProperties();
  97         properties.getListToscaDataDefinition().remove(propertyIndex);
  98         nodeFilterDataDefinition.setProperties(properties);
  99         return addOrUpdateNodeFilter(true, serviceId, componentInstanceId, nodeFilterDataDefinition);
 100     }
 101
 102     public Either<CINodeFilterDataDefinition, StorageOperationStatus> addNewProperty(String serviceId,
 103             String componentInstanceId, CINodeFilterDataDefinition nodeFilterDataDefinition,
 104             RequirementNodeFilterPropertyDataDefinition requirementNodeFilterPropertyDataDefinition) {
 105         ListDataDefinition<RequirementNodeFilterPropertyDataDefinition> properties =
 106                 nodeFilterDataDefinition.getProperties();
 107         if (properties == null) {
 108             properties = new ListDataDefinition<>();
 109             nodeFilterDataDefinition.setProperties(properties);
 110         }
 111         properties.getListToscaDataDefinition().add(requirementNodeFilterPropertyDataDefinition);
 112         nodeFilterDataDefinition.setProperties(properties);
 113         return addOrUpdateNodeFilter(true, serviceId, componentInstanceId, nodeFilterDataDefinition);
 114     }
 115
 116     public Either<CINodeFilterDataDefinition, StorageOperationStatus> updateProperties(String serviceId,
 117             String componentInstanceId, CINodeFilterDataDefinition nodeFilterDataDefinition,
 118             List<RequirementNodeFilterPropertyDataDefinition> requirementNodeFilterPropertyDataDefinition) {
 119         ListDataDefinition<RequirementNodeFilterPropertyDataDefinition> properties =
 120                 nodeFilterDataDefinition.getProperties();
 121         properties.getListToscaDataDefinition().clear();
 122         properties.getListToscaDataDefinition().addAll(requirementNodeFilterPropertyDataDefinition);
 123         nodeFilterDataDefinition.setProperties(properties);
 124         return addOrUpdateNodeFilter(true, serviceId, componentInstanceId, nodeFilterDataDefinition);
 125     }
 126
 127     public Either<CINodeFilterDataDefinition, StorageOperationStatus> updateNodeFilter(String serviceId,
 128             String componentInstanceId, CINodeFilterDataDefinition ciNodeFilterDataDefinition) {
 129         return addOrUpdateNodeFilter(true, serviceId, componentInstanceId, ciNodeFilterDataDefinition);
 130     }
 131
 132     private Either<CINodeFilterDataDefinition, StorageOperationStatus> addOrUpdateNodeFilter(boolean isUpdateAction,
 133             String serviceId, String componentInstanceId, CINodeFilterDataDefinition ciNodeFilterDataDefinition) {
 134
 135         StorageOperationStatus statusRes;
 136         Either<GraphVertex, JanusGraphOperationStatus> getToscaElementRes;
 137
 138         getToscaElementRes = janusGraphDao.getVertexById(serviceId, JsonParseFlagEnum.NoParse);
 139         if (getToscaElementRes.isRight()) {
 140             JanusGraphOperationStatus status = getToscaElementRes.right().value();
 141             CommonUtility.addRecordToLog(logger, CommonUtility.LogLevelEnum.DEBUG,
 142                     "Failed to get tosca element {} upon adding the properties. Status is {}. ", serviceId, status);
 143             statusRes = DaoStatusConverter.convertJanusGraphStatusToStorageStatus(status);
 144             return Either.right(statusRes);
 145         }
 146         GraphVertex serviceVertex = getToscaElementRes.left().value();
 147         ciNodeFilterDataDefinition.setID(componentInstanceId);
 148         statusRes = performUpdateToscaAction(isUpdateAction, serviceVertex, ImmutableList.of(ciNodeFilterDataDefinition));
 149         if (!statusRes.equals(StorageOperationStatus.OK)) {
 150             janusGraphDao.rollback();
 151             logger.error(
 152                     " Failed to perform tosca update for node filter in service {} , component instance {}. status is {}",
 153                     serviceId, componentInstanceId, statusRes);
 154             return Either.right(statusRes);
 155         }
 156         janusGraphDao.commit();
 157         return Either.left(ciNodeFilterDataDefinition);
 158
 159     }
 160
 161
 162     private StorageOperationStatus performUpdateToscaAction(boolean isUpdate, GraphVertex graphVertex,
 163             List<CINodeFilterDataDefinition> toscaDataList) {
 164         if (isUpdate) {
 165             return updateToscaDataOfToscaElement(graphVertex, EdgeLabelEnum.NODE_FILTER_TEMPLATE,
 166                     VertexTypeEnum.NODE_FILTER_TEMPLATE, toscaDataList, JsonPresentationFields.UNIQUE_ID);
 167         } else {
 168             return addToscaDataToToscaElement(graphVertex, EdgeLabelEnum.NODE_FILTER_TEMPLATE,
 169                     VertexTypeEnum.NODE_FILTER_TEMPLATE, toscaDataList, JsonPresentationFields.UNIQUE_ID);
 170         }
 171     }
 172
 173 }
 174
 175
 176