Пример #1
0
  public void saveEdge(AccumuloEdge edge) {
    ColumnVisibility edgeColumnVisibility = visibilityToAccumuloVisibility(edge.getVisibility());
    Mutation m = createMutationForEdge(edge, edgeColumnVisibility);
    saveEdgeMutation(m);

    String edgeLabel = edge.getNewEdgeLabel() != null ? edge.getNewEdgeLabel() : edge.getLabel();
    saveEdgeInfoOnVertex(edge, edgeLabel, edgeColumnVisibility);
  }
Пример #2
0
  public void alterEdgeLabel(AccumuloEdge edge, String newEdgeLabel) {
    ColumnVisibility edgeColumnVisibility = visibilityToAccumuloVisibility(edge.getVisibility());
    Mutation m = createAlterEdgeLabelMutation(edge, newEdgeLabel, edgeColumnVisibility);
    saveEdgeMutation(m);

    saveEdgeInfoOnVertex(edge, newEdgeLabel, edgeColumnVisibility);
  }
Пример #3
0
 private Mutation createAlterEdgeLabelMutation(
     AccumuloEdge edge, String newEdgeLabel, ColumnVisibility edgeColumnVisibility) {
   String edgeRowKey = edge.getId();
   Mutation m = new Mutation(edgeRowKey);
   m.putDelete(
       AccumuloEdge.CF_SIGNAL,
       new Text(edge.getLabel()),
       edgeColumnVisibility,
       currentTimeMillis());
   m.put(
       AccumuloEdge.CF_SIGNAL,
       new Text(newEdgeLabel),
       edgeColumnVisibility,
       currentTimeMillis(),
       ElementMutationBuilder.EMPTY_VALUE);
   return m;
 }
Пример #4
0
  public boolean alterElementVisibility(
      Mutation m, AccumuloElement element, Visibility newVisibility) {
    ColumnVisibility currentColumnVisibility =
        visibilityToAccumuloVisibility(element.getVisibility());
    ColumnVisibility newColumnVisibility = visibilityToAccumuloVisibility(newVisibility);
    if (currentColumnVisibility.equals(newColumnVisibility)) {
      return false;
    }

    if (element instanceof AccumuloEdge) {
      AccumuloEdge edge = (AccumuloEdge) element;
      m.putDelete(
          AccumuloEdge.CF_SIGNAL,
          new Text(edge.getLabel()),
          currentColumnVisibility,
          currentTimeMillis());
      m.put(
          AccumuloEdge.CF_SIGNAL,
          new Text(edge.getLabel()),
          newColumnVisibility,
          currentTimeMillis(),
          ElementMutationBuilder.EMPTY_VALUE);

      m.putDelete(
          AccumuloEdge.CF_OUT_VERTEX,
          new Text(edge.getVertexId(Direction.OUT)),
          currentColumnVisibility,
          currentTimeMillis());
      m.put(
          AccumuloEdge.CF_OUT_VERTEX,
          new Text(edge.getVertexId(Direction.OUT)),
          newColumnVisibility,
          currentTimeMillis(),
          ElementMutationBuilder.EMPTY_VALUE);

      m.putDelete(
          AccumuloEdge.CF_IN_VERTEX,
          new Text(edge.getVertexId(Direction.IN)),
          currentColumnVisibility,
          currentTimeMillis());
      m.put(
          AccumuloEdge.CF_IN_VERTEX,
          new Text(edge.getVertexId(Direction.IN)),
          newColumnVisibility,
          currentTimeMillis(),
          ElementMutationBuilder.EMPTY_VALUE);
    } else if (element instanceof AccumuloVertex) {
      m.putDelete(
          AccumuloVertex.CF_SIGNAL, EMPTY_TEXT, currentColumnVisibility, currentTimeMillis());
      m.put(
          AccumuloVertex.CF_SIGNAL,
          EMPTY_TEXT,
          newColumnVisibility,
          currentTimeMillis(),
          ElementMutationBuilder.EMPTY_VALUE);
    } else {
      throw new IllegalArgumentException("Invalid element type: " + element);
    }
    return true;
  }
Пример #5
0
  private void saveEdgeInfoOnVertex(
      AccumuloEdge edge, String edgeLabel, ColumnVisibility edgeColumnVisibility) {
    Text edgeIdText = new Text(edge.getId());

    // Update out vertex.
    Mutation addEdgeToOutMutation = new Mutation(edge.getVertexId(Direction.OUT));
    EdgeInfo edgeInfo =
        new EdgeInfo(
            getNameSubstitutionStrategy().deflate(edgeLabel), edge.getVertexId(Direction.IN));
    addEdgeToOutMutation.put(
        AccumuloVertex.CF_OUT_EDGE, edgeIdText, edgeColumnVisibility, edgeInfo.toValue());
    saveVertexMutation(addEdgeToOutMutation);

    // Update in vertex.
    Mutation addEdgeToInMutation = new Mutation(edge.getVertexId(Direction.IN));
    edgeInfo =
        new EdgeInfo(
            getNameSubstitutionStrategy().deflate(edgeLabel), edge.getVertexId(Direction.OUT));
    addEdgeToInMutation.put(
        AccumuloVertex.CF_IN_EDGE, edgeIdText, edgeColumnVisibility, edgeInfo.toValue());
    saveVertexMutation(addEdgeToInMutation);
  }
