[beam] 01/01: [BEAM-4287] Fix to use the residual instead of the current restriction on process continuations.

[ud...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

udim pushed a commit to branch release-2.18.0
in repository https://gitbox.apache.org/repos/asf/beam.git

commit 9e58be17043c9dcbd36dc5e48939f358d877c366
Author: Luke Cwik <lc...@google.com>
AuthorDate: Thu Dec 5 13:42:16 2019 -0800

    [BEAM-4287] Fix to use the residual instead of the current restriction on process continuations.
    
    This is a forward fix for https://github.com/apache/beam/pull/10258 to fix Dataflow VR postcommit suite.
---
 .../beam/runners/core/construction/SplittableParDoNaiveBounded.java     | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/runners/core-construction-java/src/main/java/org/apache/beam/runners/core/construction/SplittableParDoNaiveBounded.java b/runners/core-construction-java/src/main/java/org/apache/beam/runners/core/construction/SplittableParDoNaiveBounded.java
index d975b4f..8a41d7e 100644
--- a/runners/core-construction-java/src/main/java/org/apache/beam/runners/core/construction/SplittableParDoNaiveBounded.java
+++ b/runners/core-construction-java/src/main/java/org/apache/beam/runners/core/construction/SplittableParDoNaiveBounded.java
@@ -147,7 +147,7 @@ public class SplittableParDoNaiveBounded {
         ProcessContinuation continuation =
             invoker.invokeProcessElement(new NestedProcessContext<>(fn, c, element, w, tracker));
         if (continuation.shouldResume()) {
-          restriction = tracker.currentRestriction();
+          restriction = tracker.trySplit(0).getResidual();
           Uninterruptibles.sleepUninterruptibly(
               continuation.resumeDelay().getMillis(), TimeUnit.MILLISECONDS);
         } else {