use cancellation token in driver row source (#771)

RichardIrons-neo4j · web-flow · commit 84b0ea94a245 · 2024-01-18T15:52:41.000Z
diff --git a/Neo4j.Driver/Neo4j.Driver/Internal/Driver.cs b/Neo4j.Driver/Neo4j.Driver/Internal/Driver.cs
@@ -156,7 +156,7 @@ CancellationToken cancellationToken
     {
         async Task<int> Process(IAsyncEnumerable<IRecord> records)
         {
-            await foreach (var record in records.ConfigureAwait(false))
+            await foreach (var record in records.ConfigureAwait(false).WithCancellation(cancellationToken))
             {
                 streamProcessor(record);
             }

Original file line number	Diff line number	Diff line change
`@@ -156,7 +156,7 @@ CancellationToken cancellationToken`
`156`	`156`	`{`
`157`	`157`	`async Task<int> Process(IAsyncEnumerable<IRecord> records)`
`158`	`158`	`{`
`159`		`- await foreach (var record in records.ConfigureAwait(false))`
	`159`	`+ await foreach (var record in records.ConfigureAwait(false).WithCancellation(cancellationToken))`
`160`	`160`	`{`
`161`	`161`	`streamProcessor(record);`
`162`	`162`	`}`