org.neo4j.kernel.impl.api.parallel.ExecutionContextProcedureTransaction 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 .
*/
package org.neo4j.kernel.impl.api.parallel;
import static org.neo4j.kernel.impl.api.parallel.ExecutionContextProcedureKernelTransaction.failure;
import java.util.Map;
import java.util.Optional;
import java.util.UUID;
import org.neo4j.exceptions.KernelException;
import org.neo4j.graphdb.Entity;
import org.neo4j.graphdb.Label;
import org.neo4j.graphdb.Lock;
import org.neo4j.graphdb.Node;
import org.neo4j.graphdb.QueryExecutionException;
import org.neo4j.graphdb.Relationship;
import org.neo4j.graphdb.RelationshipType;
import org.neo4j.graphdb.ResourceIterable;
import org.neo4j.graphdb.ResourceIterator;
import org.neo4j.graphdb.Result;
import org.neo4j.graphdb.TransactionTerminatedException;
import org.neo4j.graphdb.schema.Schema;
import org.neo4j.graphdb.traversal.BidirectionalTraversalDescription;
import org.neo4j.graphdb.traversal.TraversalDescription;
import org.neo4j.internal.helpers.collection.AbstractResourceIterable;
import org.neo4j.internal.kernel.api.CursorFactory;
import org.neo4j.internal.kernel.api.NodeCursor;
import org.neo4j.internal.kernel.api.QueryContext;
import org.neo4j.internal.kernel.api.Read;
import org.neo4j.internal.kernel.api.RelationshipDataAccessor;
import org.neo4j.internal.kernel.api.RelationshipScanCursor;
import org.neo4j.internal.kernel.api.SchemaRead;
import org.neo4j.internal.kernel.api.TokenRead;
import org.neo4j.internal.kernel.api.connectioninfo.ClientConnectionInfo;
import org.neo4j.internal.kernel.api.security.SecurityContext;
import org.neo4j.io.pagecache.context.CursorContext;
import org.neo4j.kernel.api.ExecutionContext;
import org.neo4j.kernel.api.KernelTransaction;
import org.neo4j.kernel.api.ResourceMonitor;
import org.neo4j.kernel.api.exceptions.Status;
import org.neo4j.kernel.impl.coreapi.DataLookup;
import org.neo4j.kernel.impl.coreapi.InternalTransaction;
import org.neo4j.kernel.impl.coreapi.TransactionImpl;
import org.neo4j.kernel.impl.coreapi.internal.CursorIterator;
import org.neo4j.kernel.impl.coreapi.schema.SchemaImpl;
import org.neo4j.memory.MemoryTracker;
import org.neo4j.values.ElementIdMapper;
public class ExecutionContextProcedureTransaction extends DataLookup implements InternalTransaction {
private final ExecutionContext executionContext;
private final ExecutionContextProcedureKernelTransaction ktx;
public ExecutionContextProcedureTransaction(ExecutionContextProcedureKernelTransaction ktx) {
this.executionContext = ktx.executionContext();
this.ktx = ktx;
}
@Override
public Node createNode() {
throw new UnsupportedOperationException("Write operations are unsupported during parallel execution");
}
@Override
public Node createNode(Label... labels) {
throw new UnsupportedOperationException("Write operations are unsupported during parallel execution");
}
@Override
public BidirectionalTraversalDescription bidirectionalTraversalDescription() {
throw new UnsupportedOperationException("Traversal operations are unsupported during parallel execution");
}
@Override
public TraversalDescription traversalDescription() {
throw new UnsupportedOperationException("Traversal operations are unsupported during parallel execution");
}
@Override
public Result execute(String query) throws QueryExecutionException {
throw new UnsupportedOperationException("Execution of other queries is unsupported during parallel execution");
}
@Override
public Result execute(String query, Map parameters) throws QueryExecutionException {
throw new UnsupportedOperationException("Execution of other queries is unsupported during parallel execution");
}
@Override
public ResourceIterable getAllNodes() {
var result = new AbstractResourceIterable() {
@Override
protected ResourceIterator newIterator() {
NodeCursor cursor = cursors().allocateNodeCursor(cursorContext(), memoryTracker());
dataRead().allNodesScan(cursor);
return new CursorIterator<>(cursor, NodeCursor::nodeReference, c -> newNodeEntity(c.nodeReference()));
}
@Override
protected void onClosed() {
executionContext.unregisterCloseableResource(this);
}
};
executionContext.registerCloseableResource(result);
return result;
}
@Override
public ResourceIterable getAllRelationships() {
var result = new AbstractResourceIterable() {
@Override
protected ResourceIterator newIterator() {
RelationshipScanCursor cursor =
cursors().allocateRelationshipScanCursor(cursorContext(), memoryTracker());
dataRead().allRelationshipsScan(cursor);
return new CursorIterator<>(
cursor,
RelationshipScanCursor::relationshipReference,
c -> newRelationshipEntity(c.relationshipReference()));
}
@Override
protected void onClosed() {
executionContext.unregisterCloseableResource(this);
}
};
executionContext.registerCloseableResource(result);
return result;
}
// Explicit locking could be easily supported, but it would give the users
// tooling to create deadlocks within one transaction
@Override
public Lock acquireWriteLock(Entity entity) {
throw new UnsupportedOperationException("Acquiring locks is unsupported during parallel execution.");
}
@Override
public Lock acquireReadLock(Entity entity) {
throw new UnsupportedOperationException("Acquiring locks is unsupported during parallel execution.");
}
@Override
public Schema schema() {
return new SchemaImpl(kernelTransaction());
}
@Override
public void terminate() {
throw new UnsupportedOperationException(
"Terminating ongoing transaction inside of a procedure or a function is unsupported.");
}
@Override
public void commit() {
throw new UnsupportedOperationException(
"Committing ongoing transaction inside of a procedure or a function is unsupported.");
}
@Override
public void rollback() {
throw new UnsupportedOperationException(
"Rolling back ongoing transaction inside of a procedure or a function is unsupported.");
}
@Override
public void close() {
throw new UnsupportedOperationException(
"Closing ongoing transaction inside of a procedure or a function is unsupported.");
}
@Override
protected TokenRead tokenRead() {
return executionContext.tokenRead();
}
@Override
protected SchemaRead schemaRead() {
return executionContext.schemaRead();
}
@Override
protected Read dataRead() {
return executionContext.dataRead();
}
@Override
protected ResourceMonitor resourceMonitor() {
return executionContext;
}
@Override
public Node newNodeEntity(long nodeId) {
return new ExecutionContextNode(nodeId, executionContext);
}
@Override
public RelationshipType getRelationshipTypeById(int type) {
try {
return RelationshipType.withName(executionContext.tokenRead().relationshipTypeName(type));
} catch (KernelException e) {
throw new IllegalStateException("Kernel API returned non-existent relationship type: " + type, e);
}
}
@Override
public Relationship newRelationshipEntity(long relationshipId) {
return new ExecutionContextRelationship(relationshipId, executionContext);
}
@Override
public Relationship newRelationshipEntity(String elementId) {
return new ExecutionContextRelationship(elementIdMapper().relationshipId(elementId), executionContext);
}
@Override
public Relationship newRelationshipEntity(long id, long startNodeId, int typeId, long endNodeId) {
return newRelationshipEntity(id);
}
@Override
public Relationship newRelationshipEntity(RelationshipDataAccessor cursor) {
return newRelationshipEntity(cursor.relationshipReference());
}
@Override
protected CursorFactory cursors() {
return executionContext.cursors();
}
@Override
protected CursorContext cursorContext() {
return executionContext.cursorContext();
}
@Override
protected MemoryTracker memoryTracker() {
return executionContext.memoryTracker();
}
@Override
protected QueryContext queryContext() {
return executionContext.queryContext();
}
@Override
public SecurityContext securityContext() {
return executionContext.securityContext();
}
@Override
public void setTransaction(KernelTransaction transaction) {
throw failure("setTransaction");
}
@Override
public KernelTransaction kernelTransaction() {
checkInTransaction();
return ktx;
}
@Override
public KernelTransaction.Type transactionType() {
return ktx.transactionType();
}
@Override
public ClientConnectionInfo clientInfo() {
return ktx.clientInfo();
}
@Override
public KernelTransaction.Revertable overrideWith(SecurityContext context) {
return ktx.overrideWith(context);
}
@Override
public Optional terminationReason() {
return ktx.getReasonIfTerminated();
}
@Override
public void setMetaData(Map txMeta) {
ktx.setMetaData(txMeta);
}
@Override
public void checkInTransaction() {
if (ktx.isTerminated()) {
Status terminationReason = ktx.getReasonIfTerminated().orElse(Status.Transaction.Terminated);
throw new TransactionTerminatedException(terminationReason);
}
}
@Override
public boolean isOpen() {
return ktx.isOpen();
}
@Override
public void terminate(Status reason) {
terminate();
}
@Override
public UUID getDatabaseId() {
return ktx.getDatabaseId();
}
@Override
public String getDatabaseName() {
return ktx.getDatabaseName();
}
@Override
public Entity validateSameDB(Entity entity) {
return TransactionImpl.validateSameDB(this, entity);
}
@Override
public ElementIdMapper elementIdMapper() {
return executionContext.elementIdMapper();
}
@Override
public void commit(KernelTransaction.KernelTransactionMonitor monitor) {
commit();
}
@Override
protected void performCheckBeforeOperation() {
executionContext.performCheckBeforeOperation();
}
@Override
public void registerCloseableResource(AutoCloseable closeableResource) {
executionContext.registerCloseableResource(closeableResource);
}
@Override
public void unregisterCloseableResource(AutoCloseable closeableResource) {
executionContext.unregisterCloseableResource(closeableResource);
}
}