in ytdb/src/main/java/com/youtrackdb/ldbc/ytdb/loader/YtdbLoader.java [68:110]
public Map<String, Long> counts() {
var counts = new HashMap<String, Long>();
try {
List<String> vertexLabels = List.of(
LdbcSchema.PERSON,
LdbcSchema.PLACE,
LdbcSchema.ORGANISATION,
LdbcSchema.TAG_CLASS,
LdbcSchema.TAG,
LdbcSchema.FORUM,
LdbcSchema.POST,
LdbcSchema.COMMENT
);
for (String label : vertexLabels) {
counts.put(label, traversal.V().hasLabel(label).count().next());
}
List<String> edgeLabels = List.of(
LdbcSchema.KNOWS,
LdbcSchema.HAS_CREATOR,
LdbcSchema.IS_LOCATED_IN,
LdbcSchema.HAS_INTEREST,
LdbcSchema.HAS_MEMBER,
LdbcSchema.LIKES,
LdbcSchema.HAS_TAG,
LdbcSchema.REPLY_OF,
LdbcSchema.STUDY_AT,
LdbcSchema.WORK_AT,
LdbcSchema.HAS_MODERATOR,
LdbcSchema.CONTAINER_OF,
LdbcSchema.IS_PART_OF,
LdbcSchema.HAS_TYPE
);
for (String edgeLabel : edgeLabels) {
counts.put(edgeLabel, traversal.E().hasLabel(edgeLabel).count().next());
}
} catch (Exception e) {
log.error("Error computing counts", e);
}
return counts;
}