org.neo4j.internal.collector.TruncatedQuerySnapshot Maven / Gradle / Ivy
/*
* 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 queryPlanSupplier;
final MapValue queryParameters;
final long elapsedTimeMicros;
final long compilationTimeMicros;
final long startTimestampMillis;
final long estimatedHeap;
static final long SHALLOW_SIZE = HeapEstimator.shallowSizeOfInstance(TruncatedQuerySnapshot.class)
+ HeapEstimator.shallowSizeOfInstance(Supplier.class);
public TruncatedQuerySnapshot(
NamedDatabaseId databaseId,
String fullQueryText,
Supplier queryPlanSupplier,
MapValue queryParameters,
long elapsedTimeMicros,
long compilationTimeMicros,
long startTimestampMillis,
int maxQueryTextLength) {
this.databaseId = databaseId;
this.fullQueryTextHash = fullQueryText.hashCode();
this.queryText = truncateQueryText(fullQueryText, maxQueryTextLength);
this.queryPlanSupplier = queryPlanSupplier;
this.queryParameters = truncateParameters(queryParameters);
this.elapsedTimeMicros = elapsedTimeMicros;
this.compilationTimeMicros = compilationTimeMicros;
this.startTimestampMillis = startTimestampMillis;
this.estimatedHeap =
SHALLOW_SIZE + HeapEstimator.sizeOf(this.queryText) + this.queryParameters.estimatedHeapUsage();
}
private static String truncateQueryText(String queryText, int maxLength) {
return queryText.length() > maxLength ? queryText.substring(0, maxLength) : queryText;
}
private static MapValue truncateParameters(MapValue parameters) {
int size = parameters.size();
if (size == 0) {
return VirtualValues.EMPTY_MAP;
}
MapValueBuilder mapValueBuilder = new MapValueBuilder(size);
parameters.foreach((key, value) -> {
mapValueBuilder.add(
key.length() <= MAX_PARAMETER_KEY_LENGTH ? key : key.substring(0, MAX_PARAMETER_KEY_LENGTH),
value.map(VALUE_TRUNCATER));
});
return mapValueBuilder.build();
}
private static final ValueTruncater VALUE_TRUNCATER = new ValueTruncater();
private static final int MAX_TEXT_PARAMETER_LENGTH = 100;
private static final int MAX_PARAMETER_KEY_LENGTH = 1000;
@VisibleForTesting
public long estimatedHeap() {
return estimatedHeap;
}
@VisibleForTesting
public MapValue queryParameters() {
return queryParameters;
}
static class ValueTruncater implements ValueMapper {
@Override
public AnyValue mapPath(VirtualPathValue value) {
return Values.stringValue("§PATH[" + value.size() + "]");
}
@Override
public AnyValue mapNode(VirtualNodeValue value) {
if (value instanceof NodeValue) {
// Note: we do not want to keep a reference to the whole node value as it could contain a lot of data.
return VirtualValues.node(value.id());
}
return value;
}
@Override
public AnyValue mapRelationship(VirtualRelationshipValue value) {
if (value instanceof RelationshipValue) {
// Note: we do not want to keep a reference to the whole relationship value as it could contain a lot of
// data.
return VirtualValues.relationship(value.id());
}
return value;
}
@Override
public AnyValue mapMap(MapValue map) {
return Values.stringValue("§MAP[" + map.size() + "]");
}
@Override
public AnyValue mapNoValue() {
return Values.NO_VALUE;
}
@Override
public AnyValue mapSequence(SequenceValue value) {
return Values.stringValue("§LIST[" + value.intSize() + "]");
}
@Override
public AnyValue mapText(TextValue value) {
if (value.length() > MAX_TEXT_PARAMETER_LENGTH) {
return Values.stringValue(value.stringValue().substring(0, MAX_TEXT_PARAMETER_LENGTH));
}
return value;
}
@Override
public AnyValue mapBoolean(BooleanValue value) {
return value;
}
@Override
public AnyValue mapNumber(NumberValue value) {
return value;
}
@Override
public AnyValue mapDateTime(DateTimeValue value) {
return value;
}
@Override
public AnyValue mapLocalDateTime(LocalDateTimeValue value) {
return value;
}
@Override
public AnyValue mapDate(DateValue value) {
return value;
}
@Override
public AnyValue mapTime(TimeValue value) {
return value;
}
@Override
public AnyValue mapLocalTime(LocalTimeValue value) {
return value;
}
@Override
public AnyValue mapDuration(DurationValue value) {
return value;
}
@Override
public AnyValue mapPoint(PointValue value) {
return value;
}
}
}