Skip to content

Commit a274f1e

Browse files
authored
Minor: include output partition count of RepartitionExec to tree explain (#15717)
* add output partition count to RepartitionExec's tree explain * Review: better display
1 parent ac7156f commit a274f1e

File tree

2 files changed

+85
-78
lines changed

2 files changed

+85
-78
lines changed

datafusion/physical-plan/src/repartition/mod.rs

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -508,11 +508,18 @@ impl DisplayAs for RepartitionExec {
508508
}
509509
DisplayFormatType::TreeRender => {
510510
writeln!(f, "partitioning_scheme={}", self.partitioning(),)?;
511+
512+
let input_partition_count =
513+
self.input.output_partitioning().partition_count();
514+
let output_partition_count = self.partitioning().partition_count();
515+
let input_to_output_partition_str =
516+
format!("{} -> {}", input_partition_count, output_partition_count);
511517
writeln!(
512518
f,
513-
"input_partition_count={}",
514-
self.input.output_partitioning().partition_count()
519+
"partition_count(in->out)={}",
520+
input_to_output_partition_str
515521
)?;
522+
516523
if self.preserve_order {
517524
writeln!(f, "preserve_order={}", self.preserve_order)?;
518525
}

0 commit comments

Comments
 (0)