public int augmentDegree(Direction direction, int degree, int typeId) {
if (hasAddedRelationships()) {
degree = relationshipsAdded.augmentDegree(direction, degree, typeId);
}
if (hasRemovedRelationships()) {
degree = relationshipsRemoved.augmentDegree(direction, degree, typeId);
}
return degree;
}
public void removeRelationship(long relId, int typeId, Direction direction) {
if (hasAddedRelationships()) {
if (relationshipsAdded.removeRelationship(relId, typeId, direction)) {
// This was a rel that was added in this tx, no need to add it to the remove list, instead
// we just
// remove it from added relationships.
return;
}
}
if (!hasRemovedRelationships()) {
relationshipsRemoved = new RelationshipChangesForNode(DiffStrategy.REMOVE);
}
relationshipsRemoved.addRelationship(relId, typeId, direction);
}
public PrimitiveLongIterator augmentRelationships(
Direction direction, PrimitiveLongIterator rels) {
if (hasAddedRelationships()) {
return relationshipsAdded.augmentRelationships(direction, rels);
}
return rels;
}
@Override
public void clear() {
super.clear();
if (relationshipsAdded != null) {
relationshipsAdded.clear();
}
if (relationshipsRemoved != null) {
relationshipsRemoved.clear();
}
if (labelDiffSets != null) {
labelDiffSets.clear();
}
if (indexDiffs != null) {
indexDiffs.clear();
}
}
public void addRelationship(long relId, int typeId, Direction direction) {
if (!hasAddedRelationships()) {
relationshipsAdded = new RelationshipChangesForNode(DiffStrategy.ADD);
}
relationshipsAdded.addRelationship(relId, typeId, direction);
}
public PrimitiveIntIterator relationshipTypes() {
if (hasAddedRelationships()) {
return relationshipsAdded.relationshipTypes();
}
return PrimitiveIntCollections.emptyIterator();
}
public PrimitiveLongIterator addedRelationships(Direction direction, int[] types) {
if (hasAddedRelationships()) {
return relationshipsAdded.augmentRelationships(direction, types, emptyIterator());
}
return null;
}
