Method new_constraint_from_table_constraints

datafusion/sql/src/statement.rs:1899–1970

Convert each [TableConstraint] to corresponding [Constraint]

(
        &self,
        constraints: &[TableConstraint],
        df_schema: &DFSchemaRef,
    )

    /// Convert each [TableConstraint] to corresponding [Constraint]
    pub fn new_constraint_from_table_constraints(
        &self,
        constraints: &[TableConstraint],
        df_schema: &DFSchemaRef,
    ) -> Result<Constraints> {
        let constraints = constraints
            .iter()
            .map(|c: &TableConstraint| match c {
                TableConstraint::Unique(UniqueConstraint {
                    name,
                    index_name: _,
                    index_type_display: _,
                    index_type: _,
                    columns,
                    index_options: _,
                    characteristics: _,
                    nulls_distinct: _,
                }) => {
                    let constraint_name = match &name {
                        Some(name) => &format!("unique constraint with name '{name}'"),
                        None => "unique constraint",
                    };
                    // Get unique constraint indices in the schema
                    let indices = self.get_constraint_column_indices(
                        df_schema,
                        columns,
                        constraint_name,
                    )?;
                    Ok(Constraint::Unique(indices))
                }
                TableConstraint::PrimaryKey(PrimaryKeyConstraint {
                    name: _,
                    index_name: _,
                    index_type: _,
                    columns,
                    index_options: _,
                    characteristics: _,
                }) => {
                    // Get primary key indices in the schema
                    let indices = self.get_constraint_column_indices(
                        df_schema,
                        columns,
                        "primary key",
                    )?;
                    Ok(Constraint::PrimaryKey(indices))
                }
                TableConstraint::ForeignKey { .. } => {
                    _plan_err!("Foreign key constraints are not currently supported")
                }
                TableConstraint::Check { .. } => {
                    _plan_err!("Check constraints are not currently supported")
                }
                TableConstraint::Index { .. } => {
                    _plan_err!("Indexes are not currently supported")
                }
                TableConstraint::FulltextOrSpatial { .. } => {
                    _plan_err!("Indexes are not currently supported")
                }