admin/hermione
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: Enrich relationships with node types in Neo4JUitilityService

Browse Source
This commit is contained in:
AndreaTerzani
2024-08-12 00:12:18 +02:00
parent d96624cfa8
commit 2a6336cfe1
1 changed files with 32 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
src/main/java/com/olympus/hermione/services/Neo4JUitilityService.java
View File

@@ -85,10 +85,42 @@ public class Neo4JUitilityService {
relationshipObject.add("properties", properties); relationshipObject.add("properties", properties);
} }
enrichRelationshipsWithNodeTypes(session, relationshipsMap);
JsonArray relationshipsArray = new JsonArray(); JsonArray relationshipsArray = new JsonArray();
relationshipsMap.values().forEach(relationshipsArray::add); relationshipsMap.values().forEach(relationshipsArray::add);
return relationshipsArray; return relationshipsArray;
} }
private void enrichRelationshipsWithNodeTypes(Session session, Map<String, JsonObject> relationshipsMap) {
String query = "MATCH (n1)-[r]->(n2) RETURN DISTINCT type(r) AS relType, labels(n1) AS startLabels, labels(n2) AS endLabels";
List<Record> records = session.run(query).list();
for (Record record : records) {
String relType = record.get("relType").asString();
List<String> startLabels = record.get("startLabels").asList(Value::asString);
List<String> endLabels = record.get("endLabels").asList(Value::asString);
JsonObject relationshipObject = relationshipsMap.get(":`"+relType+"`");
if (relationshipObject != null) {
JsonArray startNodeTypes = relationshipObject.has("startNodeTypes") ? relationshipObject.get("startNodeTypes").getAsJsonArray() : new JsonArray();
startLabels.forEach(label -> {
if (!startNodeTypes.contains(new JsonPrimitive(label))) {
startNodeTypes.add(new JsonPrimitive(label));
}
});
relationshipObject.add("startNodeTypes", startNodeTypes);
JsonArray endNodeTypes = relationshipObject.has("endNodeTypes") ? relationshipObject.get("endNodeTypes").getAsJsonArray() : new JsonArray();
endLabels.forEach(label -> {
if (!endNodeTypes.contains(new JsonPrimitive(label))) {
endNodeTypes.add(new JsonPrimitive(label));
}
});
relationshipObject.add("endNodeTypes", endNodeTypes);
}
}
}
} }