so far. clone not required

juleswritescode · juleswritescode · commit 3426fcf9987e · 2025-05-16T15:05:48.000+02:00
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/pgt_schema_cache/Cargo.toml b/crates/pgt_schema_cache/Cargo.toml
@@ -21,6 +21,7 @@ serde.workspace           = true
 serde_json.workspace      = true
 sqlx.workspace            = true
 tokio.workspace           = true
+strum                     = { workspace = true }
 
 [dev-dependencies]
 pgt_test_utils.workspace = true
diff --git a/crates/pgt_schema_cache/src/columns.rs b/crates/pgt_schema_cache/src/columns.rs
@@ -37,7 +37,7 @@ impl From<char> for ColumnClassKind {
     }
 }
 
-#[derive(Debug, Clone, PartialEq, Eq)]
+#[derive(Debug, PartialEq, Eq)]
 pub struct Column {
     pub name: String,
 
diff --git a/crates/pgt_schema_cache/src/functions.rs b/crates/pgt_schema_cache/src/functions.rs
@@ -58,7 +58,7 @@ impl From<Option<JsonValue>> for FunctionArgs {
     }
 }
 
-#[derive(Debug, Clone, Default, Serialize, Deserialize)]
+#[derive(Debug, Default, Serialize, Deserialize)]
 pub struct Function {
     /// The Id (`oid`).
     pub id: i64,
diff --git a/crates/pgt_schema_cache/src/lib.rs b/crates/pgt_schema_cache/src/lib.rs
@@ -8,6 +8,7 @@ mod policies;
 mod schema_cache;
 mod schemas;
 mod tables;
+mod triggers;
 mod types;
 mod versions;
 
@@ -16,3 +17,4 @@ pub use functions::{Behavior, Function, FunctionArg, FunctionArgs};
 pub use schema_cache::SchemaCache;
 pub use schemas::Schema;
 pub use tables::{ReplicaIdentity, Table};
+pub use triggers::{Trigger, TriggerAffected, TriggerEvent};
diff --git a/crates/pgt_schema_cache/src/policies.rs b/crates/pgt_schema_cache/src/policies.rs
@@ -54,7 +54,7 @@ impl From<PolicyQueried> for Policy {
     }
 }
 
