Fixes Setting Job Name Not Reflected in Ballista UI (#1039)

* Fixes the issue of job name not being displayed in the Ballista UI after being set. Fixes: #1019 Authored-by: athultr1997 ([email protected])
apache · Jul 13, 2024 · 249d0e1 · 249d0e1
1 parent d1aff3e
commit 249d0e1
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/ballista/core/src/execution_plans/distributed_query.rs b/ballista/core/src/execution_plans/distributed_query.rs
@@ -21,7 +21,7 @@ use crate::serde::protobuf::execute_query_params::OptionalSessionId;
 use crate::serde::protobuf::{
     execute_query_params::Query, execute_query_result, job_status,
     scheduler_grpc_client::SchedulerGrpcClient, ExecuteQueryParams, GetJobStatusParams,
-    GetJobStatusResult, PartitionLocation,
+    GetJobStatusResult, KeyValuePair, PartitionLocation,
 };
 use crate::utils::create_grpc_client_connection;
 use datafusion::arrow::datatypes::SchemaRef;
@@ -208,7 +208,15 @@ impl<T: 'static + AsLogicalPlan> ExecutionPlan for DistributedQueryExec<T> {
 
         let query = ExecuteQueryParams {
             query: Some(Query::LogicalPlan(buf)),
-            settings: vec![],
+            settings: self
+                .config
+                .settings()
+                .iter()
+                .map(|(k, v)| KeyValuePair {
+                    key: k.to_owned(),
+                    value: v.to_owned(),
+                })
+                .collect::<Vec<_>>(),
             optional_session_id: Some(OptionalSessionId::SessionId(
                 self.session_id.clone(),
             )),