tmp

Author: kvch

pkg/migrations/dbactions.go

@@ -844,12 +844,12 @@ func (a *setReplicaIdentityAction) Execute(ctx context.Context) error {
 
 type alterReferencesAction struct {
 	conn         db.DB
-	referencedBy map[string][]*schema.ReferencedBy
+	referencedBy schema.ReferencedBy
 	table        string
 	column       string
 }
 
-func NewAlterReferencesAction(conn db.DB, referencedBy map[string][]*schema.ReferencedBy, table, column string) *alterReferencesAction {
+func NewAlterReferencesAction(conn db.DB, referencedBy schema.ReferencedBy, table, column string) *alterReferencesAction {
 	return &alterReferencesAction{
 		conn:         conn,
 		referencedBy: referencedBy,
@@ -859,30 +859,30 @@ func NewAlterReferencesAction(conn db.DB, referencedBy map[string][]*schema.Refe
 }
 
 func (a *alterReferencesAction) Execute(ctx context.Context) error {
-	for table, constraints := range a.referencedBy {
-		for _, constraint := range constraints {
-			// Drop the existing constraint
-			_, err := a.conn.ExecContext(ctx, fmt.Sprintf("ALTER TABLE %s DROP CONSTRAINT %s",
-				pq.QuoteIdentifier(table),
-				pq.QuoteIdentifier(constraint.Name),
-			))
-			if err != nil {
-				return fmt.Errorf("dropping constraint %s on %s: %w", constraint.Name, table, err)
-			}
-
-			// Recreate the constraint with the table and new column
-			newDef := strings.ReplaceAll(constraint.Definition, a.column, pq.QuoteIdentifier(TemporaryName(a.column)))
-			newDef = strings.ReplaceAll(newDef, a.table, pq.QuoteIdentifier(a.table))
-			_, err = a.conn.ExecContext(ctx, fmt.Sprintf("ALTER TABLE %s ADD CONSTRAINT %s %s",
-				pq.QuoteIdentifier(table),
-				pq.QuoteIdentifier(constraint.Name),
-				newDef,
-			))
-			if err != nil {
-				return fmt.Errorf("altering references for %s.%s: %w", a.table, a.column, err)
-			}
-
-		}
-	}
+	//for table, constraints := range a.referencedBy {
+	//	for _, constraint := range constraints {
+	//		// Drop the existing constraint
+	//		_, err := a.conn.ExecContext(ctx, fmt.Sprintf("ALTER TABLE %s DROP CONSTRAINT %s",
+	//			pq.QuoteIdentifier(table),
+	//			pq.QuoteIdentifier(constraint.Name),
+	//		))
+	//		if err != nil {
+	//			return fmt.Errorf("dropping constraint %s on %s: %w", constraint.Name, table, err)
+	//		}
+
+	//		// Recreate the constraint with the table and new column
+	//		newDef := strings.ReplaceAll(constraint.Definition, a.column, pq.QuoteIdentifier(TemporaryName(a.column)))
+	//		newDef = strings.ReplaceAll(newDef, a.table, pq.QuoteIdentifier(a.table))
+	//		_, err = a.conn.ExecContext(ctx, fmt.Sprintf("ALTER TABLE %s ADD CONSTRAINT %s %s",
+	//			pq.QuoteIdentifier(table),
+	//			pq.QuoteIdentifier(constraint.Name),
+	//			newDef,
+	//		))
+	//		if err != nil {
+	//			return fmt.Errorf("altering references for %s.%s: %w", a.table, a.column, err)
+	//		}
+
+	//	}
+	//}
 	return nil
 }

pkg/schema/schema.go

@@ -65,7 +65,8 @@ type Table struct {
 	// ReferencedBy is a map of table names that reference this table by foreign key
 	// The key is the name of the referencing table, and the value is the name and
 	// the definition of the foreign key constraint
-	ReferencedBy map[string][]*ReferencedBy `json:"referencedBy"`
+	//ReferencedBy map[string][]*ReferencedBy `json:"referencedBy"`
+	ReferencedBy ReferencedBy `json:"referencedBy"`
 
 	// Whether or not the table has been deleted in the virtual schema
 	Deleted bool `json:"-"`

pkg/state/init.sql

@@ -346,7 +346,20 @@ BEGIN
                                                 AND fk_constraint.contype = 'f' GROUP BY fk_constraint.conrelid, fk_constraint.conname, fk_constraint.confrelid, fk_cl.relname, fk_constraint.confkey, fk_constraint.confmatchtype, fk_constraint.confdeltype, fk_constraint.confupdtype) AS fk_info
                                             INNER JOIN pg_attribute ref_attr ON ref_attr.attrelid = fk_info.confrelid
                                                 AND ref_attr.attnum = ANY (fk_info.confkey) -- join the columns of the referenced table
-                                        GROUP BY fk_info.conname, fk_info.conrelid, fk_info.columns, fk_info.confrelid, fk_info.confmatchtype, fk_info.confdeltype, fk_info.confupdtype, fk_info.relname) AS fk_details)))), '{}'::json)
+                                        GROUP BY fk_info.conname, fk_info.conrelid, fk_info.columns, fk_info.confrelid, fk_info.confmatchtype, fk_info.confdeltype, fk_info.confupdtype, fk_info.relname) AS fk_details), 'referencedBy', (
+                                    SELECT
+                                        json_object_agg(ref_fk.conname, json_build_object('name', ref_fk.conname, 'table', ref_fk.relname, 'definition', ref_fk.definition))
+                                    FROM (
+                                        SELECT
+                                            ref_constraint.conname,
+                                            ref_cl.relname,
+                                            pg_get_constraintdef(ref_constraint.oid) AS definition
+                                        FROM pg_constraint AS ref_constraint
+                                        INNER JOIN pg_class ref_cl ON ref_constraint.conrelid = ref_cl.oid
+                                        WHERE
+                                            ref_constraint.confrelid = t.oid
+                                            AND ref_constraint.contype = 'f'
+                                    ) AS ref_fk)))), '{}'::json)
                     FROM pg_class AS t
                     INNER JOIN pg_namespace AS ns ON t.relnamespace = ns.oid
                     LEFT JOIN pg_description AS descr ON t.oid = descr.objoid