org.neo4j.internal.collector.GraphCountsSection Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of neo4j-data-collector Show documentation
Show all versions of neo4j-data-collector Show documentation
The Neo4j data collector daemon.
/*
* Copyright (c) "Neo4j"
* Neo4j Sweden AB [https://neo4j.com]
*
* This file is part of Neo4j.
*
* Neo4j is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see retrieve(Kernel kernel, Anonymizer anonymizer)
throws TransactionFailureException, IndexNotFoundKernelException {
try (KernelTransaction tx =
kernel.beginTransaction(KernelTransaction.Type.EXPLICIT, LoginContext.AUTH_DISABLED)) {
TokenRead tokens = tx.tokenRead();
Read read = tx.dataRead();
Map data = new HashMap<>();
data.put("nodes", nodeCounts(tokens, read, anonymizer));
data.put("relationships", relationshipCounts(tokens, read, anonymizer));
data.put("indexes", indexes(tokens, tx.schemaRead(), anonymizer));
data.put("constraints", constraints(tokens, tx.schemaRead(), anonymizer));
return Stream.of(new RetrieveResult(Sections.GRAPH_COUNTS, data));
}
}
private static List> nodeCounts(TokenRead tokens, Read read, Anonymizer anonymizer) {
List> nodeCounts = new ArrayList<>();
Map nodeCount = new HashMap<>();
nodeCount.put("count", read.estimateCountsForNode(TokenRead.ANY_LABEL));
nodeCounts.add(nodeCount);
tokens.labelsGetAllTokens().forEachRemaining(t -> {
long count = read.estimateCountsForNode(t.id());
Map labelCount = new HashMap<>();
labelCount.put("label", anonymizer.label(t.name(), t.id()));
labelCount.put("count", count);
nodeCounts.add(labelCount);
});
return nodeCounts;
}
private static List> relationshipCounts(TokenRead tokens, Read read, Anonymizer anonymizer) {
List> relationshipCounts = new ArrayList<>();
Map relationshipCount = new HashMap<>();
relationshipCount.put(
"count",
read.estimateCountsForRelationships(
TokenRead.ANY_LABEL, TokenRead.ANY_RELATIONSHIP_TYPE, TokenRead.ANY_LABEL));
relationshipCounts.add(relationshipCount);
List labels = Iterators.asList(tokens.labelsGetAllTokens());
tokens.relationshipTypesGetAllTokens().forEachRemaining(t -> {
long count = read.estimateCountsForRelationships(TokenRead.ANY_LABEL, t.id(), TokenRead.ANY_LABEL);
Map relationshipTypeCount = new HashMap<>();
relationshipTypeCount.put("relationshipType", anonymizer.relationshipType(t.name(), t.id()));
relationshipTypeCount.put("count", count);
relationshipCounts.add(relationshipTypeCount);
for (NamedToken label : labels) {
long startCount = read.estimateCountsForRelationships(label.id(), t.id(), TokenRead.ANY_LABEL);
if (startCount > 0) {
Map x = new HashMap<>();
x.put("relationshipType", anonymizer.relationshipType(t.name(), t.id()));
x.put("startLabel", anonymizer.label(label.name(), label.id()));
x.put("count", startCount);
relationshipCounts.add(x);
}
long endCount = read.estimateCountsForRelationships(TokenRead.ANY_LABEL, t.id(), label.id());
if (endCount > 0) {
Map x = new HashMap<>();
x.put("relationshipType", anonymizer.relationshipType(t.name(), t.id()));
x.put("endLabel", anonymizer.label(label.name(), label.id()));
x.put("count", endCount);
relationshipCounts.add(x);
}
}
});
return relationshipCounts;
}
private static List> indexes(TokenRead tokens, SchemaRead schemaRead, Anonymizer anonymizer)
throws IndexNotFoundKernelException {
List> indexes = new ArrayList<>();
Iterator iterator = schemaRead.indexesGetAll();
while (iterator.hasNext()) {
IndexDescriptor index = iterator.next();
IndexType indexType = index.getIndexType();
if (indexType == IndexType.FULLTEXT) {
/* For full text indexes, we currently do not return its options, which makes returning information on
* this index not useful and if the index type is ignored, this would even be misleading.
*/
continue;
}
EntityType entityType = index.schema().entityType();
Map data = new HashMap<>();
switch (entityType) {
case NODE:
data.put(
"labels",
map(
index.schema().getEntityTokenIds(),
id -> anonymizer.label(tokens.labelGetName(id), id)));
break;
case RELATIONSHIP:
data.put(
"relationshipTypes",
map(
index.schema().getEntityTokenIds(),
id -> anonymizer.relationshipType(tokens.relationshipTypeGetName(id), id)));
break;
default:
}
data.put(
"properties",
map(
index.schema().getPropertyIds(),
id -> anonymizer.propertyKey(tokens.propertyKeyGetName(id), id)));
var indexSample = schemaRead.indexSample(index);
data.put("totalSize", indexSample.indexSize());
data.put("updatesSinceEstimation", indexSample.updates());
data.put("estimatedUniqueSize", indexSample.uniqueValues());
data.put("indexType", indexType.name());
data.put("indexProvider", index.getIndexProvider().name());
indexes.add(data);
}
return indexes;
}
private static List> constraints(
TokenRead tokens, SchemaRead schemaRead, Anonymizer anonymizer) {
List> constraints = new ArrayList<>();
Iterator iterator = schemaRead.constraintsGetAll();
while (iterator.hasNext()) {
ConstraintDescriptor constraint = iterator.next();
Map data = ConstraintSubSection.constraint(tokens, anonymizer, constraint);
constraints.add(data);
}
return constraints;
}
private static List map(int[] ids, IntFunction f) {
return Arrays.stream(ids).mapToObj(f).collect(Collectors.toList());
}
}