diesel_cli 2.3.6

Provides the CLI for the Diesel crate
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

#![allow(clippy::expect_fun_call)] // My calls are so fun

use super::data_structures::ForeignKeyConstraint;
use super::inference::get_primary_keys;
use super::table_data::TableName;
use crate::database::InferConnection;

pub fn remove_unsafe_foreign_keys_for_codegen(
    conn: &mut InferConnection,
    foreign_keys: &[ForeignKeyConstraint],
    safe_tables: &[TableName],
) -> Vec<ForeignKeyConstraint> {
    let duplicates = foreign_keys
        .iter()
        .map(ForeignKeyConstraint::ordered_tables)
        .filter(|tables| {
            let dup_count = foreign_keys
                .iter()
                .filter(|fk| tables == &fk.ordered_tables())
                .count();
            if dup_count > 1 {
                tracing::debug!(
                    ?tables,
                    "Remove foreign key constraint because it's not unique"
                );
            }
            dup_count > 1
        })
        .collect::<Vec<_>>();

    foreign_keys
        .iter()
        .filter(|fk| {
            if fk.parent_table == fk.child_table {
                tracing::debug!(?fk, "Remove foreign key constraint because it's self referential")
            }
            fk.parent_table != fk.child_table
        })
        .filter(|fk|{
            if !safe_tables.contains(&fk.parent_table) {
                tracing::debug!(?fk, "Remove foreign key constraint because it references a table not in the outputted schema");
            }
            safe_tables.contains(&fk.parent_table)
        })
        .filter(|fk| {
            if !safe_tables.contains(&fk.child_table) {
                tracing::debug!(?fk, "Remove foreign key constraint because it references a table not in the outputted schema");
            }
            safe_tables.contains(&fk.child_table)
        })
        .filter(|fk| {
            if fk.foreign_key_columns.len() != 1 {
                tracing::debug!(?fk, "Remove foreign key constraint because it contains several foreign keys");
            }
            fk.foreign_key_columns.len() == 1
        })
        .filter(|fk| {
            let pk_columns = get_primary_keys(conn, &fk.parent_table).expect(&format!(
                "Error loading primary keys for `{}`",
                fk.parent_table
            ));
            let condition = pk_columns.len() == 1 && Some(&pk_columns[0]) == fk.primary_key_columns.first();
            if !condition {
                tracing::debug!(?fk, "Remove foreign key constraint because it references a table with several primary keys");
            }
            condition
        })
        .filter(|fk| !duplicates.contains(&fk.ordered_tables()))
        .cloned()
        .collect()
}