Fix check-vanilla test failure

citusdata · Aug 24, 2023 · 993599a · 993599a
1 parent e700e1d
commit 993599a
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/backend/distributed/metadata/metadata_cache.c b/src/backend/distributed/metadata/metadata_cache.c
@@ -4781,6 +4781,20 @@ InvalidateDistRelationCacheCallback(Datum argument, Oid relationId)
 			InvalidateCitusTableCacheEntrySlot(cacheSlot);
 		}
 
+		/*
+		 * if pg_dist_* table relcache entries got invalidated due to direct operations on
+		 * them, invalidate the cached oids.
+		 * TODO: We should probably invalidate the other cached oids when related tables
+		 * got invalidated.
+		 */
+
+		if (relationId == MetadataCache.distPartitionRelationId)
+		{
+			MetadataCache.distPartitionRelationId = InvalidOid;
+			MetadataCache.distPartitionLogicalRelidIndexId = InvalidOid;
+			MetadataCache.distPartitionColocationidIndexId = InvalidOid;
+		}
+
 		if (relationId == MetadataCache.distObjectRelationId)
 		{
 			InvalidateDistObjectCache();