catalog-model/src/main/java/org/openecomp/sdc/be/model/operations/impl/CommonTypeOperations.java

   1 package org.openecomp.sdc.be.model.operations.impl;
   2
   3 import static java.util.Collections.emptyList;
   4
   5 import java.util.HashMap;
   6 import java.util.List;
   7 import java.util.Map;
   8 import java.util.Optional;
   9 import java.util.function.Consumer;
  10 import java.util.function.Function;
  11 import java.util.stream.Collectors;
  12 import org.openecomp.sdc.be.dao.graph.datatype.GraphNode;
  13 import org.openecomp.sdc.be.dao.janusgraph.HealingJanusGraphGenericDao;
  14 import org.openecomp.sdc.be.dao.neo4j.GraphPropertiesDictionary;
  15 import org.openecomp.sdc.be.dao.janusgraph.JanusGraphOperationStatus;
  16 import org.openecomp.sdc.be.datatypes.enums.NodeTypeEnum;
  17 import org.openecomp.sdc.be.model.PropertyDefinition;
  18 import org.springframework.stereotype.Component;
  19
  20 @Component
  21 public class CommonTypeOperations {
  22
  23     private final HealingJanusGraphGenericDao janusGraphGenericDao;
  24     private final PropertyOperation propertyOperation;
  25     private final OperationUtils operationUtils;
  26
  27     public CommonTypeOperations(HealingJanusGraphGenericDao janusGraphGenericDao, PropertyOperation propertyOperation,
  28                                 OperationUtils operationUtils) {
  29         this.janusGraphGenericDao = janusGraphGenericDao;
  30         this.propertyOperation = propertyOperation;
  31         this.operationUtils = operationUtils;
  32     }
  33
  34     public <T extends GraphNode> void addType(T typeData, Class<T> clazz) {
  35         janusGraphGenericDao.createNode(typeData, clazz)
  36             .left()
  37             .on(operationUtils::onJanusGraphOperationFailure);
  38     }
  39
  40     public <T extends GraphNode> Optional<T> getType(String uniqueId, Class<T> clazz, NodeTypeEnum nodeType) {
  41         T type = janusGraphGenericDao.getNode(UniqueIdBuilder.getKeyByNodeType(nodeType), uniqueId, clazz)
  42                 .left()
  43                 .on(err -> null);
  44         return Optional.ofNullable(type);
  45     }
  46
  47     public <T extends GraphNode> Optional<T> getLatestType(String type, Class<T> clazz, NodeTypeEnum nodeType) {
  48         Map<String, Object> mapCriteria = new HashMap<>();
  49         mapCriteria.put(GraphPropertiesDictionary.TYPE.getProperty(), type);
  50         mapCriteria.put(GraphPropertiesDictionary.IS_HIGHEST_VERSION.getProperty(), true);
  51         return janusGraphGenericDao.getByCriteria(nodeType, mapCriteria, clazz)
  52                 .left()
  53                 .on(err -> emptyList())
  54                 .stream()
  55                 .findFirst();
  56     }
  57
  58     public void addProperties(String uniqueId, NodeTypeEnum nodeType, List<PropertyDefinition> properties) {
  59         propertyOperation.addPropertiesToElementType(uniqueId, nodeType, properties)
  60             .left()
  61             .on(operationUtils::onJanusGraphOperationFailure);
  62     }
  63
  64     public void fillProperties(String uniqueId, NodeTypeEnum nodeType, Consumer<List<PropertyDefinition>> propertySetter) {
  65         JanusGraphOperationStatus
  66             status = propertyOperation.fillPropertiesList(uniqueId, nodeType, propertySetter);
  67         if (status!= JanusGraphOperationStatus.OK) {
  68             operationUtils.onJanusGraphOperationFailure(status);
  69         }
  70     }
  71
  72     /**
  73      * Handle update of type without dervidedFrom attribute
  74      */
  75     public  <T extends GraphNode> void updateType(T typeData, List<PropertyDefinition> properties, Class<T> clazz, NodeTypeEnum nodeType) {
  76         janusGraphGenericDao.updateNode(typeData, clazz)
  77                 .left()
  78                 .on(operationUtils::onJanusGraphOperationFailure);
  79         Map<String, PropertyDefinition> newProperties = properties.stream()
  80                 .collect(Collectors.toMap(PropertyDefinition::getName, Function.identity()));
  81         propertyOperation.mergePropertiesAssociatedToNode(nodeType, typeData.getUniqueId(), newProperties)
  82                 .left()
  83                 .on(operationUtils::onJanusGraphOperationFailure);
  84     }
  85 }