Record whether the constraint is a foreign key or not

Author: sandhose

crates/syn2mas/.sqlx/query-12112011318abc0bdd7f722ed8c5d4a86bf5758f8c32d9d41a22999b2f0698ca.json

@@ -15,6 +15,7 @@ pub struct ConstraintDescription {
     pub name: String,
     pub table_name: String,
     pub definition: String,
+    pub is_fk: bool,
 }
 
 pub struct IndexDescription {
@@ -31,14 +32,21 @@ pub async fn describe_constraints_on_table(
     sqlx::query_as!(
         ConstraintDescription,
         r#"
-            SELECT conrelid::regclass::text AS "table_name!", conname AS "name!", pg_get_constraintdef(c.oid) AS "definition!"
+            SELECT
+                conrelid::regclass::text AS "table_name!",
+                conname AS "name!",
+                pg_get_constraintdef(c.oid) AS "definition!",
+                CASE WHEN contype = 'f' THEN true ELSE false END AS "is_fk!"
             FROM pg_constraint c
             JOIN pg_namespace n ON n.oid = c.connamespace
             WHERE contype IN ('f', 'p', 'u') AND conrelid::regclass::text = $1
             AND n.nspname = current_schema;
         "#,
         table_name
-    ).fetch_all(&mut *conn).await.into_database_with(|| format!("could not read constraint definitions of {table_name}"))
+    )
+    .fetch_all(&mut *conn)
+    .await
+    .into_database_with(|| format!("could not read constraint definitions of {table_name}"))
 }
 
 /// Look up and return the definitions of foreign-key constraints whose
@@ -50,14 +58,23 @@ pub async fn describe_foreign_key_constraints_to_table(
     sqlx::query_as!(
         ConstraintDescription,
         r#"
-            SELECT conrelid::regclass::text AS "table_name!", conname AS "name!", pg_get_constraintdef(c.oid) AS "definition!"
+            SELECT
+                conrelid::regclass::text AS "table_name!",
+                conname AS "name!",
+                pg_get_constraintdef(c.oid) AS "definition!",
+                true AS "is_fk!"
             FROM pg_constraint c
             JOIN pg_namespace n ON n.oid = c.connamespace
             WHERE contype = 'f' AND confrelid::regclass::text = $1
             AND n.nspname = current_schema;
         "#,
         target_table_name
-    ).fetch_all(&mut *conn).await.into_database_with(|| format!("could not read FK constraint definitions targetting {target_table_name}"))
+    )
+    .fetch_all(&mut *conn)
+    .await
+    .into_database_with(|| {
+        format!("could not read FK constraint definitions targetting {target_table_name}")
+    })
 }
 
 /// Look up and return the definitions of all indices on a given table.
@@ -122,6 +139,7 @@ pub async fn restore_constraint(
         name,
         table_name,
         definition,
+        is_fk: _,
     } = &constraint;
     info!("rebuilding constraint {name}");
 

crates/syn2mas/.sqlx/query-78ed3bf1032cd678b42230d68fb2b8e3d74161c8b6c5fe1a746b6958ccd2fd84.json renamed to crates/syn2mas/.sqlx/query-51b7f77e2f8d4a03cfceac46e1c22e47932d427950cd416eb7c46e0451582d8b.json

@@ -424,7 +424,7 @@ impl<'conn> MasWriter<'conn> {
                 .into_database("failed to get syn2mas restore data (index descriptions)")?;
             constraints_to_restore = query_as!(
                 ConstraintDescription,
-                "SELECT table_name, name, definition FROM syn2mas_restore_constraints ORDER BY order_key"
+                "SELECT table_name, name, definition, is_fk FROM syn2mas_restore_constraints ORDER BY order_key"
             )
                 .fetch_all(conn.as_mut())
                 .await
@@ -468,16 +468,18 @@ impl<'conn> MasWriter<'conn> {
                 name,
                 table_name,
                 definition,
+                is_fk,
             } in &constraints_to_restore
             {
                 query!(
                     r#"
-                    INSERT INTO syn2mas_restore_constraints (name, table_name, definition)
-                    VALUES ($1, $2, $3)
+                    INSERT INTO syn2mas_restore_constraints (name, table_name, definition, is_fk)
+                    VALUES ($1, $2, $3, $4)
                     "#,
                     name,
                     table_name,
-                    definition
+                    definition,
+                    is_fk
                 )
                 .execute(conn.as_mut())
                 .await

crates/syn2mas/.sqlx/query-575807a26f5caba0a71894f2e3fe4646e23dda898f563fa8cc91d26f8a9d6676.json

@@ -23,7 +23,8 @@ CREATE TABLE syn2mas_restore_constraints (
 
     table_name TEXT NOT NULL,
     name TEXT NOT NULL,
-    definition TEXT NOT NULL
+    definition TEXT NOT NULL,
+    is_fk BOOLEAN NOT NULL
 );
 
 -- corresponds to `IndexDescription`