Method plan_relation

crates/integrations/datafusion/src/relation_planner.rs:58–104

(
        &self,
        relation: TableFactor,
        context: &mut dyn RelationPlannerContext,
    )

impl RelationPlanner for PaimonRelationPlanner {
    fn plan_relation(
        &self,
        relation: TableFactor,
        context: &mut dyn RelationPlannerContext,
    ) -> DFResult<RelationPlanning> {
        // Only handle Table factors with a version clause.
        let TableFactor::Table {
            ref name,
            ref version,
            ..
        } = relation
        else {
            return Ok(RelationPlanning::Original(Box::new(relation)));
        };

        let extra_options = match version {
            Some(TableVersion::VersionAsOf(expr)) => resolve_version_as_of(expr)?,
            Some(TableVersion::TimestampAsOf(expr)) => resolve_timestamp_as_of(expr)?,
            _ => return Ok(RelationPlanning::Original(Box::new(relation))),
        };

        // Resolve the table reference.
        let table_ref = object_name_to_table_reference(name, context)?;
        let source = context
            .context_provider()
            .get_table_source(table_ref.clone())?;
        let provider = source_as_provider(&source)?;

        // Check if this is a Paimon table.
        let Some(paimon_provider) = provider.as_any().downcast_ref::<PaimonTableProvider>() else {
            return Ok(RelationPlanning::Original(Box::new(relation)));
        };

        let new_table = paimon_provider.table().copy_with_options(extra_options);
        let new_provider = PaimonTableProvider::try_new(new_table)?;
        let new_source = provider_as_source(Arc::new(new_provider));

        // Destructure to get alias.
        let TableFactor::Table { alias, .. } = relation else {
            unreachable!()
        };

        let plan = LogicalPlanBuilder::scan(table_ref, new_source, None)?.build()?;
        Ok(RelationPlanning::Planned(Box::new(PlannedRelation::new(
            plan, alias,
        ))))
    }
}

/// Convert a sqlparser `ObjectName` to a DataFusion `TableReference`.