Fix temporary tablespaces for shared filesets some more.

tglsfdc · tglsfdc · commit f7b5988cc0d8 · 2020-07-03T17:01:34.000-04:00
Commit ecd9e9f fixed the problem in the wrong place, causing unwanted side-effects on the behavior of GetNextTempTableSpace(). Instead, let's make SharedFileSetInit() responsible for subbing in the value of MyDatabaseTableSpace when the default tablespace is called for. The convention about what is in the tempTableSpaces[] array is evidently insufficiently documented, so try to improve that. It also looks like SharedFileSetInit() is doing the wrong thing in the case where temp_tablespaces is empty. It was hard-wiring use of the pg_default tablespace, but it seems like using MyDatabaseTableSpace is more consistent with what happens for other temp files. Back-patch the reversion of PrepareTempTablespaces()'s behavior to 9.5, as ecd9e9f was. The changes in SharedFileSetInit() go back to v11 where that was introduced. (Note there is net zero code change before v11 from these two patch sets, so nothing to release-note.) Magnus Hagander and Tom Lane Discussion: https://postgr.es/m/CABUevExg5YEsOvqMxrjoNvb3ApVyH+9jggWGKwTDFyFCVWczGQ@mail.gmail.com
diff --git a/src/backend/commands/tablespace.c b/src/backend/commands/tablespace.c
@@ -1183,6 +1183,7 @@ GetDefaultTablespace(char relpersistence, bool partitioned)
 
 typedef struct
 {
+	/* Array of OIDs to be passed to SetTempTablespaces() */
 	int			numSpcs;
 	Oid			tblSpcs[FLEXIBLE_ARRAY_MEMBER];
 } temp_tablespaces_extra;
@@ -1232,6 +1233,7 @@ check_temp_tablespaces(char **newval, void **extra, GucSource source)
 			/* Allow an empty string (signifying database default) */
 			if (curname[0] == '\0')
 			{
+				/* InvalidOid signifies database's default tablespace */
 				tblSpcs[numSpcs++] = InvalidOid;
 				continue;
 			}
@@ -1258,6 +1260,7 @@ check_temp_tablespaces(char **newval, void **extra, GucSource source)
 			 */
 			if (curoid == MyDatabaseTableSpace)
 			{
+				/* InvalidOid signifies database's default tablespace */
 				tblSpcs[numSpcs++] = InvalidOid;
 				continue;
 			}
@@ -1368,6 +1371,7 @@ PrepareTempTablespaces(void)
 		/* Allow an empty string (signifying database default) */
 		if (curname[0] == '\0')
 		{
+			/* InvalidOid signifies database's default tablespace */
 			tblSpcs[numSpcs++] = InvalidOid;
 			continue;
 		}
@@ -1386,7 +1390,8 @@ PrepareTempTablespaces(void)
 		 */
 		if (curoid == MyDatabaseTableSpace)
 		{
-			tblSpcs[numSpcs++] = curoid;
+			/* InvalidOid signifies database's default tablespace */
+			tblSpcs[numSpcs++] = InvalidOid;
 			continue;
 		}
 
diff --git a/src/backend/storage/file/fd.c b/src/backend/storage/file/fd.c
@@ -264,8 +264,10 @@ static int	numExternalFDs = 0;
 static long tempFileCounter = 0;
 
 /*
- * Array of OIDs of temp tablespaces.  When numTempTableSpaces is -1,
- * this has not been set in the current transaction.
+ * Array of OIDs of temp tablespaces.  (Some entries may be InvalidOid,
+ * indicating that the current database's default tablespace should be used.)
+ * When numTempTableSpaces is -1, this has not been set in the current
+ * transaction.
  */
 static Oid *tempTableSpaces = NULL;
 static int	numTempTableSpaces = -1;
