Add flag IsBgWorker to create_partitions_for_value_internal function

Author: Maksim Milyutin

src/partition_creation.c

@@ -270,7 +270,8 @@ create_partitions_for_value(Oid relid, Datum value, Oid value_type)
 			elog(DEBUG2, "create_partitions(): chose backend [%u]", MyProcPid);
 			last_partition = create_partitions_for_value_internal(relid,
 																  value,
-																  value_type);
+																  value_type,
+																  false);
 		}
 	}
 	else
@@ -299,7 +300,7 @@ create_partitions_for_value(Oid relid, Datum value, Oid value_type)
  * use create_partitions_for_value() instead.
  */
 Oid
-create_partitions_for_value_internal(Oid relid, Datum value, Oid value_type)
+create_partitions_for_value_internal(Oid relid, Datum value, Oid value_type, bool IsBgWorker)
 {
 	MemoryContext	old_mcxt = CurrentMemoryContext;
 	Oid				partid = InvalidOid; /* last created partition (or InvalidOid) */
@@ -405,7 +406,7 @@ create_partitions_for_value_internal(Oid relid, Datum value, Oid value_type)
 		ErrorData *edata;
 
 		/* Simply rethrow ERROR if we're in backend */
-		if (!IsBackgroundWorker)
+		if (!IsBgWorker)
 			PG_RE_THROW();
 
 		/* Switch to the original context & copy edata */

src/partition_creation.h

@@ -24,7 +24,7 @@
 
 /* Create RANGE partitions to store some value */
 Oid create_partitions_for_value(Oid relid, Datum value, Oid value_type);
-Oid create_partitions_for_value_internal(Oid relid, Datum value, Oid value_type);
+Oid create_partitions_for_value_internal(Oid relid, Datum value, Oid value_type, bool IsBgWorker);
 
 
 /* Create one RANGE partition */

src/pathman_workers.c

@@ -366,8 +366,8 @@ bgw_main_spawn_partitions(Datum main_arg)
 							   args->parallel_master_pid))
 		return;
 #endif
-
 	/* Establish connection and start transaction */
+
 	BackgroundWorkerInitializeConnectionByOid(args->dbid, args->userid);
 
 	/* Start new transaction (syscache access etc.) */
@@ -392,7 +392,8 @@ bgw_main_spawn_partitions(Datum main_arg)
 	/* Create partitions and save the Oid of the last one */
 	args->result = create_partitions_for_value_internal(args->partitioned_table,
 														value, /* unpacked Datum */
-														args->value_type);
+														args->value_type,
+														true); /* run under background woker */
 
 	/* Finish transaction in an appropriate way */
 	if (args->result == InvalidOid)