-#[derive(Debug, Clone, PartialEq, Eq)]
+#[derive(Debug, PartialEq, Eq)]
 pub struct Policy {
     name: String,
     table_name: String,
diff --git a/crates/pgt_schema_cache/src/queries/triggers.sql b/crates/pgt_schema_cache/src/queries/triggers.sql
@@ -0,0 +1,17 @@
+-- we need to join tables from the pg_catalog since "TRUNCATE" triggers are 
+-- not available in the information_schema.trigger table.
+select 
+  t.tgname as "name!",
+  c.relname as "table_name!",
+  p.proname as "proc_name!",
+  n.nspname as "schema_name!",
+  t.tgtype as "details_bitmask!"
+from 
+  pg_catalog.pg_trigger t 
+  left join pg_catalog.pg_proc p on t.tgfoid = p.oid
+  left join pg_catalog.pg_class c on t.tgrelid = c.oid
+  left join pg_catalog.pg_namespace n on c.relnamespace = n.oid
+where 
+  -- triggers enforcing constraints (e.g. unique fields) should not be included.
+  t.tgisinternal = false and 
+  t.tgconstraint = 0;
diff --git a/crates/pgt_schema_cache/src/schema_cache.rs b/crates/pgt_schema_cache/src/schema_cache.rs
@@ -1,5 +1,6 @@
 use sqlx::postgres::PgPool;
 
+use crate::Trigger;
 use crate::columns::Column;
 use crate::functions::Function;
 use crate::policies::Policy;
@@ -8,7 +9,7 @@ use crate::tables::Table;
 use crate::types::PostgresType;
 use crate::versions::Version;
 
-#[derive(Debug, Clone, Default)]
+#[derive(Debug, Default)]
 pub struct SchemaCache {
     pub schemas: Vec<Schema>,
     pub tables: Vec<Table>,
@@ -17,18 +18,20 @@ pub struct SchemaCache {
     pub versions: Vec<Version>,
     pub columns: Vec<Column>,
     pub policies: Vec<Policy>,
+    pub triggers: Vec<Trigger>,
 }
 
 impl SchemaCache {
     pub async fn load(pool: &PgPool) -> Result<SchemaCache, sqlx::Error> {
-        let (schemas, tables, functions, types, versions, columns, policies) = futures_util::try_join!(
+        let (schemas, tables, functions, types, versions, columns, policies, triggers) = futures_util::try_join!(
             Schema::load(pool),
             Table::load(pool),
             Function::load(pool),
             PostgresType::load(pool),
             Version::load(pool),
             Column::load(pool),
             Policy::load(pool),
+            Trigger::load(pool),
         )?;
 
         Ok(SchemaCache {
@@ -39,6 +42,7 @@ impl SchemaCache {
             versions,
             columns,
             policies,
+            triggers,
         })
     }
 
diff --git a/crates/pgt_schema_cache/src/schemas.rs b/crates/pgt_schema_cache/src/schemas.rs
@@ -2,7 +2,7 @@ use sqlx::PgPool;
 
 use crate::schema_cache::SchemaCacheItem;
 
-#[derive(Debug, Clone, Default)]
+#[derive(Debug, Default)]
 pub struct Schema {
     pub id: i64,
     pub name: String,
diff --git a/crates/pgt_schema_cache/src/tables.rs b/crates/pgt_schema_cache/src/tables.rs
@@ -23,7 +23,7 @@ impl From<String> for ReplicaIdentity {
     }
 }
 
-#[derive(Debug, Clone, Default, PartialEq, Eq)]
+#[derive(Debug, Default, PartialEq, Eq)]
 pub struct Table {
     pub id: i64,
     pub schema: String,
diff --git a/crates/pgt_schema_cache/src/triggers.rs b/crates/pgt_schema_cache/src/triggers.rs
@@ -0,0 +1,201 @@
+use crate::schema_cache::SchemaCacheItem;
+use strum::{EnumIter, IntoEnumIterator};
+
+#[derive(Debug, PartialEq, Eq)]
+pub enum TriggerAffected {
+    Row,
+    Statement,
+}
+
+impl From<i16> for TriggerAffected {
+    fn from(value: i16) -> Self {
+        let is_row = 0b0000_0001;
+        if value & is_row == is_row {
+            Self::Row
+        } else {
+            Self::Statement
+        }
+    }
+}
+
+#[derive(Debug, PartialEq, Eq, EnumIter)]
+pub enum TriggerEvent {
+    Insert,
+    Delete,
+    Update,
+    Truncate,
+}
+
+struct TriggerEvents(Vec<TriggerEvent>);
+
+impl From<i16> for TriggerEvents {
+    fn from(value: i16) -> Self {
+        Self(
+            TriggerEvent::iter()
+                .filter(|variant| {
+                    #[rustfmt::skip]
+                    let mask = match variant {
+                        TriggerEvent::Insert   => 0b0000_0100,
+                        TriggerEvent::Delete   => 0b0000_1000,
+                        TriggerEvent::Update   => 0b0001_0000,
+                        TriggerEvent::Truncate => 0b0010_0000,
+                    };
+                    mask & value == mask
+                })
+                .collect(),
+        )
+    }
+}
+
+#[derive(Debug, PartialEq, Eq, EnumIter)]
+pub enum TriggerTiming {
+    Before,
+    After,
+    Instead,
+}
+
+impl TryFrom<i16> for TriggerTiming {
+    type Error = ();
+    fn try_from(value: i16) -> Result<Self, ()> {
+        TriggerTiming::iter()
+            .find(|variant| {
+                #[rustfmt::skip]
+                    let mask = match variant {
+                        TriggerTiming::Instead   => 0b0100_0000,
+                        TriggerTiming::Before    => 0b0000_0010,
+                        TriggerTiming::After     => 0b0000_0000, // before/after share same bit
+                    };
+                mask & value == mask
+            })
+            .ok_or(())
+    }
+}
+
+pub struct TriggerQueried {
+    name: String,
+    table_name: String,
+    schema_name: String,
+    proc_name: String,
+    details_bitmask: i16,
+}
+
+#[derive(Debug, PartialEq, Eq)]
+pub struct Trigger {
+    name: String,
+    table_name: String,
+    schema_name: String,
+    affected: TriggerAffected,
+    timing: TriggerTiming,
+    events: Vec<TriggerEvent>,
+}
+
+impl From<TriggerQueried> for Trigger {
+    fn from(value: TriggerQueried) -> Self {
+        Self {
+            name: value.name,
+            table_name: value.table_name,
+            schema_name: value.schema_name,
+            affected: value.details_bitmask.into(),
+            timing: value.details_bitmask.try_into().unwrap(),
+            events: TriggerEvents::from(value.details_bitmask).0,
+        }
+    }
+}
+
+impl SchemaCacheItem for Trigger {
+    type Item = Trigger;
+
+    async fn load(pool: &sqlx::PgPool) -> Result<Vec<Self::Item>, sqlx::Error> {
+        let results = sqlx::query_file_as!(TriggerQueried, "src/queries/triggers.sql")
+            .fetch_all(pool)
+            .await?;
+
+        Ok(results.into_iter().map(|r| r.into()).collect())
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use pgt_test_utils::test_database::get_new_test_db;
+    use sqlx::Executor;
+
+    use crate::{
+        SchemaCache,
+        triggers::{TriggerAffected, TriggerEvent, TriggerTiming},
+    };
+
+    #[tokio::test]
+    async fn loads_triggers() {
+        let test_db = get_new_test_db().await;
+
+        let setup = r#"
+                create table public.users (
+                    id serial primary key,
+                    name text
+                );
+    
+                create or replace function public.log_user_insert()
+                returns trigger as $$
+                begin
+                    -- dummy body
+                    return new;
+                end;
+                $$ language plpgsql;
+    
+                create trigger trg_users_insert
+                    before insert on public.users
+                    for each row
+                    execute function public.log_user_insert();
+    
+                create trigger trg_users_update
+                    after update on public.users
+                    for each statement
+                    execute function public.log_user_insert();
+    
+                create trigger trg_users_delete
+                    before delete on public.users
+                    for each row
+                    execute function public.log_user_insert();
+            "#;
+
+        test_db
+            .execute(setup)
+            .await
+            .expect("Failed to setup test database");
+
+        let cache = SchemaCache::load(&test_db)
+            .await
+            .expect("Failed to load Schema Cache");
+
+        let triggers: Vec<_> = cache
+            .triggers
+            .iter()
+            .filter(|t| t.table_name == "users")
+            .collect();
+        assert_eq!(triggers.len(), 3);
+
+        let insert_trigger = triggers
+            .iter()
+            .find(|t| t.name == "trg_users_insert")
+            .unwrap();
+        assert_eq!(insert_trigger.timing, TriggerTiming::Before);
+        assert_eq!(insert_trigger.affected, TriggerAffected::Row);
+        assert!(insert_trigger.events.contains(&TriggerEvent::Insert));
+
+        let update_trigger = triggers
+            .iter()
+            .find(|t| t.name == "trg_users_update")
+            .unwrap();
+        assert_eq!(update_trigger.timing, TriggerTiming::After);
+        assert_eq!(update_trigger.affected, TriggerAffected::Statement);
+        assert!(update_trigger.events.contains(&TriggerEvent::Update));
+
+        let delete_trigger = triggers
+            .iter()
+            .find(|t| t.name == "trg_users_delete")
+            .unwrap();
+        assert_eq!(delete_trigger.timing, TriggerTiming::Before);
+        assert_eq!(delete_trigger.affected, TriggerAffected::Row);
+        assert!(delete_trigger.events.contains(&TriggerEvent::Delete));
+    }
+}
diff --git a/crates/pgt_schema_cache/src/types.rs b/crates/pgt_schema_cache/src/types.rs
@@ -36,7 +36,7 @@ impl From<Option<JsonValue>> for Enums {
     }
 }
 
-#[derive(Debug, Clone, Default)]
+#[derive(Debug, Default)]
 pub struct PostgresType {
     pub id: i64,
     pub name: String,
diff --git a/crates/pgt_schema_cache/src/versions.rs b/crates/pgt_schema_cache/src/versions.rs
@@ -2,7 +2,7 @@ use sqlx::PgPool;
 
 use crate::schema_cache::SchemaCacheItem;
 
-#[derive(Debug, Clone, Default)]
+#[derive(Debug, Default)]
 pub struct Version {
     pub version: Option<String>,
     pub version_num: Option<i64>,

Original file line number	Diff line number	Diff line change
`@@ -37,7 +37,7 @@ impl From<char> for ColumnClassKind {`
`37`	`37`	`}`
`38`	`38`	`}`
`39`	`39`
`40`		`-#[derive(Debug, Clone, PartialEq, Eq)]`
	`40`	`+#[derive(Debug, PartialEq, Eq)]`
`41`	`41`	`pub struct Column {`
`42`	`42`	`pub name: String,`
`43`	`43`
Original file line number	Diff line number	Diff line change
`@@ -58,7 +58,7 @@ impl From<Option<JsonValue>> for FunctionArgs {`
`58`	`58`	`}`
`59`	`59`	`}`
`60`	`60`
`61`		`-#[derive(Debug, Clone, Default, Serialize, Deserialize)]`
	`61`	`+#[derive(Debug, Default, Serialize, Deserialize)]`
`62`	`62`	`pub struct Function {`
`63`	`63`	/// The Id (`oid`).
`64`	`64`	`pub id: i64,`
Original file line number	Diff line number	Diff line change
`@@ -54,7 +54,7 @@ impl From<PolicyQueried> for Policy {`
`54`	`54`	`}`
`55`	`55`	`}`
`56`	`56`
`57`		`-#[derive(Debug, Clone, PartialEq, Eq)]`
	`57`	`+#[derive(Debug, PartialEq, Eq)]`
`58`	`58`	`pub struct Policy {`
`59`	`59`	`name: String,`
`60`	`60`	`table_name: String,`
Original file line number	Diff line number	Diff line change
`@@ -23,7 +23,7 @@ impl From<String> for ReplicaIdentity {`
`23`	`23`	`}`
`24`	`24`	`}`
`25`	`25`
`26`		`-#[derive(Debug, Clone, Default, PartialEq, Eq)]`
	`26`	`+#[derive(Debug, Default, PartialEq, Eq)]`
`27`	`27`	`pub struct Table {`
`28`	`28`	`pub id: i64,`
`29`	`29`	`pub schema: String,`