[beam] branch master updated: [BEAM-12419] Deadline should be reset for each gRPC call.

[ibzib]

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

ibzib pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/beam.git


The following commit(s) were added to refs/heads/master by this push:
     new eb65b1b  [BEAM-12419] Deadline should be reset for each gRPC call.
     new c0b8e65  Merge pull request #14995 from kw2542/BEAM-12419
eb65b1b is described below

commit eb65b1bd6ca1c2774f1cab09e0dfec90cfb0a1c7
Author: Ke Wu <k...@kwu-mn1.linkedin.biz>
AuthorDate: Thu Jun 10 21:17:11 2021 -0700

    [BEAM-12419] Deadline should be reset for each gRPC call.
    
    getState() in waitForTerminalState() is invoked multiple times and deadline 
should be reset each time since it is an absolute number.
---
 .../org/apache/beam/runners/portability/JobServicePipelineResult.java   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git 
a/runners/portability/java/src/main/java/org/apache/beam/runners/portability/JobServicePipelineResult.java
 
b/runners/portability/java/src/main/java/org/apache/beam/runners/portability/JobServicePipelineResult.java
index 478c755..b631a81 100644
--- 
a/runners/portability/java/src/main/java/org/apache/beam/runners/portability/JobServicePipelineResult.java
+++ 
b/runners/portability/java/src/main/java/org/apache/beam/runners/portability/JobServicePipelineResult.java
@@ -153,7 +153,7 @@ class JobServicePipelineResult implements PipelineResult, 
AutoCloseable {
         Thread.currentThread().interrupt();
         throw new RuntimeException(e);
       }
-      response = stub.getState(request);
+      response = stub.withDeadlineAfter(jobServerTimeout, 
TimeUnit.SECONDS).getState(request);
       lastState = getJavaState(response.getState());
     }
     terminalState = lastState;