You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@seatunnel.apache.org by fa...@apache.org on 2023/12/13 02:37:59 UTC

(seatunnel) branch dev updated: [Feature][Zeta] Add waitForJobCompleteV2 api (#5965)

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

fanjia pushed a commit to branch dev
in repository https://gitbox.apache.org/repos/asf/seatunnel.git


The following commit(s) were added to refs/heads/dev by this push:
     new 748efe55bb [Feature][Zeta] Add waitForJobCompleteV2 api (#5965)
748efe55bb is described below

commit 748efe55bbd5bbf3659a918751fe96312feff119
Author: hailin0 <wa...@apache.org>
AuthorDate: Wed Dec 13 10:37:52 2023 +0800

    [Feature][Zeta] Add waitForJobCompleteV2 api (#5965)
---
 .../java/org/apache/seatunnel/engine/client/job/ClientJobProxy.java | 4 ++--
 .../src/main/java/org/apache/seatunnel/engine/core/job/Job.java     | 6 +++++-
 .../main/java/org/apache/seatunnel/engine/core/job/JobResult.java   | 4 ++++
 3 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/seatunnel-engine/seatunnel-engine-client/src/main/java/org/apache/seatunnel/engine/client/job/ClientJobProxy.java b/seatunnel-engine/seatunnel-engine-client/src/main/java/org/apache/seatunnel/engine/client/job/ClientJobProxy.java
index 0e877f6412..ceec9b33dc 100644
--- a/seatunnel-engine/seatunnel-engine-client/src/main/java/org/apache/seatunnel/engine/client/job/ClientJobProxy.java
+++ b/seatunnel-engine/seatunnel-engine-client/src/main/java/org/apache/seatunnel/engine/client/job/ClientJobProxy.java
@@ -91,7 +91,7 @@ public class ClientJobProxy implements Job {
      * @return The job final status
      */
     @Override
-    public JobStatus waitForJobComplete() {
+    public JobResult waitForJobCompleteV2() {
         try {
             jobResult =
                     RetryUtils.retryWithException(
@@ -121,7 +121,7 @@ public class ClientJobProxy implements Job {
                 || jobResult.getStatus().equals(JobStatus.FAILED)) {
             throw new SeaTunnelEngineException(jobResult.getError());
         }
-        return jobResult.getStatus();
+        return jobResult;
     }
 
     public JobResult getJobResultCache() {
diff --git a/seatunnel-engine/seatunnel-engine-core/src/main/java/org/apache/seatunnel/engine/core/job/Job.java b/seatunnel-engine/seatunnel-engine-core/src/main/java/org/apache/seatunnel/engine/core/job/Job.java
index 88b690b967..3d4ee7593b 100644
--- a/seatunnel-engine/seatunnel-engine-core/src/main/java/org/apache/seatunnel/engine/core/job/Job.java
+++ b/seatunnel-engine/seatunnel-engine-core/src/main/java/org/apache/seatunnel/engine/core/job/Job.java
@@ -29,5 +29,9 @@ public interface Job {
 
     JobStatus getJobStatus();
 
-    JobStatus waitForJobComplete();
+    default JobStatus waitForJobComplete() {
+        return waitForJobCompleteV2().getStatus();
+    }
+
+    JobResult waitForJobCompleteV2();
 }
diff --git a/seatunnel-engine/seatunnel-engine-core/src/main/java/org/apache/seatunnel/engine/core/job/JobResult.java b/seatunnel-engine/seatunnel-engine-core/src/main/java/org/apache/seatunnel/engine/core/job/JobResult.java
index b2bf1536c3..5f946ffe45 100644
--- a/seatunnel-engine/seatunnel-engine-core/src/main/java/org/apache/seatunnel/engine/core/job/JobResult.java
+++ b/seatunnel-engine/seatunnel-engine-core/src/main/java/org/apache/seatunnel/engine/core/job/JobResult.java
@@ -30,4 +30,8 @@ public class JobResult implements Serializable {
     @NonNull private JobStatus status;
 
     private String error;
+
+    public JobResult(@NonNull JobStatus status) {
+        this.status = status;
+    }
 }