From 249d0e114fcabb4048f628c31822f7250c574061 Mon Sep 17 00:00:00 2001 From: Athul T R Date: Sat, 13 Jul 2024 19:43:19 +0530 Subject: [PATCH] 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 (athultr1997@gmail.com) --- .../core/src/execution_plans/distributed_query.rs | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/ballista/core/src/execution_plans/distributed_query.rs b/ballista/core/src/execution_plans/distributed_query.rs index 2f12a6740..4e7a1c2be 100644 --- 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 ExecutionPlan for DistributedQueryExec { 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::>(), optional_session_id: Some(OptionalSessionId::SessionId( self.session_id.clone(), )),