From 9a53b486d84419c2af85586d4b0d5ea75e4a06b2 Mon Sep 17 00:00:00 2001
From: ulixius9 <mayursingal9@gmail.com>
Date: Thu, 26 Sep 2024 21:04:56 +0530
Subject: [PATCH] add partial status support

---
 .../service/apps/scheduler/AbstractOmAppJobListener.java     | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/apps/scheduler/AbstractOmAppJobListener.java b/openmetadata-service/src/main/java/org/openmetadata/service/apps/scheduler/AbstractOmAppJobListener.java
index 8e23e7be7f40..939b9270f6e2 100644
--- a/openmetadata-service/src/main/java/org/openmetadata/service/apps/scheduler/AbstractOmAppJobListener.java
+++ b/openmetadata-service/src/main/java/org/openmetadata/service/apps/scheduler/AbstractOmAppJobListener.java
@@ -94,7 +94,10 @@ public void jobWasExecuted(
     if (jobException == null
         && !(runRecord.getStatus() == AppRunRecord.Status.FAILED
             || runRecord.getStatus() == AppRunRecord.Status.ACTIVE_ERROR)) {
-      runRecord.withStatus(AppRunRecord.Status.SUCCESS);
+      runRecord.withStatus(
+          runRecord.getStatus() == AppRunRecord.Status.PARTIAL_SUCCESS
+              ? AppRunRecord.Status.PARTIAL_SUCCESS
+              : AppRunRecord.Status.SUCCESS);
       SuccessContext context = new SuccessContext();
       if (runRecord.getSuccessContext() != null) {
         context = runRecord.getSuccessContext();