refactor: Extract a function for creating the df planner

pull/24376/head
Carol (Nichols || Goulding) 2023-03-29 17:31:43 -04:00
parent 7bbf0fcd79
commit 07c2c768e9
No known key found for this signature in database
GPG Key ID: E907EE5A736F87D4
1 changed files with 9 additions and 5 deletions

View File

@ -23,7 +23,7 @@ use super::{
df_plan_exec::{
dedicated::DedicatedDataFusionPlanExec, noop::NoopDataFusionPlanExec, DataFusionPlanExec,
},
df_planner::planner_v1::V1DataFusionPlanner,
df_planner::{planner_v1::V1DataFusionPlanner, DataFusionPlanner},
divide_initial::multiple_branches::MultipleBranchesDivideInitial,
file_classifier::{
logging::LoggingFileClassifierWrapper, split_based::SplitBasedFileClassifier,
@ -144,10 +144,7 @@ pub fn hardcoded_components(config: &Config) -> Arc<Components> {
partition_done_sink,
commit,
ir_planner: make_ir_planner(config),
df_planner: Arc::new(V1DataFusionPlanner::new(
config.parquet_store_scratchpad.clone(),
Arc::clone(&config.exec),
)),
df_planner: make_df_planner(config),
df_plan_exec,
parquet_files_sink,
round_split: Arc::new(ManyFilesRoundSplit::new()),
@ -398,3 +395,10 @@ fn make_ir_planner(config: &Config) -> Arc<dyn IRPlanner> {
config.split_percentage,
)))
}
fn make_df_planner(config: &Config) -> Arc<dyn DataFusionPlanner> {
Arc::new(V1DataFusionPlanner::new(
config.parquet_store_scratchpad.clone(),
Arc::clone(&config.exec),
))
}