vespertide-query 0.3.0

Converts migration actions into SQL statements with bind parameters
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

use sea_query::{Alias, ForeignKey};

use vespertide_core::TableConstraint;

use crate::sql::helpers::quote_ident;
use crate::sql::types::BuiltQuery;
use crate::sql::{DatabaseBackend, RawSql};

pub fn build_remove_constraint(table: &str, constraint: &TableConstraint) -> Vec<BuiltQuery> {
    match constraint {
        TableConstraint::PrimaryKey { .. } => {
            let pg_table = quote_ident(table, DatabaseBackend::Postgres);
            let pg_pkey = quote_ident(&format!("{table}_pkey"), DatabaseBackend::Postgres);
            vec![BuiltQuery::Raw(RawSql::uniform(format!(
                "ALTER TABLE {pg_table} DROP CONSTRAINT {pg_pkey}"
            )))]
        }
        TableConstraint::Unique { name, columns, .. } => {
            let constraint_name =
                vespertide_naming::build_unique_constraint_name(table, columns, name.as_deref());
            let pg_constraint = quote_ident(&constraint_name, DatabaseBackend::Postgres);
            vec![BuiltQuery::Raw(RawSql::uniform(format!(
                "DROP INDEX {pg_constraint}"
            )))]
        }
        TableConstraint::ForeignKey { name, columns, .. } => {
            let constraint_name =
                vespertide_naming::build_foreign_key_name(table, columns, name.as_deref());
            let fk_drop = ForeignKey::drop()
                .name(&constraint_name)
                .table(Alias::new(table))
                .to_owned();
            vec![BuiltQuery::DropForeignKey(Box::new(fk_drop))]
        }
        TableConstraint::Index { name, columns } => {
            let index_name = vespertide_naming::build_index_name(table, columns, name.as_deref());
            vec![super::build_drop_index_query(table, &index_name)]
        }
        TableConstraint::Check { name, .. } => {
            let pg_table = quote_ident(table, DatabaseBackend::Postgres);
            let pg_name = quote_ident(name, DatabaseBackend::Postgres);
            vec![BuiltQuery::Raw(RawSql::uniform(format!(
                "ALTER TABLE {pg_table} DROP CONSTRAINT {pg_name}"
            )))]
        }
        _ => unreachable!("TableConstraint is #[non_exhaustive]; all variants are matched above"),
    }
}