introduce function copy_foreign_keys(), generate fkeys on children

funbringer · funbringer · commit 562cab6c78bd · 2016-09-06T18:42:49.000+03:00
diff --git a/hash.sql b/hash.sql
@@ -63,12 +63,14 @@ BEGIN
 		EXECUTE format('ALTER TABLE %s ADD CONSTRAINT %s
 						CHECK (@extschema@.get_hash_part_idx(%s(%s), %s) = %s)',
 					   v_child_relname,
-					   @extschema@.build_check_constraint_name(v_child_relname::regclass,
+					   @extschema@.build_check_constraint_name(v_child_relname::REGCLASS,
 															   attribute),
 					   v_hashfunc,
 					   attribute,
 					   partitions_count,
 					   partnum);
+
+		PERFORM @extschema@.copy_foreign_keys(parent_relid, v_child_relname::REGCLASS);
 	END LOOP;
 
 	/* Notify backend about changes */
diff --git a/init.sql b/init.sql
@@ -546,6 +546,31 @@ $$ LANGUAGE plpgsql
 SET pg_pathman.enable_partitionfilter = off; /* ensures that PartitionFilter is OFF */
 
 
+/*
+ * Copy all of parent's foreign keys.
+ */
+CREATE OR REPLACE FUNCTION @extschema@.copy_foreign_keys(
+	parent_relid	REGCLASS,
+	partition		REGCLASS)
+RETURNS VOID AS
+$$
+DECLARE
+	rec		RECORD;
+
+BEGIN
+	PERFORM @extschema@.validate_relname(parent_relid);
+	PERFORM @extschema@.validate_relname(partition);
+
+	FOR rec IN (SELECT oid as conid FROM pg_catalog.pg_constraint
+				WHERE conrelid = parent_relid AND contype = 'f')
+	LOOP
+		EXECUTE format('ALTER TABLE %s ADD %s',
+					   partition::TEXT,
+					   pg_get_constraintdef(rec.conid));
+	END LOOP;
+END
+$$ LANGUAGE plpgsql;
+
 
 /*
  * Create DDL trigger to call pathman_ddl_trigger_func().
diff --git a/range.sql b/range.sql
@@ -478,6 +478,8 @@ BEGIN
 													 p_start_value,
 													 p_end_value));
 
+	PERFORM @extschema@.copy_foreign_keys(parent_relid, v_child_relname::REGCLASS);
+
 	RETURN v_child_relname;
 END
 $$ LANGUAGE plpgsql
