Method build

crates/integrations/datafusion/src/table/mod.rs:105–143

Build a [`PaimonTableScan`] from the configured parameters.

(self)

impl PaimonScanBuilder<'_> {
    /// Build a [`PaimonTableScan`] from the configured parameters.
    pub(crate) fn build(self) -> DFResult<Arc<dyn ExecutionPlan>> {
        let (projected_schema, projected_columns) = if let Some(indices) = self.projection {
            let fields: Vec<Field> = indices
                .iter()
                .map(|&i| self.schema.field(i).clone())
                .collect();
            let column_names: Vec<String> = fields.iter().map(|f| f.name().clone()).collect();
            (Arc::new(Schema::new(fields)), Some(column_names))
        } else {
            let column_names: Vec<String> = self
                .schema
                .fields()
                .iter()
                .map(|f| f.name().clone())
                .collect();
            (self.schema.clone(), Some(column_names))
        };

        let splits = self.plan.splits().to_vec();
        let planned_partitions: Vec<Arc<[_]>> = if splits.is_empty() {
            vec![Arc::from(Vec::new())]
        } else {
            let num_partitions = splits.len().min(self.target_partitions.max(1));
            bucket_round_robin(splits, num_partitions)
                .into_iter()
                .map(Arc::from)
                .collect()
        };

        Ok(Arc::new(PaimonTableScan::new(
            projected_schema,
            self.table.clone(),
            projected_columns,
            self.pushed_predicate,
            planned_partitions,
            self.limit,
            self.filter_exact,
        )))
    }
}

#[async_trait]