use of org.neo4j.kernel.api.Statement in project neo4j by neo4j.
the class Neo4jTransactionalContextFactory method create.
@Deprecated
public static TransactionalContextFactory create(GraphDatabaseQueryService queryService, PropertyContainerLocker locker) {
DependencyResolver resolver = queryService.getDependencyResolver();
ThreadToStatementContextBridge txBridge = resolver.resolveDependency(ThreadToStatementContextBridge.class);
Guard guard = resolver.resolveDependency(Guard.class);
Neo4jTransactionalContext.Creator contextCreator = (Supplier<Statement> statementSupplier, InternalTransaction tx, Statement initialStatement, ExecutingQuery executingQuery) -> new Neo4jTransactionalContext(queryService, statementSupplier, guard, txBridge, locker, tx, initialStatement, executingQuery);
return new Neo4jTransactionalContextFactory(txBridge, contextCreator);
}
use of org.neo4j.kernel.api.Statement in project neo4j by neo4j.
the class Neo4jTransactionalContextFactory method newContext.
@Override
public final Neo4jTransactionalContext newContext(ClientConnectionInfo clientConnection, InternalTransaction tx, String queryText, Map<String, Object> queryParameters) {
Statement initialStatement = statementSupplier.get();
ClientConnectionInfo connectionWithUserName = clientConnection.withUsername(tx.securityContext().subject().username());
ExecutingQuery executingQuery = initialStatement.queryRegistration().startQueryExecution(connectionWithUserName, queryText, queryParameters);
return contextCreator.create(statementSupplier, tx, initialStatement, executingQuery);
}
use of org.neo4j.kernel.api.Statement in project neo4j by neo4j.
the class Neo4jTransactionalContextTest method neverStopsExecutingQueryDuringCommitAndRestartTx.
@SuppressWarnings("ConstantConditions")
@Test
public void neverStopsExecutingQueryDuringCommitAndRestartTx() {
// Given
KernelTransaction initialKTX = mock(KernelTransaction.class);
InternalTransaction initialTransaction = mock(InternalTransaction.class, new ReturnsDeepStubs());
KernelTransaction.Type transactionType = KernelTransaction.Type.implicit;
SecurityContext securityContext = SecurityContext.AUTH_DISABLED;
when(initialTransaction.transactionType()).thenReturn(transactionType);
when(initialTransaction.securityContext()).thenReturn(securityContext);
QueryRegistryOperations initialQueryRegistry = mock(QueryRegistryOperations.class);
ExecutingQuery executingQuery = mock(ExecutingQuery.class);
PropertyContainerLocker locker = null;
ThreadToStatementContextBridge txBridge = mock(ThreadToStatementContextBridge.class);
KernelTransaction secondKTX = mock(KernelTransaction.class);
InternalTransaction secondTransaction = mock(InternalTransaction.class);
Statement secondStatement = mock(Statement.class);
QueryRegistryOperations secondQueryRegistry = mock(QueryRegistryOperations.class);
when(executingQuery.queryText()).thenReturn("X");
when(executingQuery.queryParameters()).thenReturn(Collections.emptyMap());
when(initialStatement.queryRegistration()).thenReturn(initialQueryRegistry);
when(queryService.beginTransaction(transactionType, securityContext)).thenReturn(secondTransaction);
when(txBridge.getKernelTransactionBoundToThisThread(true)).thenReturn(initialKTX, secondKTX);
when(txBridge.get()).thenReturn(secondStatement);
when(secondStatement.queryRegistration()).thenReturn(secondQueryRegistry);
Neo4jTransactionalContext context = new Neo4jTransactionalContext(queryService, null, guard, txBridge, locker, initialTransaction, initialStatement, executingQuery);
// When
context.commitAndRestartTx();
// Then
Object[] mocks = { txBridge, initialTransaction, initialQueryRegistry, initialKTX, secondQueryRegistry, secondKTX };
InOrder order = Mockito.inOrder(mocks);
// (0) Constructor
order.verify(initialTransaction).transactionType();
order.verify(initialTransaction).securityContext();
// (1) Unbind old
order.verify(txBridge).getKernelTransactionBoundToThisThread(true);
order.verify(txBridge).unbindTransactionFromCurrentThread();
// (2) Register and unbind new
order.verify(txBridge).get();
order.verify(secondQueryRegistry).registerExecutingQuery(executingQuery);
order.verify(txBridge).getKernelTransactionBoundToThisThread(true);
order.verify(txBridge).unbindTransactionFromCurrentThread();
// (3) Rebind, unregister, and close old
order.verify(txBridge).bindTransactionToCurrentThread(initialKTX);
order.verify(initialQueryRegistry).unregisterExecutingQuery(executingQuery);
order.verify(initialTransaction).success();
order.verify(initialTransaction).close();
order.verify(txBridge).unbindTransactionFromCurrentThread();
// (4) Rebind new
order.verify(txBridge).bindTransactionToCurrentThread(secondKTX);
verifyNoMoreInteractions(mocks);
}
use of org.neo4j.kernel.api.Statement in project neo4j by neo4j.
the class Neo4jTransactionalContextTest method rollsBackNewlyCreatedTransactionIfTerminationDetectedOnCloseDuringPeriodicCommit.
@SuppressWarnings("ConstantConditions")
@Test
public void rollsBackNewlyCreatedTransactionIfTerminationDetectedOnCloseDuringPeriodicCommit() {
// Given
InternalTransaction initialTransaction = mock(InternalTransaction.class, new ReturnsDeepStubs());
KernelTransaction.Type transactionType = KernelTransaction.Type.implicit;
SecurityContext securityContext = SecurityContext.AUTH_DISABLED;
when(initialTransaction.transactionType()).thenReturn(transactionType);
when(initialTransaction.securityContext()).thenReturn(securityContext);
GraphDatabaseQueryService queryService = mock(GraphDatabaseQueryService.class);
KernelTransaction initialKTX = mock(KernelTransaction.class);
Statement initialStatement = mock(Statement.class);
QueryRegistryOperations initialQueryRegistry = mock(QueryRegistryOperations.class);
ExecutingQuery executingQuery = mock(ExecutingQuery.class);
PropertyContainerLocker locker = new PropertyContainerLocker();
ThreadToStatementContextBridge txBridge = mock(ThreadToStatementContextBridge.class);
KernelTransaction secondKTX = mock(KernelTransaction.class);
InternalTransaction secondTransaction = mock(InternalTransaction.class);
Statement secondStatement = mock(Statement.class);
QueryRegistryOperations secondQueryRegistry = mock(QueryRegistryOperations.class);
when(executingQuery.queryText()).thenReturn("X");
when(executingQuery.queryParameters()).thenReturn(Collections.emptyMap());
Mockito.doThrow(RuntimeException.class).when(initialTransaction).close();
when(initialStatement.queryRegistration()).thenReturn(initialQueryRegistry);
when(queryService.beginTransaction(transactionType, securityContext)).thenReturn(secondTransaction);
when(txBridge.getKernelTransactionBoundToThisThread(true)).thenReturn(initialKTX, secondKTX);
when(txBridge.get()).thenReturn(secondStatement);
when(secondStatement.queryRegistration()).thenReturn(secondQueryRegistry);
Neo4jTransactionalContext context = new Neo4jTransactionalContext(queryService, null, guard, txBridge, locker, initialTransaction, initialStatement, executingQuery);
// When
try {
context.commitAndRestartTx();
throw new AssertionError("Expected RuntimeException to be thrown");
} catch (RuntimeException e) {
// Then
Object[] mocks = { txBridge, initialTransaction, initialQueryRegistry, initialKTX, secondQueryRegistry, secondKTX, secondTransaction };
InOrder order = Mockito.inOrder(mocks);
// (0) Constructor
order.verify(initialTransaction).transactionType();
order.verify(initialTransaction).securityContext();
// (1) Unbind old
order.verify(txBridge).getKernelTransactionBoundToThisThread(true);
order.verify(txBridge).unbindTransactionFromCurrentThread();
// (2) Register and unbind new
order.verify(txBridge).get();
order.verify(secondQueryRegistry).registerExecutingQuery(executingQuery);
order.verify(txBridge).getKernelTransactionBoundToThisThread(true);
order.verify(txBridge).unbindTransactionFromCurrentThread();
// (3) Rebind, unregister, and close old
order.verify(txBridge).bindTransactionToCurrentThread(initialKTX);
order.verify(initialQueryRegistry).unregisterExecutingQuery(executingQuery);
order.verify(initialTransaction).success();
order.verify(initialTransaction).close();
order.verify(txBridge).bindTransactionToCurrentThread(secondKTX);
order.verify(secondTransaction).failure();
order.verify(secondTransaction).close();
order.verify(txBridge).unbindTransactionFromCurrentThread();
verifyNoMoreInteractions(mocks);
}
}
use of org.neo4j.kernel.api.Statement in project neo4j by neo4j.
the class BuiltInProceduresIT method listRelationshipTypes.
@Test
public void listRelationshipTypes() throws Throwable {
// Given
Statement statement = statementInNewTransaction(AnonymousContext.writeToken());
int relType = statement.tokenWriteOperations().relationshipTypeGetOrCreateForName("MyRelType");
long startNodeId = statement.dataWriteOperations().nodeCreate();
long endNodeId = statement.dataWriteOperations().nodeCreate();
statement.dataWriteOperations().relationshipCreate(relType, startNodeId, endNodeId);
commit();
// When
RawIterator<Object[], ProcedureException> stream = procedureCallOpsInNewTx().procedureCallRead(procedureName("db", "relationshipTypes"), new Object[0]);
// Then
assertThat(asList(stream), contains(equalTo(new Object[] { "MyRelType" })));
}
Aggregations