fix JDC stack overflow

plebhash · Jan 18, 2024 · 9e26716 · 9e26716
1 parent 3f07a6a
commit 9e26716
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/roles/jd-client/src/job_declarator/mod.rs b/roles/jd-client/src/job_declarator/mod.rs
@@ -166,7 +166,14 @@ impl JobDeclarator {
                         })
                         .unwrap();
                 }
-                tokio::task::yield_now().await;
+
+                // we block on the receiver to avoid infinite recursion
+                let receiver = self_mutex.safe_lock(|s| s.receiver.clone()).unwrap();
+                while receiver.len() == 0 {
+                    tokio::time::sleep(tokio::time::Duration::from_millis(1)).await;
+                    tokio::task::yield_now().await;
+                }
+
                 Self::get_last_token(self_mutex).await
             }
             1 => {