Skip to content

Commit

Permalink
Make sure to fail and remove transaction when a query fails
Browse files Browse the repository at this point in the history
  • Loading branch information
@systay
systay committed Sep 28, 2016
1 parent 5050009 commit b36da8c
Show file tree
Hide file tree
Showing 2 changed files with 54 additions and 51 deletions.
82 changes: 40 additions & 42 deletions community/bolt/src/main/java/org/neo4j/bolt/v1/runtime/TransactionStateMachine.java
View file
Expand Up @@ -30,6 +30,7 @@
import org.neo4j.bolt.v1.runtime.spi.BoltResult;
import org.neo4j.bolt.v1.runtime.spi.BookmarkResult;
import org.neo4j.cypher.InvalidSemanticsException;
import org.neo4j.function.ThrowingAction;
import org.neo4j.function.ThrowingConsumer;
import org.neo4j.kernel.api.KernelTransaction;
import org.neo4j.kernel.api.exceptions.KernelException;
Expand Down Expand Up @@ -171,16 +172,19 @@ else if ( statement.equalsIgnoreCase( ROLLBACK ) )
}
else if ( spi.isPeriodicCommit( statement ) )
{
ctx.currentTransaction = null;
ctx.currentResultHandle = executeQuery( ctx, spi, statement, params );
ctx.currentResult = ctx.currentResultHandle.start();
BoltResultHandle resultHandle = executeQuery( ctx, spi, statement, params, () -> {} );
ctx.currentResultHandle = resultHandle;
ctx.currentResult = resultHandle.start();
ctx.currentTransaction = null; // Periodic commit will change the current transaction, so
// we can't trust this to point to the actual current transaction;
return AUTO_COMMIT;
}
else
{
ctx.currentTransaction = spi.beginTransaction( ctx.authSubject );
ctx.currentResultHandle = execute( ctx, spi, statement, params );
ctx.currentResult = ctx.currentResultHandle.start();
BoltResultHandle resultHandle = execute( ctx, spi, statement, params );
ctx.currentResultHandle = resultHandle;
ctx.currentResult = resultHandle.start();
return AUTO_COMMIT;
}
}
Expand All @@ -190,29 +194,21 @@ else if ( spi.isPeriodicCommit( statement ) )
* transaction to null.
*/
private BoltResultHandle execute( MutableTransactionState ctx, SPI spi,
String statement, Map<String,Object> params )
String statement, Map<String,Object> params )
throws TransactionFailureException, QueryExecutionKernelException
{
try
return executeQuery( ctx, spi, statement, params, () ->
{
return executeQuery( ctx, spi, statement, params );
}
catch ( Throwable e )
{
if (ctx.currentTransaction != null)
try // On fail
{
try
{
ctx.currentTransaction.failure();
ctx.currentTransaction.close();
}
finally
{
ctx.currentTransaction = null;
}
ctx.currentTransaction.failure();
ctx.currentTransaction.close();
}
throw e;
}
finally
{
ctx.currentTransaction = null;
}
} );
}

@Override
Expand Down Expand Up @@ -276,21 +272,17 @@ else if( spi.isPeriodicCommit( statement ) )
}

private BoltResultHandle execute( MutableTransactionState ctx, SPI spi,
String statement, Map<String,Object> params )
String statement, Map<String,Object> params )
throws QueryExecutionKernelException
{
try
{
return executeQuery( ctx, spi, statement, params );
}
catch ( Throwable e )
{
if (ctx.currentTransaction != null)
{
ctx.currentTransaction.failure();
}
throw e;
}
return executeQuery( ctx, spi, statement, params,
() ->
{
if ( ctx.currentTransaction != null )
{
ctx.currentTransaction.failure();
}
} );
}

@Override
Expand Down Expand Up @@ -332,15 +324,20 @@ void terminateQueryAndRollbackTransaction( MutableTransactionState ctx ) throws
}

private static BoltResultHandle executeQuery( MutableTransactionState ctx, SPI spi, String statement,
Map<String,Object> params )
Map<String,Object> params, ThrowingAction<KernelException> onFail )
throws QueryExecutionKernelException
{
return spi.executeQuery( ctx.querySource, ctx.authSubject, statement, params );
return spi.executeQuery( ctx.querySource, ctx.authSubject, statement, params, onFail );
}

/**
* This interface makes it possible to abort queries even before they have returned a Result object.
* In some cases, creating the Result object will take as long as running the query takes. This way, we can
* terminate the underlying transaction while the Result object is created.
*/
interface BoltResultHandle
{
BoltResult start() throws QueryExecutionKernelException;
BoltResult start() throws KernelException;
void terminate();
}

Expand Down Expand Up @@ -392,8 +389,9 @@ interface SPI
boolean isPeriodicCommit( String query );

BoltResultHandle executeQuery( String querySource,
AuthSubject authSubject,
String statement,
Map<String,Object> params ) throws QueryExecutionKernelException;
AuthSubject authSubject,
String statement,
Map<String,Object> params,
ThrowingAction<KernelException> onFail ) throws QueryExecutionKernelException;
}
}
23 changes: 14 additions & 9 deletions community/bolt/src/main/java/org/neo4j/bolt/v1/runtime/TransactionStateMachineSPI.java
View file
Expand Up @@ -26,6 +26,7 @@
import org.neo4j.bolt.v1.runtime.TransactionStateMachine.BoltResultHandle;
import org.neo4j.bolt.v1.runtime.cypher.CypherAdapterStream;
import org.neo4j.bolt.v1.runtime.spi.BoltResult;
import org.neo4j.function.ThrowingAction;
import org.neo4j.graphdb.Result;
import org.neo4j.kernel.GraphDatabaseQueryService;
import org.neo4j.kernel.api.KernelTransaction;
Expand Down Expand Up @@ -113,19 +114,19 @@ public boolean isPeriodicCommit( String query )

@Override
public BoltResultHandle executeQuery( String querySource,
AuthSubject authSubject,
String statement,
Map<String,Object> params ) throws QueryExecutionKernelException
AuthSubject authSubject,
String statement,
Map<String,Object> params, ThrowingAction<KernelException> onFail ) throws QueryExecutionKernelException
{
InternalTransaction transaction = queryService.beginTransaction( implicit, authSubject );
InternalTransaction internalTransaction = queryService.beginTransaction( implicit, authSubject );
QuerySource sourceDetails = new QuerySource( "bolt-session", querySource );
TransactionalContext transactionalContext =
contextFactory.newContext( sourceDetails, transaction, statement, params );
contextFactory.newContext( sourceDetails, internalTransaction, statement, params );

return new BoltResultHandle()
{
@Override
public BoltResult start() throws QueryExecutionKernelException
public BoltResult start() throws KernelException
{
try
{
Expand All @@ -134,9 +135,14 @@ public BoltResult start() throws QueryExecutionKernelException
}
catch ( KernelException e )
{
onFail.apply();
throw new QueryExecutionKernelException( e );
}

catch ( Throwable e )
{
onFail.apply();
throw e;
}
}

@Override
Expand All @@ -145,6 +151,5 @@ public void terminate()
transactionalContext.terminate();
}
};

}
}
}

0 comments on commit b36da8c

Please sign in to comment.