pathman: return partitions count

zilder · zilder · commit 0b3ae134b394 · 2016-01-27T12:19:27.000+03:00
diff --git a/contrib/pg_pathman/sql/hash.sql b/contrib/pg_pathman/sql/hash.sql
@@ -5,13 +5,13 @@ CREATE OR REPLACE FUNCTION @extschema@.create_hash_partitions(
     relation TEXT
     , attribute TEXT
     , partitions_count INTEGER
-) RETURNS VOID AS
+) RETURNS INTEGER AS
 $$
 BEGIN
     relation := @extschema@.validate_relname(relation);
 
     IF EXISTS (SELECT * FROM @extschema@.pathman_config WHERE relname = relation) THEN
-        RAISE EXCEPTION 'Reltion "%s" has already been partitioned', relation;
+        RAISE EXCEPTION 'Relation "%s" has already been partitioned', relation;
     END IF;
 
     /* Create partitions and update pg_pathman configuration */
@@ -42,6 +42,8 @@ BEGIN
 
     /* Notify backend about changes */
     PERFORM @extschema@.on_create_partitions(relation::regclass::oid);
+
+    RETURN partitions_count;
 END
 $$ LANGUAGE plpgsql;
 
diff --git a/contrib/pg_pathman/sql/range.sql b/contrib/pg_pathman/sql/range.sql
@@ -7,7 +7,7 @@ CREATE OR REPLACE FUNCTION @extschema@.create_range_partitions(
     , p_start_value ANYELEMENT
     , p_interval INTERVAL
     , p_premake INTEGER)
-RETURNS VOID AS
+RETURNS INTEGER AS
 $$
 DECLARE
     v_value     TEXT;
@@ -16,7 +16,7 @@ BEGIN
     p_relation := @extschema@.validate_relname(p_relation);
 
     IF EXISTS (SELECT * FROM @extschema@.pathman_config WHERE relname = p_relation) THEN
-        RAISE EXCEPTION 'Reltion "%" has already been partitioned', p_relation;
+        RAISE EXCEPTION 'Relation "%" has already been partitioned', p_relation;
     END IF;
 
     EXECUTE format('DROP SEQUENCE IF EXISTS %s_seq', p_relation);
@@ -39,6 +39,8 @@ BEGIN
     -- PERFORM create_hash_update_trigger(relation, attribute, partitions_count);
     /* Notify backend about changes */
     PERFORM @extschema@.on_create_partitions(p_relation::regclass::oid);
+
+    RETURN p_premake+1;
 END
 $$ LANGUAGE plpgsql;
 
@@ -51,7 +53,7 @@ CREATE OR REPLACE FUNCTION @extschema@.create_range_partitions(
     , p_start_value ANYELEMENT
     , p_interval ANYELEMENT
     , p_premake INTEGER)
-RETURNS VOID AS
+RETURNS INTEGER AS
 $$
 DECLARE
     v_value     TEXT;
@@ -60,7 +62,7 @@ BEGIN
     p_relation := @extschema@.validate_relname(p_relation);
 
     IF EXISTS (SELECT * FROM @extschema@.pathman_config WHERE relname = p_relation) THEN
-        RAISE EXCEPTION 'Reltion "%" has already been partitioned', p_relation;
+        RAISE EXCEPTION 'Relation "%" has already been partitioned', p_relation;
     END IF;
 
     EXECUTE format('DROP SEQUENCE IF EXISTS %s_seq', p_relation);
@@ -83,6 +85,8 @@ BEGIN
     -- PERFORM create_hash_update_trigger(relation, attribute, partitions_count);
     /* Notify backend about changes */
     PERFORM @extschema@.on_create_partitions(p_relation::regclass::oid);
+
+    RETURN p_premake+1;
 END
 $$ LANGUAGE plpgsql;
 
