private void connectRelationship(
NodeRecord firstNode, NodeRecord secondNode, RelationshipRecord rel) {
assert firstNode.getNextRel() != rel.getId();
assert secondNode.getNextRel() != rel.getId();
rel.setFirstNextRel(firstNode.getNextRel());
rel.setSecondNextRel(secondNode.getNextRel());
connect(firstNode, rel);
connect(secondNode, rel);
firstNode.setNextRel(rel.getId());
secondNode.setNextRel(rel.getId());
}
public Iterable<SimpleRelationship> getSimpleRelationships(long nodeId) {
NodeRecord nodeRecord = getNodeRecord(nodeId);
long nextRel = nodeRecord.getNextRel();
List<SimpleRelationship> rels = new ArrayList<SimpleRelationship>();
while (nextRel != Record.NO_NEXT_RELATIONSHIP.intValue()) {
RelationshipRecord relRecord = getRelationshipRecord(nextRel);
RelationshipType type = new RelationshipTypeImpl(typeHolder.getName(relRecord.getType()));
rels.add(
new SimpleRelationship(
relRecord.getId(), relRecord.getFirstNode(), relRecord.getSecondNode(), type));
long firstNode = relRecord.getFirstNode();
long secondNode = relRecord.getSecondNode();
if (firstNode == nodeId) {
nextRel = relRecord.getFirstNextRel();
} else if (secondNode == nodeId) {
nextRel = relRecord.getSecondNextRel();
} else {
throw new InvalidRecordException(
"Node["
+ nodeId
+ "] not part of firstNode["
+ firstNode
+ "] or secondNode["
+ secondNode
+ "]");
}
}
return rels;
}
@Override
public Iterable<Long> getRelationshipIds(long nodeId) {
NodeRecord nodeRecord = getNodeRecord(nodeId);
long nextRel = nodeRecord.getNextRel();
List<Long> ids = new ArrayList<Long>();
while (nextRel != Record.NO_NEXT_RELATIONSHIP.intValue()) {
RelationshipRecord relRecord = getRelationshipRecord(nextRel);
ids.add(relRecord.getId());
long firstNode = relRecord.getFirstNode();
long secondNode = relRecord.getSecondNode();
if (firstNode == nodeId) {
nextRel = relRecord.getFirstNextRel();
} else if (secondNode == nodeId) {
nextRel = relRecord.getSecondNextRel();
} else {
throw new InvalidRecordException(
"Node["
+ nodeId
+ "] not part of firstNode["
+ firstNode
+ "] or secondNode["
+ secondNode
+ "]");
}
}
return ids;
}
private void connect(NodeRecord node, RelationshipRecord rel) {
if (node.getNextRel() != Record.NO_NEXT_RELATIONSHIP.intValue()) {
RelationshipRecord nextRel = getRelationshipStore().getRecord(node.getNextRel());
boolean changed = false;
if (nextRel.getFirstNode() == node.getId()) {
nextRel.setFirstPrevRel(rel.getId());
changed = true;
}
if (nextRel.getSecondNode() == node.getId()) {
nextRel.setSecondPrevRel(rel.getId());
changed = true;
}
if (!changed) {
throw new InvalidRecordException(node + " dont match " + nextRel);
}
getRelationshipStore().updateRecord(nextRel);
}
}
private void connectRelationship(
NodeRecord firstNode,
NodeRecord secondNode,
RelationshipRecord rel,
RecordAccess<Long, RelationshipRecord, Void> relRecords,
RecordAccess<Long, RelationshipGroupRecord, Integer> relGroupRecords) {
// Assertion interpreted: if node is a normal node and we're trying to create a
// relationship that we already have as first rel for that node --> error
assert firstNode.getNextRel() != rel.getId() || firstNode.isDense();
assert secondNode.getNextRel() != rel.getId() || secondNode.isDense();
if (!firstNode.isDense()) {
rel.setFirstNextRel(firstNode.getNextRel());
}
if (!secondNode.isDense()) {
rel.setSecondNextRel(secondNode.getNextRel());
}
if (!firstNode.isDense()) {
connect(firstNode, rel, relRecords);
} else {
connectRelationshipToDenseNode(firstNode, rel, relRecords, relGroupRecords);
}
if (!secondNode.isDense()) {
if (firstNode.getId() != secondNode.getId()) {
connect(secondNode, rel, relRecords);
} else {
rel.setFirstInFirstChain(true);
rel.setSecondPrevRel(rel.getFirstPrevRel());
}
} else if (firstNode.getId() != secondNode.getId()) {
connectRelationshipToDenseNode(secondNode, rel, relRecords, relGroupRecords);
}
if (!firstNode.isDense()) {
firstNode.setNextRel(rel.getId());
}
if (!secondNode.isDense()) {
secondNode.setNextRel(rel.getId());
}
}
private void convertNodeToDenseIfNecessary(
NodeRecord node,
RecordAccess<Long, RelationshipRecord, Void> relRecords,
RecordAccess<Long, RelationshipGroupRecord, Integer> relGroupRecords) {
if (node.isDense()) {
return;
}
long relId = node.getNextRel();
if (relId != Record.NO_NEXT_RELATIONSHIP.intValue()) {
RecordProxy<Long, RelationshipRecord, Void> relChange = relRecords.getOrLoad(relId, null);
RelationshipRecord rel = relChange.forReadingLinkage();
if (RelationshipCounter.relCount(node.getId(), rel) >= neoStore.getDenseNodeThreshold()) {
convertNodeToDenseNode(node, relChange.forChangingLinkage(), relRecords, relGroupRecords);
}
}
}
public RecordProxy<Long, RelationshipGroupRecord, Integer> getRelationshipGroup(
NodeRecord node, int type) {
long groupId = node.getNextRel();
long previousGroupId = Record.NO_NEXT_RELATIONSHIP.intValue();
Set<Integer> allTypes = new HashSet<>();
while (groupId != Record.NO_NEXT_RELATIONSHIP.intValue()) {
RecordProxy<Long, RelationshipGroupRecord, Integer> change =
recordChangeSet.getRelGroupRecords().getOrLoad(groupId, type);
RelationshipGroupRecord record = change.forReadingData();
record.setPrev(previousGroupId); // not persistent so not a "change"
allTypes.add(record.getType());
if (record.getType() == type) {
return change;
}
previousGroupId = groupId;
groupId = record.getNext();
}
return null;
}
private void connect(
NodeRecord node,
RelationshipRecord rel,
RecordAccess<Long, RelationshipRecord, Void> relRecords) {
connect(node.getId(), node.getNextRel(), rel, relRecords);
}
public void updateFirstRelationships() {
for (RecordProxy<Long, NodeRecord, Void> change : recordChangeSet.getNodeRecords().changes()) {
NodeRecord record = change.forReadingLinkage();
txState.setFirstIds(record.getId(), record.getNextRel(), record.getNextProp());
}
}