[incubator-gobblin] branch master updated: [GOBBLIN-1378] Fix the issue that task Hang after restart

[lesun]

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

lesun pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-gobblin.git


The following commit(s) were added to refs/heads/master by this push:
     new ad69a59  [GOBBLIN-1378] Fix the issue that task Hang after restart
ad69a59 is described below

commit ad69a596664242e1ad8bed53acec4ad437ca04f8
Author: Zihan Li <zi...@linkedin.com>
AuthorDate: Mon Feb 1 10:32:02 2021 -0800

    [GOBBLIN-1378] Fix the issue that task Hang after restart
    
    Closes #3218 from ZihanLi58/GOBBLIN-1378
---
 .../src/main/java/org/apache/gobblin/runtime/fork/Fork.java            | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git 
a/gobblin-runtime/src/main/java/org/apache/gobblin/runtime/fork/Fork.java 
b/gobblin-runtime/src/main/java/org/apache/gobblin/runtime/fork/Fork.java
index 8de98f5..f9bc862 100644
--- a/gobblin-runtime/src/main/java/org/apache/gobblin/runtime/fork/Fork.java
+++ b/gobblin-runtime/src/main/java/org/apache/gobblin/runtime/fork/Fork.java
@@ -434,7 +434,8 @@ public class Fork<S, D> implements Closeable, FinalState, 
RecordStreamConsumer<S
   }
 
   public boolean isDone() {
-    return this.forkState.get() == ForkState.SUCCEEDED || this.forkState.get() 
== ForkState.FAILED;
+    return this.forkState.get() == ForkState.SUCCEEDED || this.forkState.get() 
== ForkState.FAILED
+        || this.forkState.get() == ForkState.COMMITTED;
   }
 
   @Override