gerrit.onap Code Review - sdc.git/blob

   1 package org.openecomp.sdc.be.model.operations.impl;
   2
   3 import org.apache.commons.lang3.tuple.Pair;
   4 import org.openecomp.sdc.be.dao.graph.datatype.GraphNode;
   5 import org.openecomp.sdc.be.dao.graph.datatype.GraphRelation;
   6 import org.openecomp.sdc.be.dao.neo4j.GraphEdgeLabels;
   7 import org.openecomp.sdc.be.dao.titan.TitanGenericDao;
   8 import org.openecomp.sdc.be.dao.titan.TitanOperationStatus;
   9 import org.openecomp.sdc.be.datatypes.enums.NodeTypeEnum;
  10 import org.openecomp.sdc.be.model.operations.api.DerivedFromOperation;
  11 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
  12 import org.openecomp.sdc.be.resources.data.UniqueIdData;
  13 import org.slf4j.Logger;
  14 import org.slf4j.LoggerFactory;
  15 import org.springframework.stereotype.Component;
  16
  17 import fj.data.Either;
  18
  19 @Component
  20 public class DefaultDerivedFromOperation implements DerivedFromOperation {
  21
  22     private static final Logger log = LoggerFactory.getLogger(DefaultDerivedFromOperation.class);
  23     private TitanGenericDao titanGenericDao;
  24
  25     public DefaultDerivedFromOperation(TitanGenericDao titanGenericDao) {
  26         this.titanGenericDao = titanGenericDao;
  27     }
  28
  29     @Override
  30     public Either<GraphRelation, StorageOperationStatus> addDerivedFromRelation(String parentUniqueId, String derivedFromUniqueId, NodeTypeEnum nodeType) {
  31         UniqueIdData from = new UniqueIdData(NodeTypeEnum.PolicyType, parentUniqueId);
  32         UniqueIdData to = new UniqueIdData(NodeTypeEnum.PolicyType, derivedFromUniqueId);
  33         return titanGenericDao.createRelation(from, to, GraphEdgeLabels.DERIVED_FROM, null)
  34                 .right()
  35                 .map(DaoStatusConverter::convertTitanStatusToStorageStatus);
  36     }
  37
  38     @Override
  39     public <T extends GraphNode> Either<T, StorageOperationStatus> getDerivedFromChild(String uniqueId, NodeTypeEnum nodeType, Class<T> clazz) {
  40         log.debug("#getDerivedFromChild - fetching derived from entity for node type {} with id {}", nodeType, uniqueId);
  41         return titanGenericDao.getChild(UniqueIdBuilder.getKeyByNodeType(nodeType), uniqueId, GraphEdgeLabels.DERIVED_FROM, nodeType, clazz)
  42                 .bimap(Pair::getKey,
  43                        DaoStatusConverter::convertTitanStatusToStorageStatus);
  44     }
  45
  46     @Override
  47     public StorageOperationStatus removeDerivedFromRelation(String uniqueId, String derivedFromUniqueId, NodeTypeEnum nodeType) {
  48         UniqueIdData from = new UniqueIdData(NodeTypeEnum.PolicyType, uniqueId);
  49         UniqueIdData to = new UniqueIdData(NodeTypeEnum.PolicyType, derivedFromUniqueId);
  50         return isDerivedFromExists(from, to)
  51                 .either(isRelationExist -> isRelationExist ? deleteDerivedFrom(from, to) : StorageOperationStatus.OK,
  52                         DaoStatusConverter::convertTitanStatusToStorageStatus);
  53
  54
  55     }
  56
  57     private StorageOperationStatus deleteDerivedFrom(UniqueIdData from,  UniqueIdData to) {
  58         return titanGenericDao.deleteRelation(from, to, GraphEdgeLabels.DERIVED_FROM)
  59                 .either(deletedRelation -> StorageOperationStatus.OK,
  60                         DaoStatusConverter::convertTitanStatusToStorageStatus);
  61     }
  62
  63     private Either<Boolean, TitanOperationStatus> isDerivedFromExists(UniqueIdData from, UniqueIdData to) {
  64         return titanGenericDao.isRelationExist(from, to, GraphEdgeLabels.DERIVED_FROM);
  65     }
  66
  67
  68 }