Пример #6
0
  public Iterable<KeyValuePair> getVertexTableKeyValuePairsEdge(AccumuloEdge edge) {
    List<KeyValuePair> results = new ArrayList<>();
    ColumnVisibility edgeColumnVisibility = visibilityToAccumuloVisibility(edge.getVisibility());
    String edgeLabel = edge.getNewEdgeLabel() != null ? edge.getNewEdgeLabel() : edge.getLabel();
    Text edgeIdText = new Text(edge.getId());
    long timestamp = edge.getTimestamp();

    // out vertex.
    Text vertexOutIdRowKey = new Text(edge.getVertexId(Direction.OUT));
    org.vertexium.accumulo.iterator.model.EdgeInfo edgeInfo =
        new EdgeInfo(
            getNameSubstitutionStrategy().deflate(edgeLabel), edge.getVertexId(Direction.IN));
    results.add(
        new KeyValuePair(
            new Key(
                vertexOutIdRowKey,
                AccumuloVertex.CF_OUT_EDGE,
                edgeIdText,
                edgeColumnVisibility,
                timestamp),
            edgeInfo.toValue()));

    // in vertex.
    Text vertexInIdRowKey = new Text(edge.getVertexId(Direction.IN));
    edgeInfo =
        new EdgeInfo(
            getNameSubstitutionStrategy().deflate(edgeLabel), edge.getVertexId(Direction.OUT));
    results.add(
        new KeyValuePair(
            new Key(
                vertexInIdRowKey,
                AccumuloVertex.CF_IN_EDGE,
                edgeIdText,
                edgeColumnVisibility,
                timestamp),
            edgeInfo.toValue()));

    return results;
  }
Пример #7
0
 private Mutation createMutationForEdge(AccumuloEdge edge, ColumnVisibility edgeColumnVisibility) {
   String edgeRowKey = edge.getId();
   Mutation m = new Mutation(edgeRowKey);
   String edgeLabel = edge.getLabel();
   if (edge.getNewEdgeLabel() != null) {
     edgeLabel = edge.getNewEdgeLabel();
     m.putDelete(
         AccumuloEdge.CF_SIGNAL,
         new Text(edge.getLabel()),
         edgeColumnVisibility,
         currentTimeMillis());
   }
   m.put(
       AccumuloEdge.CF_SIGNAL,
       new Text(edgeLabel),
       edgeColumnVisibility,
       edge.getTimestamp(),
       ElementMutationBuilder.EMPTY_VALUE);
   m.put(
       AccumuloEdge.CF_OUT_VERTEX,
       new Text(edge.getVertexId(Direction.OUT)),
       edgeColumnVisibility,
       edge.getTimestamp(),
       ElementMutationBuilder.EMPTY_VALUE);
   m.put(
       AccumuloEdge.CF_IN_VERTEX,
       new Text(edge.getVertexId(Direction.IN)),
       edgeColumnVisibility,
       edge.getTimestamp(),
       ElementMutationBuilder.EMPTY_VALUE);
   for (PropertyDeleteMutation propertyDeleteMutation : edge.getPropertyDeleteMutations()) {
     addPropertyDeleteToMutation(m, propertyDeleteMutation);
   }
   for (PropertySoftDeleteMutation propertySoftDeleteMutation :
       edge.getPropertySoftDeleteMutations()) {
     addPropertySoftDeleteToMutation(m, propertySoftDeleteMutation);
   }
   for (Property property : edge.getProperties()) {
     addPropertyToMutation(edge.getGraph(), m, edgeRowKey, property);
   }
   return m;
 }
Пример #8
0
  public Iterable<KeyValuePair> getEdgeTableKeyValuePairsEdge(AccumuloEdge edge) {
    List<KeyValuePair> results = new ArrayList<>();

    ColumnVisibility edgeColumnVisibility = visibilityToAccumuloVisibility(edge.getVisibility());
    Text edgeRowKey = new Text(edge.getId());
    String edgeLabel = edge.getLabel();
    if (edge.getNewEdgeLabel() != null) {
      throw new VertexiumException("Cannot get key/value pairs for label changes");
    }
    results.add(
        new KeyValuePair(
            new Key(
                edgeRowKey,
                AccumuloEdge.CF_SIGNAL,
                new Text(edgeLabel),
                edgeColumnVisibility,
                edge.getTimestamp()),
            ElementMutationBuilder.EMPTY_VALUE));
    results.add(
        new KeyValuePair(
            new Key(
                edgeRowKey,
                AccumuloEdge.CF_OUT_VERTEX,
                new Text(edge.getVertexId(Direction.OUT)),
                edgeColumnVisibility,
                edge.getTimestamp()),
            ElementMutationBuilder.EMPTY_VALUE));
    results.add(
        new KeyValuePair(
            new Key(
                edgeRowKey,
                AccumuloEdge.CF_IN_VERTEX,
                new Text(edge.getVertexId(Direction.IN)),
                edgeColumnVisibility,
                edge.getTimestamp()),
            ElementMutationBuilder.EMPTY_VALUE));
    if (edge.getPropertyDeleteMutations().iterator().hasNext()) {
      throw new VertexiumException("Cannot get key/value pairs for property deletions");
    }
    for (PropertySoftDeleteMutation propertySoftDeleteMutation :
        edge.getPropertySoftDeleteMutations()) {
      addPropertySoftDeleteToKeyValuePairs(results, edgeRowKey, propertySoftDeleteMutation);
    }
    for (Property property : edge.getProperties()) {
      addPropertyToKeyValuePairs(results, edgeRowKey, property);
    }
    return results;
  }