20
20
21
21
import java .util .Collections ;
22
22
import java .util .Map ;
23
+ import java .util .concurrent .CompletionStage ;
23
24
24
25
import org .neo4j .driver .ResultResourcesHandler ;
25
26
import org .neo4j .driver .internal .async .AsyncConnection ;
35
36
import org .neo4j .driver .internal .types .InternalTypeSystem ;
36
37
import org .neo4j .driver .internal .util .BiConsumer ;
37
38
import org .neo4j .driver .v1 .Record ;
38
- import org .neo4j .driver .v1 .Response ;
39
39
import org .neo4j .driver .v1 .Statement ;
40
40
import org .neo4j .driver .v1 .StatementResult ;
41
41
import org .neo4j .driver .v1 .StatementResultCursor ;
47
47
import org .neo4j .driver .v1 .types .TypeSystem ;
48
48
import org .neo4j .driver .v1 .util .Function ;
49
49
50
+ import static org .neo4j .driver .internal .async .Futures .asCompletionStage ;
50
51
import static org .neo4j .driver .internal .util .ErrorUtil .isRecoverable ;
51
52
import static org .neo4j .driver .v1 .Values .ofValue ;
52
53
import static org .neo4j .driver .v1 .Values .value ;
@@ -215,9 +216,9 @@ private void rollbackTx()
215
216
}
216
217
217
218
@ Override
218
- public Response <Void > commitAsync ()
219
+ public CompletionStage <Void > commitAsync ()
219
220
{
220
- return internalCommitAsync ();
221
+ return asCompletionStage ( internalCommitAsync () );
221
222
}
222
223
223
224
InternalFuture <Void > internalCommitAsync ()
@@ -238,9 +239,9 @@ else if ( state == State.ROLLED_BACK )
238
239
}
239
240
240
241
@ Override
241
- public Response <Void > rollbackAsync ()
242
+ public CompletionStage <Void > rollbackAsync ()
242
243
{
243
- return internalRollbackAsync ();
244
+ return asCompletionStage ( internalRollbackAsync () );
244
245
}
245
246
246
247
InternalFuture <Void > internalRollbackAsync ()
@@ -317,7 +318,7 @@ public StatementResult run( String statementText, Value statementParameters )
317
318
}
318
319
319
320
@ Override
320
- public Response <StatementResultCursor > runAsync ( String statementText , Value parameters )
321
+ public CompletionStage <StatementResultCursor > runAsync ( String statementText , Value parameters )
321
322
{
322
323
return runAsync ( new Statement ( statementText , parameters ) );
323
324
}
@@ -329,7 +330,7 @@ public StatementResult run( String statementText )
329
330
}
330
331
331
332
@ Override
332
- public Response <StatementResultCursor > runAsync ( String statementTemplate )
333
+ public CompletionStage <StatementResultCursor > runAsync ( String statementTemplate )
333
334
{
334
335
return runAsync ( statementTemplate , Values .EmptyMap );
335
336
}
@@ -342,7 +343,8 @@ public StatementResult run( String statementText, Map<String,Object> statementPa
342
343
}
343
344
344
345
@ Override
345
- public Response <StatementResultCursor > runAsync ( String statementTemplate , Map <String ,Object > statementParameters )
346
+ public CompletionStage <StatementResultCursor > runAsync ( String statementTemplate ,
347
+ Map <String ,Object > statementParameters )
346
348
{
347
349
Value params = statementParameters == null ? Values .EmptyMap : value ( statementParameters );
348
350
return runAsync ( statementTemplate , params );
@@ -356,7 +358,7 @@ public StatementResult run( String statementTemplate, Record statementParameters
356
358
}
357
359
358
360
@ Override
359
- public Response <StatementResultCursor > runAsync ( String statementTemplate , Record statementParameters )
361
+ public CompletionStage <StatementResultCursor > runAsync ( String statementTemplate , Record statementParameters )
360
362
{
361
363
Value params = statementParameters == null ? Values .EmptyMap : value ( statementParameters .asMap () );
362
364
return runAsync ( statementTemplate , params );
@@ -388,10 +390,10 @@ public StatementResult run( Statement statement )
388
390
}
389
391
390
392
@ Override
391
- public Response <StatementResultCursor > runAsync ( Statement statement )
393
+ public CompletionStage <StatementResultCursor > runAsync ( Statement statement )
392
394
{
393
395
ensureNotFailed ();
394
- return QueryRunner .runAsync ( asyncConnection , statement , this );
396
+ return asCompletionStage ( QueryRunner .runAsync ( asyncConnection , statement , this ) );
395
397
}
396
398
397
399
@ Override
0 commit comments