Struct DistributedLeafExec 

pub struct DistributedLeafExec { /* private fields */ }

Represents a leaf node ready to be distributed across N tasks, where the variant of the node belonging to each task is stored in a Vec of N positions.

While sending this plan over the wire to a remote worker, only the appropriate variant is sent.

This ExecutionPlan implementation is typically returned by [crate::TaskEstimator::scale_up_leaf_node], which will be called for scaling up a node for distribution. The process typically looks like this:

1. The distributed planner calls [crate::TaskEstimator::scale_up_leaf_node] providing a leaf node and the amount of tasks in which it should be distributed:

┌──────────────┐
│DataSourceExec│ + 3 tasks
└──────────────┘

2. The [crate::TaskEstimator] implementation, either user provided or a default one, returns a DistributedLeafExec adhering to this task count:

┌────────────────────────────────────────────────┐
│              DistributedLeafExec               │
│                                                │
│┌──────────────┐┌──────────────┐┌──────────────┐│
││DataSourceExec││DataSourceExec││DataSourceExec││
││  for task 0  ││  for task 1  ││  for task 2  ││
│└──────────────┘└──────────────┘└──────────────┘│
└────────────────────────────────────────────────┘

3. The crate::DistributedExec node, upon being executed, will send the different variants of the leaf node to the respective workers, instead of sending the full DistributedLeafExec:

┌──────────────────┐┌──────────────────┐┌──────────────────┐
│     Worker 0     ││     Worker 1     ││     Worker 2     │
│                  ││                  ││                  │
│       ...        ││       ...        ││       ...        │
│                  ││                  ││                  │
│ ┌──────────────┐ ││ ┌──────────────┐ ││ ┌──────────────┐ │
│ │   SomeExec   │ ││ │   SomeExec   │ ││ │   SomeExec   │ │
│ │              │ ││ │              │ ││ │              │ │
│ └──────────────┘ ││ └──────────────┘ ││ └──────────────┘ │
│ ┌──────────────┐ ││ ┌──────────────┐ ││ ┌──────────────┐ │
│ │DataSourceExec│ ││ │DataSourceExec│ ││ │DataSourceExec│ │
│ │  for task 0  │ ││ │  for task 1  │ ││ │  for task 2  │ │
│ └──────────────┘ ││ └──────────────┘ ││ └──────────────┘ │
└──────────────────┘└──────────────────┘└──────────────────┘

This way, the different workers get to execute different versions of the same plan, each handling its own range of non-overlapping data.

Note that every variant in variants must agree on the same schema and partition count with every other variant, but variants are not required to match the partition count of the original leaf plan. Variants can scale their partition count higher or lower as needed for worker execution.

Implementations

impl DistributedLeafExec

pub fn try_new( original: Arc<dyn ExecutionPlan>, variants: impl IntoIterator<Item = Arc<dyn ExecutionPlan>>, ) -> Result<Self>

Builds a new DistributedLeafExec based on the provided original plan and its per-task variants.

Every variant must expose the same schema and partition count as every other variant. Variants do not need to match the partition count of the original plan.

pub fn original(&self) -> &Arc<dyn ExecutionPlan>

The plan this leaf was built from (the leaf passed to [crate::TaskEstimator::scale_up_leaf_node]). Useful for recognising which DistributedLeafExec you are looking at — e.g. by downcasting it to your own leaf type — before inspecting its DistributedLeafExec::variants.

pub fn variants(&self) -> &[Arc<dyn ExecutionPlan>]

The per-task variants, in task order: variants()[i] is the plan sent to task i. Useful for inspecting per-task information (e.g. data locality) when routing tasks to workers via [crate::TaskEstimator::route_tasks].

Trait Implementations

impl Debug for DistributedLeafExec

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl DisplayAs for DistributedLeafExec

