Struct PlannerContext 

pub struct PlannerContext {
    prepare_param_data_types: Arc<Vec<FieldRef>>,
    ctes: HashMap<String, Arc<LogicalPlan>>,
    outer_query_schema: Option<DFSchemaRef>,
    outer_from_schema: Option<DFSchemaRef>,
    create_table_schema: Option<DFSchemaRef>,
    lambdas_parameters: HashSet<String>,
}

Struct to store the states used by the Planner. The Planner will leverage the states to resolve CTEs, Views, subqueries and PREPARE statements. The states include Common Table Expression (CTE) provided with WITH clause and Parameter Data Types provided with PREPARE statement and the query schema of the outer query plan.

Cloning

Only the ctes are truly cloned when the PlannerContext is cloned. This helps resolve scoping issues of CTEs. By using cloning, a subquery can inherit CTEs from the outer query and can also define its own private CTEs without affecting the outer query.

Fields

prepare_param_data_types: Arc<Vec<FieldRef>>

Data types for numbered parameters ($1, $2, etc), if supplied in PREPARE statement

ctes: HashMap<String, Arc<LogicalPlan>>

Map of CTE name to logical plan of the WITH clause. Use Arc<LogicalPlan> to allow cheap cloning

outer_query_schema: Option<DFSchemaRef>

The query schema of the outer query plan, used to resolve the columns in subquery

outer_from_schema: Option<DFSchemaRef>

The joined schemas of all FROM clauses planned so far. When planning LATERAL FROM clauses, this should become a suffix of the outer_query_schema.

create_table_schema: Option<DFSchemaRef>

The query schema defined by the table

lambdas_parameters: HashSet<String>

The lambda introduced columns names

Implementations

impl PlannerContext

pub fn new() -> Self

Create an empty PlannerContext

pub fn with_prepare_param_data_types( self, prepare_param_data_types: Vec<FieldRef>, ) -> Self

Update the PlannerContext with provided prepare_param_data_types

pub fn outer_query_schema(&self) -> Option<&DFSchema>

pub fn set_outer_query_schema( &mut self, schema: Option<DFSchemaRef>, ) -> Option<DFSchemaRef>

Sets the outer query schema, returning the existing one, if any

pub fn set_table_schema( &mut self, schema: Option<DFSchemaRef>, ) -> Option<DFSchemaRef>

pub fn table_schema(&self) -> Option<DFSchemaRef>

pub fn outer_from_schema(&self) -> Option<Arc<DFSchema>>

pub fn set_outer_from_schema( &mut self, schema: Option<DFSchemaRef>, ) -> Option<DFSchemaRef>

Sets the outer FROM schema, returning the existing one, if any

pub fn extend_outer_from_schema(&mut self, schema: &DFSchemaRef) -> Result<()>

Extends the FROM schema, returning the existing one, if any

pub fn prepare_param_data_types(&self) -> &[FieldRef]

Return the types of parameters ($1, $2, etc) if known

pub fn contains_cte(&self, cte_name: &str) -> bool

Returns true if there is a Common Table Expression (CTE) / Subquery for the specified name

pub fn insert_cte(&mut self, cte_name: impl Into<String>, plan: LogicalPlan)

Inserts a LogicalPlan for the Common Table Expression (CTE) / Subquery for the specified name

pub fn get_cte(&self, cte_name: &str) -> Option<&LogicalPlan>

Return a plan for the Common Table Expression (CTE) / Subquery for the specified name

pub fn lambdas_parameters(&self) -> &HashSet<String>

pub fn with_lambda_parameters( self, arguments: impl IntoIterator<Item = String>, ) -> Self

pub(crate) fn remove_cte(&mut self, cte_name: &str)

Remove the plan of CTE / Subquery for the specified name

Trait Implementations

impl Clone for PlannerContext

fn clone(&self) -> PlannerContext

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for PlannerContext

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

Formats the value using the given formatter.

impl Default for PlannerContext

fn default() -> Self

Returns the “default value” for a type.