controller: Fix bug (don't exit upon arrival of new worker)

1 files changed, 3 insertions, 4 deletions

diff --git a/controller/src/lib.rs b/controller/src/lib.rs
index 60b3639..0918542 100644
--- a/controller/src/lib.rs
+++ b/controller/src/lib.rs
@@ -404,15 +404,14 @@ fn thread_entry(
 
                     if let Some(worker) = workers_locked.get_mut(&worker_id) {
                         if worker.loaded_core != current_core_id {
-                            let current_core = current_core.unwrap().clone();
+                            let current_core = current_core.as_ref().unwrap().clone();
                             worker_set_new_core(
                                 worker, current_core_id, current_core, workers.clone(),
                                 collector_sink.clone()
                             ).await;
-                            return;
+                        } else {
+                            free_workers.push_back(worker_id);
                         }
-
-                        free_workers.push_back(worker_id);
                     }
                 }
             }