Task retrying if the context load failed

apache · Nov 11, 2018 · 274c73f · 274c73f
1 parent eb919bc
commit 274c73f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/...helix/helix-spectator/src/main/java/org/apache/airavata/helix/impl/task/AiravataTask.java b/...helix/helix-spectator/src/main/java/org/apache/airavata/helix/impl/task/AiravataTask.java
@@ -347,8 +347,10 @@ public TaskResult onRun(TaskHelper helper) {
                 publishTaskState(TaskState.EXECUTING);
             }
             return onRun(helper, getTaskContext());
+        } catch (TaskOnFailException e) {
+            return onFail("Captured a task fail : " + e.getReason(), e.isCritical(), e);
         } catch (Exception e) {
-            return onFail("Unknown error while running task " + getTaskId(), true, e);
+            return onFail("Unknown error while running task " + getTaskId(), false, e);
         } finally {
             MDC.clear();
         }
@@ -422,7 +424,7 @@ private void loadContext() throws TaskOnFailException {
 
         } catch (Exception e) {
             logger.error("Error occurred while initializing the task " + getTaskId() + " of experiment " + getExperimentId(), e);
-            throw new TaskOnFailException("Error occurred while initializing the task " + getTaskId() + " of experiment " + getExperimentId(), true, e);
+            throw new TaskOnFailException("Error occurred while initializing the task " + getTaskId() + " of experiment " + getExperimentId(), false, e);
         }
     }