vespertide-query 0.2.1

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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71

use sea_query::Alias;

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);
            let mysql_table = quote_ident(table, DatabaseBackend::MySql);
            let pg_sql = format!("ALTER TABLE {pg_table} DROP CONSTRAINT {pg_pkey}");
            vec![BuiltQuery::Raw(RawSql::per_backend(
                pg_sql.clone(),
                format!("ALTER TABLE {mysql_table} DROP PRIMARY KEY"),
                pg_sql,
            ))]
        }
        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);
            let mysql_table = quote_ident(table, DatabaseBackend::MySql);
            let mysql_constraint = quote_ident(&constraint_name, DatabaseBackend::MySql);
            let pg_sql = format!("DROP INDEX {pg_constraint}");
            vec![BuiltQuery::Raw(RawSql::per_backend(
                pg_sql.clone(),
                format!("ALTER TABLE {mysql_table} DROP INDEX {mysql_constraint}"),
                pg_sql,
            ))]
        }
        TableConstraint::ForeignKey { name, columns, .. } => {
            let constraint_name =
                vespertide_naming::build_foreign_key_name(table, columns, name.as_deref());
            let pg_table = quote_ident(table, DatabaseBackend::Postgres);
            let pg_constraint = quote_ident(&constraint_name, DatabaseBackend::Postgres);
            let mysql_table = quote_ident(table, DatabaseBackend::MySql);
            let mysql_constraint = quote_ident(&constraint_name, DatabaseBackend::MySql);
            let pg_sql = format!("ALTER TABLE {pg_table} DROP CONSTRAINT {pg_constraint}");
            vec![BuiltQuery::Raw(RawSql::per_backend(
                pg_sql.clone(),
                format!("ALTER TABLE {mysql_table} DROP FOREIGN KEY {mysql_constraint}"),
                pg_sql,
            ))]
        }
        TableConstraint::Index { name, columns } => {
            let index_name = vespertide_naming::build_index_name(table, columns, name.as_deref());
            let idx_drop = sea_query::Index::drop()
                .table(Alias::new(table))
                .name(&index_name)
                .to_owned();
            vec![BuiltQuery::DropIndex(Box::new(idx_drop))]
        }
        TableConstraint::Check { name, .. } => {
            let pg_table = quote_ident(table, DatabaseBackend::Postgres);
            let pg_name = quote_ident(name, DatabaseBackend::Postgres);
            let mysql_table = quote_ident(table, DatabaseBackend::MySql);
            let mysql_name = quote_ident(name, DatabaseBackend::MySql);
            let pg_sql = format!("ALTER TABLE {pg_table} DROP CONSTRAINT {pg_name}");
            vec![BuiltQuery::Raw(RawSql::per_backend(
                pg_sql.clone(),
                format!("ALTER TABLE {mysql_table} DROP CHECK {mysql_name}"),
                pg_sql,
            ))]
        }
        _ => unreachable!("TableConstraint is #[non_exhaustive]; all variants are matched above"),
    }
}