@@ -2779,6 +2781,9 @@ closeAllVfds(void)
  * unless this function is called again before then.  It is caller's
  * responsibility that the passed-in array has adequate lifespan (typically
  * it'd be allocated in TopTransactionContext).
+ *
+ * Some entries of the array may be InvalidOid, indicating that the current
+ * database's default tablespace should be used.
  */
 void
 SetTempTablespaces(Oid *tableSpaces, int numSpaces)
@@ -2818,7 +2823,10 @@ TempTablespacesAreSet(void)
  * GetTempTablespaces
  *
  * Populate an array with the OIDs of the tablespaces that should be used for
- * temporary files.  Return the number that were copied into the output array.
+ * temporary files.  (Some entries may be InvalidOid, indicating that the
+ * current database's default tablespace should be used.)  At most numSpaces
+ * entries will be filled.
+ * Returns the number of OIDs that were copied into the output array.
  */
 int
 GetTempTablespaces(Oid *tableSpaces, int numSpaces)
diff --git a/src/backend/storage/file/sharedfileset.c b/src/backend/storage/file/sharedfileset.c
@@ -63,9 +63,25 @@ SharedFileSetInit(SharedFileSet *fileset, dsm_segment *seg)
 						   lengthof(fileset->tablespaces));
 	if (fileset->ntablespaces == 0)
 	{
-		fileset->tablespaces[0] = DEFAULTTABLESPACE_OID;
+		/* If the GUC is empty, use current database's default tablespace */
+		fileset->tablespaces[0] = MyDatabaseTableSpace;
 		fileset->ntablespaces = 1;
 	}
+	else
+	{
+		int			i;
+
+		/*
+		 * An entry of InvalidOid means use the default tablespace for the
+		 * current database.  Replace that now, to be sure that all users of
+		 * the SharedFileSet agree on what to do.
+		 */
+		for (i = 0; i < fileset->ntablespaces; i++)
+		{
+			if (fileset->tablespaces[i] == InvalidOid)
+				fileset->tablespaces[i] = MyDatabaseTableSpace;
+		}
+	}
 
 	/* Register our cleanup callback. */
 	on_dsm_detach(seg, SharedFileSetOnDetach, PointerGetDatum(fileset));

Original file line number	Diff line number	Diff line change
`@@ -1183,6 +1183,7 @@ GetDefaultTablespace(char relpersistence, bool partitioned)`
`1183`	`1183`
`1184`	`1184`	`typedef struct`
`1185`	`1185`	`{`
	`1186`	`+ /* Array of OIDs to be passed to SetTempTablespaces() */`
`1186`	`1187`	`int numSpcs;`
`1187`	`1188`	`Oid tblSpcs[FLEXIBLE_ARRAY_MEMBER];`
`1188`	`1189`	`} temp_tablespaces_extra;`
`@@ -1232,6 +1233,7 @@ check_temp_tablespaces(char newval, void extra, GucSource source)`
`1232`	`1233`	`/* Allow an empty string (signifying database default) */`
`1233`	`1234`	`if (curname[0] == '\0')`
`1234`	`1235`	`{`
	`1236`	`+ /* InvalidOid signifies database's default tablespace */`
`1235`	`1237`	`tblSpcs[numSpcs++] = InvalidOid;`
`1236`	`1238`	`continue;`
`1237`	`1239`	`}`
`@@ -1258,6 +1260,7 @@ check_temp_tablespaces(char newval, void extra, GucSource source)`
`1258`	`1260`	`*/`
`1259`	`1261`	`if (curoid == MyDatabaseTableSpace)`
`1260`	`1262`	`{`
	`1263`	`+ /* InvalidOid signifies database's default tablespace */`
`1261`	`1264`	`tblSpcs[numSpcs++] = InvalidOid;`
`1262`	`1265`	`continue;`
`1263`	`1266`	`}`
`@@ -1368,6 +1371,7 @@ PrepareTempTablespaces(void)`
`1368`	`1371`	`/* Allow an empty string (signifying database default) */`
`1369`	`1372`	`if (curname[0] == '\0')`
`1370`	`1373`	`{`
	`1374`	`+ /* InvalidOid signifies database's default tablespace */`
`1371`	`1375`	`tblSpcs[numSpcs++] = InvalidOid;`
`1372`	`1376`	`continue;`
`1373`	`1377`	`}`
`@@ -1386,7 +1390,8 @@ PrepareTempTablespaces(void)`
`1386`	`1390`	`*/`
`1387`	`1391`	`if (curoid == MyDatabaseTableSpace)`
`1388`	`1392`	`{`
`1389`		`- tblSpcs[numSpcs++] = curoid;`
	`1393`	`+ /* InvalidOid signifies database's default tablespace */`
	`1394`	`+ tblSpcs[numSpcs++] = InvalidOid;`
`1390`	`1395`	`continue;`
`1391`	`1396`	`}`
`1392`	`1397`