fix

risingwavelabs · Jan 5, 2024 · 0b055d6 · 0b055d6
1 parent 3afd98a
commit 0b055d6
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/frontend/src/scheduler/distributed/stage.rs b/src/frontend/src/scheduler/distributed/stage.rs
@@ -705,7 +705,9 @@ impl StageRunner {
                 .worker_node_manager
                 .manager
                 .get_workers_by_parallel_unit_ids(&parallel_unit_ids)?;
-
+            if candidates.is_empty() {
+                return Err(SchedulerError::EmptyWorkerNodes);
+            }
             return Ok(Some(
                 candidates
                     .get(self.stage.session_id.0 as usize % candidates.len())
@@ -736,6 +738,9 @@ impl StageRunner {
                 .worker_node_manager
                 .manager
                 .get_workers_by_parallel_unit_ids(&[pu])?;
+            if candidates.is_empty() {
+                return Err(SchedulerError::EmptyWorkerNodes);
+            }
             Ok(Some(candidates[0].clone()))
         } else {
             Ok(None)

diff --git a/src/frontend/src/scheduler/local.rs b/src/frontend/src/scheduler/local.rs
@@ -580,6 +580,9 @@ impl LocalQueryExecution {
                     .worker_node_manager
                     .manager
                     .get_workers_by_parallel_unit_ids(&parallel_unit_ids)?;
+                if candidates.is_empty() {
+                    return Err(SchedulerError::EmptyWorkerNodes);
+                }
                 candidates
                     .get(stage.session_id.0 as usize % candidates.len())
                     .context("no available worker node for dml")?