Fix executor metadata decode bug (#1004)

apache · Jun 12, 2024 · dcaa1cf · dcaa1cf
1 parent e6829c9
commit dcaa1cf
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/ballista/scheduler/src/cluster/kv.rs b/ballista/scheduler/src/cluster/kv.rs
@@ -429,6 +429,12 @@ impl<S: KeyValueStore, T: 'static + AsLogicalPlan, U: 'static + AsExecutionPlan>
             metadata.value().clone()
         } else {
             let value = self.store.get(Keyspace::Executors, executor_id).await?;
+            if value.is_empty() {
+                return Err(BallistaError::Internal(format!(
+                    "Executor {} not registered",
+                    executor_id
+                )));
+            }
             let decoded =
                 decode_into::<protobuf::ExecutorMetadata, ExecutorMetadata>(&value)?;
             self.executors