elefant-tools 0.0.8

A library for doing things like pg_dump and pg_restore, with extra features, and probably more bugs.
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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98

use crate::postgres_client_wrapper::{QueryResult, RowEnumExt};
use crate::ReferenceAction;

pub struct ForeignKeyResult {
    pub constraint_name: String,
    // pub constraint_schema_name: String,
    pub source_table_name: String,
    pub source_table_schema_name: String,
    pub target_table_name: String,
    pub target_table_schema_name: String,
    pub update_action: ReferenceAction,
    pub delete_action: ReferenceAction,
    pub comment: Option<String>,
    pub is_enforced: bool,
    pub constraint_definition: Option<String>,
}

impl<'a> elefant_client::FromSqlRow<'a> for ForeignKeyResult {
    fn from_sql_row(
        row: &'a elefant_client::PostgresDataRow<'_, '_>,
    ) -> Result<Self, elefant_client::ElefantClientError> {
        Ok(Self {
            constraint_name: row.get(0)?,
            // constraint_schema_name: row.get(1)?,
            source_table_name: row.get(2)?,
            source_table_schema_name: row.get(3)?,
            target_table_name: row.get(4)?,
            target_table_schema_name: row.get(5)?,
            update_action: row.try_get_enum_value(6)?,
            delete_action: row.try_get_enum_value(7)?,
            comment: row.get(8)?,
            is_enforced: row.get(9)?,
            constraint_definition: row.get(10)?,
        })
    }
}

//language=postgresql
pub(in crate::schema_reader) const QUERY_V18: &str = r#"
select con.conname              as constraint_name,
       con_ns.nspname           as constraint_schema_name,
       tab.relname              as source_table_name,
       tab_ns.nspname           as source_schema_name,
       target.relname           as target_table_name,
       target_ns.nspname        as target_schema_name,
       con.confupdtype    as update_action,
       con.confdeltype    as delete_action,
       d.description       as comment,
       con.conenforced     as is_enforced,
       case when con.conperiod then pg_get_constraintdef(con.oid) else null end as constraint_definition
from pg_catalog.pg_constraint con
         left join pg_catalog.pg_namespace con_ns on con_ns.oid = con.connamespace
         join pg_catalog.pg_class tab on con.conrelid = tab.oid
         left join pg_namespace tab_ns on tab_ns.oid = tab.relnamespace
         join pg_catalog.pg_class target on con.confrelid = target.oid
         left join pg_namespace target_ns on target_ns.oid = target.relnamespace
         left join pg_description d on d.objoid = con.oid
         left join pg_depend dep on dep.objid = con_ns.oid
where con.contype = 'f'
  and (dep.objid is null or dep.deptype <> 'e' )
order by constraint_schema_name, source_table_name, constraint_name;
"#;

//language=postgresql
pub(in crate::schema_reader) const QUERY_LEGACY: &str = r#"
select con.conname              as constraint_name,
       con_ns.nspname           as constraint_schema_name,
       tab.relname              as source_table_name,
       tab_ns.nspname           as source_schema_name,
       target.relname           as target_table_name,
       target_ns.nspname        as target_schema_name,
       con.confupdtype    as update_action,
       con.confdeltype    as delete_action,
       d.description       as comment,
       true                as is_enforced,
       null::text          as constraint_definition
from pg_catalog.pg_constraint con
         left join pg_catalog.pg_namespace con_ns on con_ns.oid = con.connamespace
         join pg_catalog.pg_class tab on con.conrelid = tab.oid
         left join pg_namespace tab_ns on tab_ns.oid = tab.relnamespace
         join pg_catalog.pg_class target on con.confrelid = target.oid
         left join pg_namespace target_ns on target_ns.oid = target.relnamespace
         left join pg_description d on d.objoid = con.oid
         left join pg_depend dep on dep.objid = con_ns.oid
where con.contype = 'f'
  and (dep.objid is null or dep.deptype <> 'e' )
order by constraint_schema_name, source_table_name, constraint_name;
"#;

impl QueryResult for ForeignKeyResult {
    fn query(version: i32) -> &'static str {
        if version >= 180 {
            QUERY_V18
        } else {
            QUERY_LEGACY
        }
    }
}