From 7e0c775071fa0adb7e20c409629913bef10f2abd Mon Sep 17 00:00:00 2001
From: LiaCastaneda <lia.castaneda@datadoghq.com>
Date: Mon, 14 Jul 2025 14:56:57 +0200
Subject: [PATCH] Add guard to avoid panic

---
 src/planning.rs | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/planning.rs b/src/planning.rs
index 112dff6e4..8e04213d1 100644
--- a/src/planning.rs
+++ b/src/planning.rs
@@ -451,6 +451,10 @@ pub async fn distribute_stages(
         .collect();
 
     for attempt in 0..3 {
+        if workers.is_empty() {
+            return Err(anyhow!("No workers available to distribute stages").into());
+        }
+
         // all stages to workers
         let (task_datas, final_addrs) =
             assign_to_workers(query_id, &stages, workers.values().collect())?;