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
99
100
101
102

use crate::postgres_client_wrapper::{FromPgChar, QueryResult, RowEnumExt};
use crate::{ElefantToolsError, TablePartitionStrategy};

#[derive(Debug, Eq, PartialEq)]
pub struct TablesResult {
    pub schema_name: String,
    pub table_name: String,
    pub comment: Option<String>,
    pub table_type: TableType,
    pub partition_expression: Option<String>,
    pub partition_strategy: Option<TablePartitionStrategy>,
    pub default_partition_name: Option<String>,
    pub partition_column_indices: Option<Vec<i16>>,
    pub partition_expression_columns: Option<String>,
    pub parent_tables: Option<Vec<String>>,
    pub is_partition: bool,
    pub storage_parameters: Option<Vec<String>>,
    pub oid: i64,
    pub depends_on: Option<Vec<i64>>,
    pub type_oid: i64,
}

#[derive(Debug, Eq, PartialEq, Copy, Clone, Default)]
pub enum TableType {
    #[default]
    Table,
    PartitionedTable,
}

impl FromPgChar for TableType {
    fn from_pg_char(c: char) -> Result<Self, ElefantToolsError> {
        match c {
            'r' => Ok(TableType::Table),
            'p' => Ok(TableType::PartitionedTable),
            _ => Err(ElefantToolsError::InvalidTableType(c.to_string())),
        }
    }
}

impl<'a> elefant_client::FromSqlRow<'a> for TablesResult {
    fn from_sql_row(
        row: &'a elefant_client::PostgresDataRow<'_, '_>,
    ) -> Result<Self, elefant_client::ElefantClientError> {
        Ok(TablesResult {
            schema_name: row.get(0)?,
            table_name: row.get(1)?,
            comment: row.get(2)?,
            table_type: row.try_get_enum_value(3)?,
            partition_expression: row.get(4)?,
            partition_strategy: row.try_get_opt_enum_value(5)?,
            default_partition_name: row.get(6)?,
            partition_column_indices: row.get(7)?,
            partition_expression_columns: row.get(8)?,
            parent_tables: row.get(9)?,
            is_partition: row.get(10)?,
            storage_parameters: row.get(11)?,
            oid: row.get(12)?,
            depends_on: row.get(13)?,
            type_oid: row.get(14)?,
        })
    }
}

//language=postgresql
pub(in crate::schema_reader) const QUERY: &str = r#"
select
    ns.nspname,
    cl.relname,
    des.description,
    cl.relkind,
    pg_get_expr(cl.relpartbound, cl.oid) as partition_expression,
    pt.partstrat,
    default_partition.relname as default_partition,
    pt.partattrs::int2[],
    pg_get_expr(pt.partexprs, pt.partrelid) as partexprs,
    (select array_agg(parent.relname) from (select parent.relname from pg_inherits i
        join pg_class parent on i.inhparent = parent.oid
          where i.inhrelid = cl.oid
          order by i.inhseqno) parent) as parent_table,
    cl.relispartition,
    cl.reloptions,
   cl.oid::int8,
   (select array_agg(refobjid::int8) from pg_depend dep where cl.oid = dep.objid and dep.deptype <> 'e' and dep.refobjid > 16384 and dep.objid <> dep.refobjid) as depends_on,
   cl.reltype::int8
from pg_class cl
         join pg_catalog.pg_namespace ns on ns.oid = cl.relnamespace
         left join pg_description des on des.objoid = cl.oid and des.objsubid = 0
         left join pg_partitioned_table pt on pt.partrelid = cl.oid
         left join pg_class default_partition on default_partition.oid = pt.partdefid
         left join pg_depend dep on dep.objid = ns.oid
where cl.relkind in ('r', 'p')
  and cl.oid > 16384
  and (dep.objid is null or dep.deptype <> 'e' )
    and has_table_privilege(cl.oid, 'SELECT, INSERT, UPDATE')
order by ns.nspname, cl.relname;
"#;

impl QueryResult for TablesResult {
    fn query(_version: i32) -> &'static str {
        QUERY
    }
}