fn fmt_as(&self, t: DisplayFormatType, f: &mut Formatter<'_>) -> Result

Format according to DisplayFormatType, used when verbose representation looks different from the default one

impl ExecutionPlan for DistributedLeafExec

fn name(&self) -> &str

Short name for the ExecutionPlan, such as ‘DataSourceExec’.

fn properties(&self) -> &Arc<PlanProperties>

Return properties of the output of the ExecutionPlan, such as output ordering(s), partitioning information etc.

fn children(&self) -> Vec<&Arc<dyn ExecutionPlan>>

Get a list of children ExecutionPlans that act as inputs to this plan. The returned list will be empty for leaf nodes such as scans, will contain a single value for unary nodes, or two values for binary nodes (such as joins).

fn apply_expressions( &self, f: &mut dyn FnMut(&Arc<dyn PhysicalExpr>) -> Result<TreeNodeRecursion>, ) -> Result<TreeNodeRecursion>

Apply a closure f to each root expression that this node owns and uses during execution, either by evaluating it or updating it dynamically.

fn with_new_children( self: Arc<Self>, children: Vec<Arc<dyn ExecutionPlan>>, ) -> Result<Arc<dyn ExecutionPlan>>

👎Deprecated since 55.0.0:

Use ExecutionPlan::replace_children with ReplaceChildrenOptions

Deprecated.

fn execute( &self, partition: usize, context: Arc<TaskContext>, ) -> Result<SendableRecordBatchStream>

Begin execution of partition, returning a Stream of RecordBatches.

fn metrics(&self) -> Option<MetricsSet>

Return a snapshot of the set of Metrics for this ExecutionPlan. If no Metrics are available, return None.

fn statistics_from_inputs( &self, _input_stats: &[Arc<Statistics>], args: &StatisticsArgs, ) -> Result<Arc<Statistics>>

Returns statistics for a specific partition of this ExecutionPlan node, given pre-computed child statistics.

fn static_name() -> &'static str

where Self: Sized,

Short name for the ExecutionPlan, such as ‘DataSourceExec’. Like name but can be called without an instance.

fn downcast_delegate(&self) -> Option<&(dyn ExecutionPlan + 'static)>

Returns the plan that provides this plan’s public ExecutionPlan downcast identity.

fn schema(&self) -> Arc<Schema>

Get the schema for this execution plan

fn check_invariants(&self, check: InvariantLevel) -> Result<(), DataFusionError>

Returns an error if this individual node does not conform to its invariants. These invariants are typically only checked in debug mode.

fn dynamic_expressions_produced(&self) -> Vec<Arc<dyn PhysicalExpr>>

Returns the dynamic expressions produced by this plan node.

fn required_input_distribution(&self) -> Vec<Distribution>

👎Deprecated since 55.0.0:

Use input_distribution_requirements

Specifies simple per-child input distribution requirements.

fn input_distribution_requirements(&self) -> InputDistributionRequirements

Specifies the input distribution requirements for this plan.

fn required_input_ordering(&self) -> Vec<Option<OrderingRequirements>>

Specifies the ordering required for all of the children of this ExecutionPlan.

fn maintains_input_order(&self) -> Vec<bool>

Returns false if this ExecutionPlan’s implementation may reorder rows within or between partitions.

fn benefits_from_input_partitioning(&self) -> Vec<bool>

Specifies whether the ExecutionPlan benefits from increased parallelization at its input for each child.

fn replace_children( self: Arc<Self>, children: Vec<Arc<dyn ExecutionPlan>>, options: ReplaceChildrenOptions, ) -> Result<Arc<dyn ExecutionPlan>, DataFusionError>

Returns a clone of the existing plan with the children replaced, skipping recomputation of plan properties when the options indicate the new children’s properties are unchanged.

fn with_new_children_and_same_properties( self: Arc<Self>, children: Vec<Arc<dyn ExecutionPlan>>, ) -> Result<Arc<dyn ExecutionPlan>, DataFusionError>

👎Deprecated since 55.0.0:

Use ExecutionPlan::replace_children with ReplaceChildrenOptions

Deprecated. Implement ExecutionPlan::replace_children instead.

fn reset_state( self: Arc<Self>, ) -> Result<Arc<dyn ExecutionPlan>, DataFusionError>

Reset any internal state within this ExecutionPlan.

fn repartitioned( &self, _target_partitions: usize, _config: &ConfigOptions, ) -> Result<Option<Arc<dyn ExecutionPlan>>, DataFusionError>

If supported, attempt to increase the partitioning of this ExecutionPlan to produce target_partitions partitions.

fn partition_statistics( &self, partition: Option<usize>, ) -> Result<Arc<Statistics>, DataFusionError>

👎Deprecated since 55.0.0:

Use StatisticsContext::compute instead

Returns statistics for a specific partition of this ExecutionPlan node.

fn child_stats_requests(&self, _partition: Option<usize>) -> Vec<ChildStats>

Returns, per child, which statistics the StatisticsContext should resolve before calling Self::statistics_from_inputs.

fn supports_limit_pushdown(&self) -> bool

Returns true if a limit can be safely pushed down through this ExecutionPlan node.

fn with_fetch(&self, _limit: Option<usize>) -> Option<Arc<dyn ExecutionPlan>>

Returns a fetching variant of this ExecutionPlan node, if it supports fetch limits. Returns None otherwise.

fn fetch(&self) -> Option<usize>

Gets the fetch count for the operator, None means there is no fetch.

fn cardinality_effect(&self) -> CardinalityEffect

Gets the effect on cardinality, if known

fn try_swapping_with_projection( &self, _projection: &ProjectionExec, ) -> Result<Option<Arc<dyn ExecutionPlan>>, DataFusionError>

Attempts to push down the given projection into the input of this ExecutionPlan.

fn gather_filters_for_pushdown( &self, _phase: FilterPushdownPhase, parent_filters: Vec<Arc<dyn PhysicalExpr>>, _config: &ConfigOptions, ) -> Result<FilterDescription, DataFusionError>

Collect filters that this node can push down to its children. Filters that are being pushed down from parents are passed in, and the node may generate additional filters to push down. For example, given the plan FilterExec -> HashJoinExec -> DataSourceExec, what will happen is that we recurse down the plan calling ExecutionPlan::gather_filters_for_pushdown:

fn handle_child_pushdown_result( &self, _phase: FilterPushdownPhase, child_pushdown_result: ChildPushdownResult, _config: &ConfigOptions, ) -> Result<FilterPushdownPropagation<Arc<dyn ExecutionPlan>>, DataFusionError>

Handle the result of a child pushdown.

fn with_new_state( &self, _state: Arc<dyn Any + Send + Sync>, ) -> Option<Arc<dyn ExecutionPlan>>

Injects arbitrary run-time state into this execution plan, returning a new plan instance that incorporates that state if it is relevant to the concrete node implementation.

fn try_pushdown_sort( &self, _order: &[PhysicalSortExpr], ) -> Result<SortOrderPushdownResult<Arc<dyn ExecutionPlan>>, DataFusionError>

Try to push down sort ordering requirements to this node.

fn with_preserve_order( &self, _preserve_order: bool, ) -> Option<Arc<dyn ExecutionPlan>>

Returns a variant of this ExecutionPlan that is aware of order-sensitivity.

fn try_to_proto( &self, _ctx: &ExecutionPlanEncodeCtx<'_>, ) -> Result<Option<PhysicalPlanNode>, DataFusionError>

Serialize this plan to its protobuf representation, if it knows how.