catalog-model/src/main/java/org/openecomp/sdc/be/model/jsontitan/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.jsontitan.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 import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
  30 import org.openecomp.sdc.be.dao.jsongraph.types.EdgeLabelEnum;
  31 import org.openecomp.sdc.be.dao.jsongraph.types.JsonParseFlagEnum;
  32 import org.openecomp.sdc.be.dao.jsongraph.types.VertexTypeEnum;
  33 import org.openecomp.sdc.be.dao.titan.TitanOperationStatus;
  34 import org.openecomp.sdc.be.datatypes.elements.CINodeFilterDataDefinition;
  35 import org.openecomp.sdc.be.datatypes.elements.ListDataDefinition;
  36 import org.openecomp.sdc.be.datatypes.elements.RequirementNodeFilterPropertyDataDefinition;
  37 import org.openecomp.sdc.be.datatypes.enums.JsonPresentationFields;
  38 import org.openecomp.sdc.be.model.Service;
  39 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
  40 import org.openecomp.sdc.be.model.operations.impl.DaoStatusConverter;
  41 import org.openecomp.sdc.common.jsongraph.util.CommonUtility;
  42 import org.openecomp.sdc.common.log.wrappers.Logger;
  43
  44 @org.springframework.stereotype.Component("service-filter-operations")
  45 public class NodeFilterOperation extends BaseOperation {
  46
  47     private static Logger logger = Logger.getLogger(NodeFilterOperation.class);
  48
  49     public Either<Set<String>, StorageOperationStatus> deleteNodeFilters(Service service,
  50             Set<String> componentInstanceIds) {
  51         Either<GraphVertex, TitanOperationStatus> getComponentVertex;
  52         Either<GraphVertex, TitanOperationStatus> getNodeFilterVertex;
  53         StorageOperationStatus status;
  54
  55         getComponentVertex = titanDao.getVertexById(service.getUniqueId(), JsonParseFlagEnum.NoParse);
  56         if (getComponentVertex.isRight()) {
  57             return Either.right(
  58                     DaoStatusConverter.convertTitanStatusToStorageStatus(getComponentVertex.right().value()));
  59         }
  60
  61         getNodeFilterVertex =
  62                 titanDao.getChildVertex(getComponentVertex.left().value(), EdgeLabelEnum.NODE_FILTER_TEMPLATE,
  63                         JsonParseFlagEnum.NoParse);
  64         if (getNodeFilterVertex.isLeft()) {
  65             status = deleteToscaDataElements(service.getUniqueId(), EdgeLabelEnum.NODE_FILTER_TEMPLATE,
  66                     new ArrayList<>(componentInstanceIds));
  67             if (status != StorageOperationStatus.OK) {
  68                 return Either.right(status);
  69             }
  70         }
  71
  72         return Either.left(componentInstanceIds);
  73     }
  74
  75
  76     public Either<String, StorageOperationStatus> deleteNodeFilter(Service service, String componentInstanceId) {
  77         final Either<Set<String>, StorageOperationStatus> listStorageOperationStatusEither =
  78                 deleteNodeFilters(service, ImmutableSet.of(componentInstanceId));
  79         if (listStorageOperationStatusEither.isRight()) {
  80             return Either.right(listStorageOperationStatusEither.right().value());
  81         }
  82         return Either.left(componentInstanceId);
  83     }
  84
  85
  86     public Either<CINodeFilterDataDefinition, StorageOperationStatus> createNodeFilter(String serviceId,
  87             String componentInstanceId) {
  88         CINodeFilterDataDefinition nodeFilterDataDefinition = new CINodeFilterDataDefinition();
  89         return addOrUpdateNodeFilter(false, serviceId, componentInstanceId, nodeFilterDataDefinition);
  90     }
  91
  92     public Either<CINodeFilterDataDefinition, StorageOperationStatus> deleteConstraint(String serviceId,
  93             String componentInstanceId, CINodeFilterDataDefinition nodeFilterDataDefinition, int propertyIndex) {
  94         ListDataDefinition<RequirementNodeFilterPropertyDataDefinition> properties =
  95                 nodeFilterDataDefinition.getProperties();
  96         properties.getListToscaDataDefinition().remove(propertyIndex);
  97         nodeFilterDataDefinition.setProperties(properties);
  98         return addOrUpdateNodeFilter(true, serviceId, componentInstanceId, nodeFilterDataDefinition);
  99     }
 100
 101     public Either<CINodeFilterDataDefinition, StorageOperationStatus> addNewProperty(String serviceId,
 102             String componentInstanceId, CINodeFilterDataDefinition nodeFilterDataDefinition,
 103             RequirementNodeFilterPropertyDataDefinition requirementNodeFilterPropertyDataDefinition) {
 104         ListDataDefinition<RequirementNodeFilterPropertyDataDefinition> properties =
 105                 nodeFilterDataDefinition.getProperties();
 106         if (properties == null) {
 107             properties = new ListDataDefinition<>();
 108             nodeFilterDataDefinition.setProperties(properties);
 109         }
 110         properties.getListToscaDataDefinition().add(requirementNodeFilterPropertyDataDefinition);
 111         nodeFilterDataDefinition.setProperties(properties);
 112         return addOrUpdateNodeFilter(true, serviceId, componentInstanceId, nodeFilterDataDefinition);
 113     }
 114
 115     public Either<CINodeFilterDataDefinition, StorageOperationStatus> updateProperties(String serviceId,
 116             String componentInstanceId, CINodeFilterDataDefinition nodeFilterDataDefinition,
 117             List<RequirementNodeFilterPropertyDataDefinition> requirementNodeFilterPropertyDataDefinition) {
 118         ListDataDefinition<RequirementNodeFilterPropertyDataDefinition> properties =
 119                 nodeFilterDataDefinition.getProperties();
 120         properties.getListToscaDataDefinition().clear();
 121         properties.getListToscaDataDefinition().addAll(requirementNodeFilterPropertyDataDefinition);
 122         nodeFilterDataDefinition.setProperties(properties);
 123         return addOrUpdateNodeFilter(true, serviceId, componentInstanceId, nodeFilterDataDefinition);
 124     }
 125
 126     public Either<CINodeFilterDataDefinition, StorageOperationStatus> updateNodeFilter(String serviceId,
 127             String componentInstanceId, CINodeFilterDataDefinition ciNodeFilterDataDefinition) {
 128         return addOrUpdateNodeFilter(true, serviceId, componentInstanceId, ciNodeFilterDataDefinition);
 129     }
 130
 131     private Either<CINodeFilterDataDefinition, StorageOperationStatus> addOrUpdateNodeFilter(boolean isUpdateAction,
 132             String serviceId, String componentInstanceId, CINodeFilterDataDefinition ciNodeFilterDataDefinition) {
 133
 134         StorageOperationStatus statusRes;
 135         Either<GraphVertex, TitanOperationStatus> getToscaElementRes;
 136
 137         getToscaElementRes = titanDao.getVertexById(serviceId, JsonParseFlagEnum.NoParse);
 138         if (getToscaElementRes.isRight()) {
 139             TitanOperationStatus status = getToscaElementRes.right().value();
 140             CommonUtility.addRecordToLog(logger, CommonUtility.LogLevelEnum.DEBUG,
 141                     "Failed to get tosca element {} upon adding the properties. Status is {}. ", serviceId, status);
 142             statusRes = DaoStatusConverter.convertTitanStatusToStorageStatus(status);
 143             return Either.right(statusRes);
 144         }
 145         GraphVertex serviceVertex = getToscaElementRes.left().value();
 146         ciNodeFilterDataDefinition.setID(componentInstanceId);
 147         statusRes = performUpdateToscaAction(isUpdateAction, serviceVertex, ImmutableList.of(ciNodeFilterDataDefinition));
 148         if (!statusRes.equals(StorageOperationStatus.OK)) {
 149             titanDao.rollback();
 150             logger.error(
 151                     " Failed to perform tosca update for node filter in service {} , component instance {}. status is {}",
 152                     serviceId, componentInstanceId, statusRes);
 153             return Either.right(statusRes);
 154         }
 155         titanDao.commit();
 156         return Either.left(ciNodeFilterDataDefinition);
 157
 158     }
 159
 160
 161     private StorageOperationStatus performUpdateToscaAction(boolean isUpdate, GraphVertex graphVertex,
 162             List<CINodeFilterDataDefinition> toscaDataList) {
 163         if (isUpdate) {
 164             return updateToscaDataOfToscaElement(graphVertex, EdgeLabelEnum.NODE_FILTER_TEMPLATE,
 165                     VertexTypeEnum.NODE_FILTER_TEMPLATE, toscaDataList, JsonPresentationFields.UNIQUE_ID);
 166         } else {
 167             return addToscaDataToToscaElement(graphVertex, EdgeLabelEnum.NODE_FILTER_TEMPLATE,
 168                     VertexTypeEnum.NODE_FILTER_TEMPLATE, toscaDataList, JsonPresentationFields.UNIQUE_ID);
 169         }
 170     }
 171
 172 }
 173
 174
 175