Limit workflow timeout to non-negative

Signed-off-by: Junwei Dai <[email protected]>
junweid62 · Jan 13, 2025 · ffdc699 · ffdc699
1 parent 84f829c
commit ffdc699
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/main/java/org/opensearch/flowframework/util/WorkflowTimeoutUtility.java b/src/main/java/org/opensearch/flowframework/util/WorkflowTimeoutUtility.java
@@ -31,8 +31,7 @@
 public class WorkflowTimeoutUtility {
 
     private static final Logger logger = LogManager.getLogger(WorkflowTimeoutUtility.class);
-    private static final TimeValue MAX_TIMEOUT_MILLIS = TimeValue.timeValueSeconds(300);
-    private static final TimeValue MIN_TIMEOUT_MILLIS = TimeValue.timeValueSeconds(1);
+    private static final TimeValue MIN_TIMEOUT_MILLIS = TimeValue.timeValueSeconds(0);
 
     /**
      * Schedules a timeout task for a workflow execution.
@@ -53,8 +52,8 @@ public static ActionListener<WorkflowResponse> scheduleTimeoutHandler(
         long timeout,
         AtomicBoolean isResponseSent
     ) {
-        long adjustedTimeout = Math.min(timeout, MAX_TIMEOUT_MILLIS.millis());
-        adjustedTimeout = Math.max(adjustedTimeout, MIN_TIMEOUT_MILLIS.millis());
+        // Ensure timeout is within the valid range (non-negative)
+        long adjustedTimeout = Math.max(timeout, MIN_TIMEOUT_MILLIS.millis());
         Scheduler.ScheduledCancellable scheduledCancellable = threadPool.schedule(
             new WorkflowTimeoutListener(client, workflowId, listener, isResponseSent),
             TimeValue.timeValueMillis(adjustedTimeout),