pathman:

zilder · zilder · commit e39af8269c6d · 2016-01-15T16:26:17.000+03:00
* split partition function
* PL functions enchancement
diff --git a/contrib/pathman/pathman.c b/contrib/pathman/pathman.c
@@ -75,6 +75,7 @@ PG_FUNCTION_INFO_V1( on_partitions_created );
 PG_FUNCTION_INFO_V1( on_partitions_updated );
 PG_FUNCTION_INFO_V1( on_partitions_removed );
 PG_FUNCTION_INFO_V1( find_range_partition );
+PG_FUNCTION_INFO_V1( get_partition_range );
 
 
 /*
@@ -1162,3 +1163,54 @@ find_range_partition(PG_FUNCTION_ARGS)
 
 	PG_RETURN_NULL();
 }
+
+/*
+ * Returns range (min, max) as output parameters
+ *
+ * first argument is parent relid
+ * second is partition relid
+ * third and forth are MIN and MAX output parameters
+ */
+Datum
+get_partition_range(PG_FUNCTION_ARGS)
+{
+	int		parent_oid = DatumGetInt32(PG_GETARG_DATUM(0));
+	int		child_oid = DatumGetInt32(PG_GETARG_DATUM(1));
+	int nelems = 2;
+	Datum *elems = palloc(nelems * sizeof(Datum));
+	Oid elemtype = INT4OID;
+	PartRelationInfo   *prel;
+	RangeRelation	   *rangerel;
+	RangeEntry		   *ranges;
+	bool	found;
+	int 	i;
+
+	prel = (PartRelationInfo *)
+		hash_search(relations, (const void *) &parent_oid, HASH_FIND, 0);
+	
+	rangerel = (RangeRelation *)
+		hash_search(range_restrictions, (const void *) &parent_oid, HASH_FIND, NULL);
+
+	if (!prel || !rangerel)
+		PG_RETURN_NULL();
+
+	ranges = dsm_array_get_pointer(&rangerel->ranges);
+	for(i=0; i<rangerel->ranges.length; i++)
+		if (ranges[i].child_oid == child_oid)
+		{
+			found = true;
+			break;
+		}
+
+	if (found)
+	{
+		elems[0] = ranges[i].min;
+		elems[1] = ranges[i].max;
+
+		ArrayType *arr =
+			construct_array(elems, nelems, prel->atttype, sizeof(Oid), true, 'i');
+		PG_RETURN_ARRAYTYPE_P(arr);
+	}
+
+	PG_RETURN_NULL();
+}
diff --git a/contrib/pathman/sql/init.sql b/contrib/pathman/sql/init.sql
@@ -169,22 +169,32 @@ RETURNS VOID AS 'pathman', 'on_partitions_removed' LANGUAGE C STRICT;
 CREATE OR REPLACE FUNCTION find_range_partition(relid OID, value ANYELEMENT)
 RETURNS OID AS 'pathman', 'find_range_partition' LANGUAGE C STRICT;
 
+/*
+ * Returns min and max values for specified RANGE partition.
+ */
+CREATE OR REPLACE FUNCTION get_partition_range(
+    parent_relid ANYELEMENT,
+    partition_relid ANYELEMENT)
+RETURNS ANYARRAY AS 'pathman', 'get_partition_range' LANGUAGE C STRICT;
+
+/*
+ * Copy rows to partitions
+ */
 CREATE OR REPLACE FUNCTION partition_data(p_parent text)
 RETURNS bigint AS
 $$
 DECLARE
     rec RECORD;
 BEGIN
     FOR rec IN  (SELECT child.relname, pg_constraint.consrc
-                   FROM pg_pathman_rels
-                   JOIN pg_class AS parent ON parent.relname = pg_pathman_rels.relname
-                   JOIN pg_inherits ON inhparent = parent.relfilenode
-                   JOIN pg_constraint ON conrelid = inhrelid AND contype='c'
-                   JOIN pg_class AS child ON child.relfilenode = inhrelid
-                   WHERE pg_pathman_rels.relname = p_parent)
+                 FROM pg_pathman_rels
+                 JOIN pg_class AS parent ON parent.relname = pg_pathman_rels.relname
+                 JOIN pg_inherits ON inhparent = parent.relfilenode
+                 JOIN pg_constraint ON conrelid = inhrelid AND contype='c'
+                 JOIN pg_class AS child ON child.relfilenode = inhrelid
+                 WHERE pg_pathman_rels.relname = p_parent)
     LOOP
-        RAISE NOTICE 'child %, condition %', rec.relname, rec.consrc;
-
+        RAISE NOTICE 'Copying data to % (condition: %)', rec.relname, rec.consrc;
         EXECUTE format('WITH part_data AS (
                             DELETE FROM ONLY %s WHERE %s RETURNING *)
                         INSERT INTO %s SELECT * FROM part_data'
diff --git a/contrib/pathman/sql/range.sql b/contrib/pathman/sql/range.sql
