org.neo4j.kernel.api.query.ExecutingQuery Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of neo4j-kernel Show documentation
Show all versions of neo4j-kernel Show documentation
Neo4j kernel is a lightweight, embedded Java database designed to
store data structured as graphs rather than tables. For more
information, see http://neo4j.org.
/*
* 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 WAIT_TIME =
newUpdater(ExecutingQuery.class, "waitTimeNanos");
private final long queryId;
private final LockTracer lockTracer;
private final String executingUsername;
private final String authenticatedUsername;
private final ClientConnectionInfo clientConnection;
private final String rawQueryText;
private final MapValue rawQueryParameters;
private final long startTimeNanos;
private final long startTimestampMillis;
private final Map transactionAnnotationData;
private final long threadExecutingTheQueryId;
@SuppressWarnings({"unused", "FieldCanBeLocal"})
private final String threadExecutingTheQueryName;
private final SystemNanoClock clock;
private final CpuClock cpuClock;
private final long cpuTimeNanosWhenQueryStarted;
/** Uses write barrier of {@link #status}. */
private CompilerInfo compilerInfo;
private long compilationCompletedNanos;
private String obfuscatedQueryText;
private MapValue obfuscatedQueryParameters;
private Supplier planDescriptionSupplier;
private volatile ExecutingQueryStatus status = SimpleState.parsing();
private volatile ExecutingQuery previousQuery;
/** Updated through {@link #WAIT_TIME} */
@SuppressWarnings("unused")
private volatile long waitTimeNanos;
private HeapHighWaterMarkTracker memoryTracker;
// Accumulated statistics of transactions that have executed this query but are already committed
private volatile long pageHitsOfClosedTransactionsExcludingCommits;
private volatile long pageFaultsOfClosedTransactionsExcludingCommits;
private volatile long pageHitsOfClosedTransactionsIncludingCommits;
private volatile long pageFaultsOfClosedTransactionsIncludingCommits;
private volatile long pageHitsOfClosedTransactionCommits;
private volatile long pageFaultsOfClosedTransactionCommits;
/**
* Map of all transactions that are active executing this query.
* Needs to be thread-safe because other Threads traverse the list when calling {@link #snapshot()}.
*/
private final ConcurrentMap openTransactionBindings = new ConcurrentHashMap<>();
/**
* Database id of the outer (first) transaction binding.
*/
private NamedDatabaseId namedDatabaseId;
/**
* Transaction id of the outer (first) transaction.
*/
private long outerTransactionId = -1L;
/**
* The database name of the parent transactions (for composite databases)
*/
private String parentDbName;
/**
* The transaction id of the parent transactions (for composite databases)
*/
private long parentTransactionId = -1L;
private QueryCacheUsage executableQueryCacheUsage;
private QueryCacheUsage logicalPlanCacheUsage;
public ExecutingQuery(
long queryId,
ClientConnectionInfo clientConnection,
String executingUsername,
String authenticatedUsername,
String queryText,
MapValue queryParameters,
Map transactionAnnotationData,
long threadExecutingTheQueryId,
String threadExecutingTheQueryName,
LockTracer systemLockTracer,
SystemNanoClock clock,
CpuClock cpuClock) {
// Capture timestamps first
this.cpuTimeNanosWhenQueryStarted = cpuClock.cpuTimeNanos(threadExecutingTheQueryId);
this.startTimeNanos = clock.nanos();
this.startTimestampMillis = clock.millis();
// then continue with assigning fields
this.queryId = queryId;
this.clientConnection = clientConnection;
this.executingUsername = executingUsername;
this.authenticatedUsername = authenticatedUsername;
this.rawQueryText = queryText;
this.rawQueryParameters = queryParameters;
this.transactionAnnotationData = transactionAnnotationData;
this.threadExecutingTheQueryId = threadExecutingTheQueryId;
this.threadExecutingTheQueryName = threadExecutingTheQueryName;
this.clock = clock;
this.cpuClock = cpuClock;
this.lockTracer = systemLockTracer.combine(this::waitForLock);
this.memoryTracker = HeapHighWaterMarkTracker.ZERO;
}
// NOTE: test/benchmarking constructor
public ExecutingQuery(
long queryId,
ClientConnectionInfo clientConnection,
NamedDatabaseId namedDatabaseId,
String executingUsername,
String authenticatedUsername,
String queryText,
MapValue queryParameters,
Map transactionAnnotationData,
LongSupplier activeLockCount,
LongSupplier hitsSupplier,
LongSupplier faultsSupplier,
long threadExecutingTheQueryId,
String threadExecutingTheQueryName,
LockTracer systemLockTracer,
SystemNanoClock clock,
CpuClock cpuClock) {
this(
queryId,
clientConnection,
executingUsername,
authenticatedUsername,
queryText,
queryParameters,
transactionAnnotationData,
threadExecutingTheQueryId,
threadExecutingTheQueryName,
systemLockTracer,
clock,
cpuClock);
onTransactionBound(new TransactionBinding(namedDatabaseId, hitsSupplier, faultsSupplier, activeLockCount, 1));
}
public void setParentTransaction(String parentDbName, long parentTransactionId) {
this.parentTransactionId = parentTransactionId;
this.parentDbName = parentDbName;
}
public long getOuterTransactionId() {
return outerTransactionId;
}
public static class TransactionBinding {
private final NamedDatabaseId namedDatabaseId;
private final LongSupplier hitsSupplier;
private final LongSupplier faultsSupplier;
private final LongSupplier activeLockCount;
private final long initialActiveLocks;
private final long transactionId;
public TransactionBinding(
NamedDatabaseId namedDatabaseId,
LongSupplier hitsSupplier,
LongSupplier faultsSupplier,
LongSupplier activeLockCount,
long transactionId) {
this.namedDatabaseId = namedDatabaseId;
this.hitsSupplier = hitsSupplier;
this.faultsSupplier = faultsSupplier;
this.activeLockCount = activeLockCount;
this.initialActiveLocks = activeLockCount.getAsLong();
this.transactionId = transactionId;
}
/**
* @return the number of active locks, already subtracting the initially active locks count
*/
public long getActiveLocks() {
return activeLockCount.getAsLong() - initialActiveLocks;
}
}
// update state
/**
* Called before this query (or part of this query) starts executing in a transaction.
* Adds a TransactionBinding used to fetch statistics from the transaction.
*/
public void onTransactionBound(TransactionBinding transactionBinding) {
if (this.openTransactionBindings.isEmpty()) {
namedDatabaseId = transactionBinding.namedDatabaseId;
outerTransactionId = transactionBinding.transactionId;
}
this.openTransactionBindings.put(transactionBinding.transactionId, transactionBinding);
}
/**
* Called when a transaction, that this query (or part of this query) has executed in, is closed.
* Removes the TransactionBinding for that transaction.
*/
public void onTransactionUnbound(long userTransactionId) {
var binding = openTransactionBindings.remove(userTransactionId);
if (binding == null) {
throw new IllegalStateException("Unbound a transaction that was never bound. ID: " + userTransactionId);
}
}
/**
* Called when a transaction, that this query (or part of this query) has executed in, is about to close.
* Captures some statistics that we might need even after the transaction has closed.
* NOTE: this only captures statistics that happened before commit, not during the commit, since this is called before
*/
public void onPrepareTransactionOnbound(long userTransactionId) {
var binding = openTransactionBindings.get(userTransactionId);
if (binding == null) {
throw new IllegalStateException("Unbound a transaction that was never bound. ID: " + userTransactionId);
}
recordStatisticsOfTransactionAboutToClose(binding.hitsSupplier.getAsLong(), binding.faultsSupplier.getAsLong());
}
@VisibleForTesting
void recordStatisticsOfTransactionAboutToClose(long hits, long faults) {
// We only have one thread writing to these fields
//noinspection NonAtomicOperationOnVolatileField
pageHitsOfClosedTransactionsExcludingCommits += hits;
//noinspection NonAtomicOperationOnVolatileField
pageFaultsOfClosedTransactionsExcludingCommits += faults;
}
/**
* A transaction executing part of this query is closing; record its page cache statistics (including commit).
*/
public void recordStatisticsOfClosedTransaction(ExecutionStatistics statistics) {
// We only have one thread writing to these fields
//noinspection NonAtomicOperationOnVolatileField
pageHitsOfClosedTransactionsIncludingCommits += statistics.pageHits();
//noinspection NonAtomicOperationOnVolatileField
pageFaultsOfClosedTransactionsIncludingCommits += statistics.pageFaults();
pageHitsOfClosedTransactionCommits =
(pageHitsOfClosedTransactionsIncludingCommits - pageHitsOfClosedTransactionsExcludingCommits);
pageFaultsOfClosedTransactionCommits =
(pageFaultsOfClosedTransactionsIncludingCommits - pageFaultsOfClosedTransactionsExcludingCommits);
}
public void onObfuscatorReady(QueryObfuscator queryObfuscator) {
if (status != SimpleState.parsing()) // might get called multiple times due to caching and/or internal queries
{
return;
}
try {
obfuscatedQueryText = queryObfuscator.obfuscateText(rawQueryText);
obfuscatedQueryParameters = queryObfuscator.obfuscateParameters(rawQueryParameters);
} catch (Exception ignore) {
obfuscatedQueryText = null;
obfuscatedQueryParameters = null;
}
this.status = SimpleState.planning();
}
public void onCompilationCompleted(
CompilerInfo compilerInfo, Supplier planDescriptionSupplier) {
assertExpectedStatus(SimpleState.planning());
this.compilerInfo = compilerInfo;
this.compilationCompletedNanos = clock.nanos();
this.planDescriptionSupplier = planDescriptionSupplier;
this.status = SimpleState.planned(); // write barrier - must be last
}
public void onExecutionStarted(HeapHighWaterMarkTracker memoryTracker) {
assertExpectedStatus(SimpleState.planned());
this.memoryTracker = memoryTracker;
this.status = SimpleState.running(); // write barrier - must be last
}
public void onRetryAttempted() {
assertExpectedStatus(SimpleState.running());
this.compilerInfo = null;
this.compilationCompletedNanos = 0;
this.planDescriptionSupplier = null;
this.memoryTracker = HeapHighWaterMarkTracker.NONE;
this.obfuscatedQueryParameters = null;
this.obfuscatedQueryText = null;
this.status = SimpleState.parsing();
}
/**
* ONLY FOR TESTING
*
* sets compiler info without asserting we maintain the integrity of the status field
*/
@VisibleForTesting
public void setCompilerInfoForTesting(CompilerInfo compilerInfo) {
this.compilerInfo = compilerInfo;
}
public LockTracer lockTracer() {
return lockTracer;
}
// snapshot state
public QuerySnapshot snapshot() {
// capture a consistent snapshot of the "live" state
ExecutingQueryStatus status;
long waitTimeNanos;
long currentTimeNanos;
long cpuTimeNanos;
String queryText;
MapValue queryParameters;
do {
status = this.status; // read barrier, must be first
waitTimeNanos = this.waitTimeNanos; // the reason for the retry loop: don't count the wait time twice
cpuTimeNanos = cpuClock.cpuTimeNanos(threadExecutingTheQueryId);
currentTimeNanos = clock.nanos(); // capture the time as close to the snapshot as possible
queryText = this.obfuscatedQueryText;
queryParameters = this.obfuscatedQueryParameters;
} while (this.status != status);
// guarded by barrier - unused if status is planning, stable otherwise
long compilationCompletedNanos = this.compilationCompletedNanos;
// guarded by barrier - like compilationCompletedNanos
CompilerInfo planner = status.isParsingOrPlanning() ? null : this.compilerInfo;
List waitingOnLocks = status.isWaitingOnLocks() ? status.waitingOnLocks() : Collections.emptyList();
// activeLockCount is not atomic to capture, so we capture it after the most sensitive part.
long activeLocks = 0;
long hits = pageHitsOfClosedTransactions();
long faults = pageFaultsOfClosedTransactions();
for (TransactionBinding tx : openTransactionBindings.values()) {
activeLocks += tx.getActiveLocks();
hits += tx.hitsSupplier.getAsLong();
faults += tx.faultsSupplier.getAsLong();
}
// - at this point we are done capturing the "live" state, and can start computing the snapshot -
long compilationTimeNanos =
(status.isParsingOrPlanning() ? currentTimeNanos : compilationCompletedNanos) - startTimeNanos;
long elapsedTimeNanos = currentTimeNanos - startTimeNanos;
cpuTimeNanos -= cpuTimeNanosWhenQueryStarted;
waitTimeNanos += status.waitTimeNanos(currentTimeNanos);
return new QuerySnapshot(
this,
planner,
hits,
faults,
NANOSECONDS.toMicros(compilationTimeNanos),
NANOSECONDS.toMicros(elapsedTimeNanos),
cpuTimeNanos == 0 && cpuTimeNanosWhenQueryStarted == -1 ? -1 : NANOSECONDS.toMicros(cpuTimeNanos),
NANOSECONDS.toMicros(waitTimeNanos),
status.name(),
status.toMap(currentTimeNanos),
waitingOnLocks,
activeLocks,
memoryTracker.heapHighWaterMark(),
Optional.ofNullable(queryText),
Optional.ofNullable(queryParameters),
outerTransactionId,
parentDbName,
parentTransactionId,
executableQueryCacheUsage,
logicalPlanCacheUsage);
}
public String cypherRuntime() {
return this.compilerInfo == null ? "" : this.compilerInfo.runtime();
}
// basic methods
@Override
public boolean equals(Object o) {
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
ExecutingQuery that = (ExecutingQuery) o;
return queryId == that.queryId;
}
@Override
public int hashCode() {
return (int) (queryId ^ (queryId >>> 32));
}
@Override
public String toString() {
return ToStringBuilder.reflectionToString(this);
}
// access stable state
public long internalQueryId() {
return queryId;
}
public String id() {
return Long.toString(internalQueryId());
}
public String executingUsername() {
return executingUsername;
}
public String authenticatedUsername() {
return authenticatedUsername;
}
public String rawQueryText() {
return rawQueryText;
}
public MapValue rawQueryParameters() {
return rawQueryParameters;
}
Supplier planDescriptionSupplier() {
return planDescriptionSupplier;
}
public Optional databaseId() {
return Optional.ofNullable(namedDatabaseId);
}
public long startTimestampMillis() {
return startTimestampMillis;
}
public long elapsedNanos() {
return clock.nanos() - startTimeNanos;
}
public long elapsedMillis() {
return NANOSECONDS.toMillis(elapsedNanos());
}
public Map transactionAnnotationData() {
return transactionAnnotationData;
}
public long reportedWaitingTimeNanos() {
return waitTimeNanos;
}
public long totalWaitingTimeNanos(long currentTimeNanos) {
return waitTimeNanos + status.waitTimeNanos(currentTimeNanos);
}
public String threadExecutingTheQueryName() {
return this.threadExecutingTheQueryName;
}
ClientConnectionInfo clientConnection() {
return clientConnection;
}
private LockWaitEvent waitForLock(
LockType lockType, ResourceType resourceType, long userTransactionId, long[] resourceIds) {
WaitingOnLockEvent event = new WaitingOnLockEvent(
lockType, resourceType, userTransactionId, resourceIds, this, clock.nanos(), status);
status = event;
return event;
}
void doneWaitingOnLock(WaitingOnLockEvent waiting) {
if (status != waiting) {
return; // already closed
}
WAIT_TIME.addAndGet(this, waiting.waitTimeNanos(clock.nanos()));
status = waiting.previousStatus();
}
private void assertExpectedStatus(ExecutingQueryStatus expectedStatus) {
if (status != expectedStatus) {
throw new IllegalStateException(String.format(
"Expected query in '%s' state, actual state is '%s'.", expectedStatus.name(), status.name()));
}
}
public ExecutingQuery getPreviousQuery() {
return previousQuery;
}
public void setPreviousQuery(ExecutingQuery previousQuery) {
this.previousQuery = previousQuery;
}
public long pageHitsOfClosedTransactions() {
return pageHitsOfClosedTransactionsExcludingCommits + pageHitsOfClosedTransactionCommits;
}
public long pageFaultsOfClosedTransactions() {
return pageFaultsOfClosedTransactionsExcludingCommits + pageFaultsOfClosedTransactionCommits;
}
public long pageHitsOfClosedTransactionCommits() {
return pageHitsOfClosedTransactionCommits;
}
public long pageFaultsOfClosedTransactionCommits() {
return pageFaultsOfClosedTransactionCommits;
}
public void executableQueryCacheHit() {
this.executableQueryCacheUsage = QueryCacheUsage.HIT;
}
public void executableQueryCacheMiss() {
this.executableQueryCacheUsage = QueryCacheUsage.MISS;
}
public void logicalPlanCacheHit() {
this.logicalPlanCacheUsage = QueryCacheUsage.HIT;
}
public void logicalPlanCacheMiss() {
this.logicalPlanCacheUsage = QueryCacheUsage.MISS;
}
}