fix: skip parsing partitioned Postgres tables

src/postgres/query/constraints/mod.rs

@@ -9,7 +9,7 @@ pub use referential_constraints::*;
 pub use table_constraints::*;
 
 use super::{InformationSchema, SchemaQueryBuilder};
-use crate::sqlx_types::postgres::PgRow;
+use crate::{postgres::query::select_table_and_view, sqlx_types::postgres::PgRow};
 use sea_query::{Alias, Condition, Expr, Iden, JoinType, Order, Query, SeaRc, SelectStatement};
 
 #[derive(Debug, Default)]
@@ -162,6 +162,9 @@ impl SchemaQueryBuilder {
                 Expr::col((Schema::TableConstraints, Tcf::TableSchema)).eq(schema.to_string()),
             )
             .and_where(Expr::col((Schema::TableConstraints, Tcf::TableName)).eq(table.to_string()))
+            .and_where(
+                Expr::col((rcsq.clone(), Kcuf::TableName)).not_in_subquery(select_table_and_view()),
+            )
             .order_by((Schema::TableConstraints, Tcf::ConstraintName), Order::Asc)
             .order_by((Schema::KeyColumnUsage, Kcuf::OrdinalPosition), Order::Asc)
             .order_by((rcsq.clone(), RefC::UniqueConstraintName), Order::Asc)

src/postgres/query/schema.rs

@@ -1,7 +1,9 @@
+use sea_query::{Condition, Expr, Iden, JoinType, Query, SelectStatement};
+
 #[derive(Debug, Default)]
 pub struct SchemaQueryBuilder;
 
-#[derive(Debug, sea_query::Iden)]
+#[derive(Debug, Iden)]
 /// Ref: https://www.postgresql.org/docs/13/information-schema.html
 pub enum InformationSchema {
     #[iden = "information_schema"]
@@ -14,3 +16,37 @@ pub enum InformationSchema {
     TableConstraints,
     ConstraintColumnUsage,
 }
+
+pub(crate) fn select_table_and_view() -> SelectStatement {
+    #[derive(Debug, Iden)]
+    pub enum PgClass {
+        Table,
+        Relname,
+        Relkind,
+        Oid,
+    }
+
+    #[derive(Debug, Iden)]
+    pub enum PgInherits {
+        Table,
+        Inhrelid,
+    }
+
+    Query::select()
+        .column((PgClass::Table, PgClass::Relname))
+        .from(PgInherits::Table)
+        .join(
+            JoinType::Join,
+            PgClass::Table,
+            Condition::all()
+                .add(
+                    Expr::col((PgInherits::Table, PgInherits::Inhrelid))
+                        .equals((PgClass::Table, PgClass::Oid)),
+                )
+                .add(
+                    Expr::col((PgClass::Table, PgClass::Relkind))
+                        .is_in(["r", "t", "v", "m", "f", "p"]),
+                ),
+        )
+        .to_owned()
+}

src/postgres/query/table.rs

@@ -1,4 +1,4 @@
-use super::{InformationSchema, SchemaQueryBuilder};
+use super::{select_table_and_view, InformationSchema, SchemaQueryBuilder};
 use crate::sqlx_types::postgres::PgRow;
 use sea_query::{Expr, Iden, Query, SeaRc, SelectStatement};
 
@@ -46,6 +46,7 @@ impl SchemaQueryBuilder {
             .from((InformationSchema::Schema, InformationSchema::Tables))
             .and_where(Expr::col(TablesFields::TableSchema).eq(schema.to_string()))
             .and_where(Expr::col(TablesFields::TableType).eq(TableType::BaseTable.to_string()))
+            .and_where(Expr::col(TablesFields::TableName).not_in_subquery(select_table_and_view()))
             .take()
     }
 }