Add disabling of VACUUM FULL to custom TOASTers

Nikita Glukhov · Nikita Glukhov · commit f834d3e9a713 · 2023-01-12T02:12:43.000+03:00
diff --git a/src/backend/commands/vacuum.c b/src/backend/commands/vacuum.c
@@ -32,6 +32,7 @@
 #include "access/htup_details.h"
 #include "access/multixact.h"
 #include "access/tableam.h"
+#include "access/toasterapi.h"
 #include "access/transam.h"
 #include "access/xact.h"
 #include "catalog/namespace.h"
@@ -1832,6 +1833,76 @@ vac_truncate_clog(TransactionId frozenXID,
 	LWLockRelease(WrapLimitsVacuumLock);
 }
 
+static Oid
+get_main_rel_for_toast_rel(Oid toastrelid)
+{
+	Relation	class_rel = table_open(RelationRelationId, AccessShareLock);
+	SysScanDesc scan = systable_beginscan(class_rel, InvalidOid, false,
+										  NULL, 0, NULL);
+	HeapTuple	tup;
+	Oid			relid = InvalidOid;
+
+	while ((tup = systable_getnext(scan)) != NULL)
+	{
+		Form_pg_class relform = (Form_pg_class) GETSTRUCT(tup);
+
+		if (relform->reltoastrelid == toastrelid)
+		{
+			relid = relform->oid;
+			break;
+		}
+	}
+
+	systable_endscan(scan);
+	table_close(class_rel, NoLock);
+
+	if (!OidIsValid(relid))
+		elog(ERROR, "could not find main relation for TOAST relation %u",
+			 toastrelid);
+
+	return relid;
+}
+
+static bool
+toastrel_vacuum_full_is_disabled(Relation toastrel, VacuumParams *params)
+{
+	Oid			mainrelid = get_main_rel_for_toast_rel(RelationGetRelid(toastrel));
+	Relation	mainrel = vacuum_open_relation(mainrelid, NULL, params->options,
+											   params->log_min_duration >= 0,
+											   AccessShareLock);
+	TupleDesc	maindesc;
+	bool		res = false;
+
+	if (!mainrel)
+		return true;
+
+	maindesc = RelationGetDescr(mainrel);
+
+	for (int i = 0; i < maindesc->natts; i++)
+	{
+		Oid			toasterid = TupleDescAttr(maindesc, i)->atttoaster;
+
+		if (OidIsValid(toasterid))
+		{
+			TsrRoutine *toaster = SearchTsrCache(toasterid);
+
+			if (toaster->relinfo &&
+				(toaster->relinfo(toastrel) & TOASTREL_VACUUM_FULL_DISABLED))
+			{
+				ereport(WARNING,
+						(errmsg("skipping \"%s\" --- %s is disabled by toaster",
+								RelationGetRelationName(toastrel),
+								"VACUUM FULL")));
+				res = true;
+				break;
+			}
+		}
+	}
+
+	relation_close(mainrel, AccessShareLock);
+
+	return res;
+}
 
 /*
  *	vacuum_rel() -- vacuum one heap relation
@@ -1997,6 +2068,20 @@ vacuum_rel(Oid relid, RangeVar *relation, VacuumParams *params)
 		return true;
 	}
 
+	/*
+	 * Check if VACUUM FULL of TOAST relation is disabled by the
+	 * toasters.
+	 */
+	if (rel->rd_rel->relkind == RELKIND_TOASTVALUE &&
+		(params->options & VACOPT_FULL) != 0 &&
+		toastrel_vacuum_full_is_disabled(rel, params))
+	{
+		relation_close(rel, lmode);
+		PopActiveSnapshot();
+		CommitTransactionCommand();
+		return false;
+	}
+
 	/*
 	 * Get a session-level lock too. This will protect our access to the
 	 * relation across multiple transactions, so that we can vacuum the
diff --git a/src/include/access/toasterapi.h b/src/include/access/toasterapi.h
@@ -92,6 +92,9 @@ typedef bool (*toastervalidate_function) (Oid typeoid,
 										  char storage, char compression,
 										  Oid amoid, bool false_ok);
 
+#define TOASTREL_VACUUM_FULL_DISABLED 0x01
+typedef int (*toast_rel_info_function)(Relation toast_rel);
+
 /*
  * API struct for Toaster.  Note this must be stored in a single palloc'd
  * chunk of memory.
@@ -109,6 +112,7 @@ typedef struct TsrRoutine
 	del_toast_function deltoast;
 	get_vtable_function get_vtable;
 	toastervalidate_function toastervalidate;
+	toast_rel_info_function relinfo;
 } TsrRoutine;
 
 
