Skip to content

Commit

Permalink
[BEAM-12419] Deadline should be reset for each gRPC call.
Browse files Browse the repository at this point in the history
getState() in waitForTerminalState() is invoked multiple times and deadline should be reset each time since it is an absolute number.
  • Loading branch information
Ke Wu committed Jun 11, 2021
1 parent 0b369a5 commit eb65b1b
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -153,7 +153,7 @@ private void waitForTerminalState() {
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;
Expand Down

0 comments on commit eb65b1b

Please sign in to comment.