Remove executor metadata

systemxlabs · Aug 21, 2024 · 69b61b2 · 69b61b2
1 parent b28b84f
commit 69b61b2
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ballista/scheduler/src/cluster/kv.rs b/ballista/scheduler/src/cluster/kv.rs
@@ -371,7 +371,10 @@ impl<S: KeyValueStore, T: 'static + AsLogicalPlan, U: 'static + AsExecutionPlan>
             .await?;
         self.executor_heartbeats.remove(executor_id);
 
-        // TODO Check the Executor reservation logic for push-based scheduling
+        self.store
+            .put(Keyspace::Executors, executor_id.to_string(), vec![])
+            .await?;
+        self.executors.remove(executor_id);
 
         Ok(())
     }