restrict partitioning of temporary tables

funbringer · funbringer · commit 951cde131a12 · 2016-08-26T02:47:40.000+03:00
diff --git a/hash.sql b/hash.sql
@@ -69,6 +69,9 @@ BEGIN
 	PERFORM @extschema@.partition_data(parent_relid);
 
 	RETURN partitions_count;
+
+EXCEPTION WHEN others THEN
+	RAISE EXCEPTION '%', SQLERRM;
 END
 $$ LANGUAGE plpgsql
 SET client_min_messages = WARNING;
diff --git a/init.sql b/init.sql
@@ -88,8 +88,18 @@ $$
 DECLARE
 	v_rec			RECORD;
 	is_referenced	BOOLEAN;
+	rel_persistence	CHAR;
 
 BEGIN
+	/* Ignore temporary tables */
+	SELECT relpersistence FROM pg_catalog.pg_class
+	WHERE oid = p_relation INTO rel_persistence;
+
+	IF rel_persistence = 't'::CHAR THEN
+		RAISE EXCEPTION 'Temporary table "%" cannot be partitioned',
+			quote_ident(p_relation::TEXT);
+	END IF;
+
 	IF EXISTS (SELECT * FROM @extschema@.pathman_config
 			   WHERE partrel = p_relation) THEN
 		RAISE EXCEPTION 'Relation "%" has already been partitioned', p_relation;
diff --git a/range.sql b/range.sql
@@ -875,7 +875,7 @@ BEGIN
 	WHERE oid = p_partition INTO rel_persistence;
 
 	IF rel_persistence = 't'::CHAR THEN
-		RAISE EXCEPTION 'Temporary table \"%\" cannot be used as a partition',
+		RAISE EXCEPTION 'Temporary table "%" cannot be used as a partition',
 			quote_ident(p_partition::TEXT);
 	END IF;
 
