From 424e23386e77a215e7300d5fcaa3c664d5a2ead3 Mon Sep 17 00:00:00 2001
From: EmelSimsek <emelhasdal@gmail.com>
Date: Thu, 24 Aug 2023 15:20:38 +0300
Subject: [PATCH] Fix check-validate failure

---
 .../distributed/metadata/metadata_cache.c     | 76 +++++++++++++++++++
 1 file changed, 76 insertions(+)

diff --git a/src/backend/distributed/metadata/metadata_cache.c b/src/backend/distributed/metadata/metadata_cache.c
index e6e8debd3b9..2bb88737ab0 100644
--- a/src/backend/distributed/metadata/metadata_cache.c
+++ b/src/backend/distributed/metadata/metadata_cache.c
@@ -4795,9 +4795,85 @@ InvalidateDistRelationCacheCallback(Datum argument, Oid relationId)
 			MetadataCache.distPartitionColocationidIndexId = InvalidOid;
 		}
 
+		if (relationId == MetadataCache.distNodeRelationId)
+		{
+			MetadataCache.distNodeRelationId = InvalidOid;
+			MetadataCache.distNodeNodeIdIndexId = InvalidOid;
+		}
+
+		if (relationId == MetadataCache.distLocalGroupRelationId)
+		{
+			MetadataCache.distLocalGroupRelationId = InvalidOid;
+		}
+
+		if (relationId == MetadataCache.distCleanupRelationId)
+		{
+			MetadataCache.distCleanupRelationId = InvalidOid;
+			MetadataCache.distCleanupPrimaryKeyIndexId = InvalidOid;
+		}
+
+		if (relationId == MetadataCache.distColocationRelationId)
+		{
+			MetadataCache.distColocationRelationId = InvalidOid;
+			MetadataCache.distColocationConfigurationIndexId = InvalidOid;
+		}
+
+		if (relationId == Metadata.distTenantSchemaRelationId)
+		{
+			Metadata.distTenantSchemaRelationId = InvalidOid;
+		}
+
+		if (relationId == MetadataCache.distShardLogicalRelidIndexId)
+		{
+			MetadataCache.distShardLogicalRelidIndexId = InvalidOid;
+			MetadataCache.distShardShardidIndexId = InvalidOid;
+		}
+
+		if (relationId == MetadataCache.distPlacementShardidIndexId)
+		{
+			MetadataCache.distPlacementShardidIndexId = InvalidOid;
+			MetadataCache.distPlacementPlacementidIndexId = InvalidOid;
+			MetadataCache.distPlacementGroupidIndexId = InvalidOid;
+		}
+
+		if (relationId == MetadataCache.distColocationidIndexId)
+		{
+			MetadataCache.distColocationidIndexId = InvalidOid;
+		}
+
+		if (relationId == MetadataCache.distTransactionRelationId)
+		{
+			MetadataCache.distTransactionRelationId = InvalidOid;
+			MetadataCache.distTransactionGroupIndexId = InvalidOid;
+		}
+
+		if (relationId == MetadataCache.distTenantSchemaPrimaryKeyIndexId)
+		{
+			MetadataCache.distTenantSchemaPrimaryKeyIndexId = InvalidOid;
+			MetadataCache.distTenantSchemaUniqueColocationIdIndexId = InvalidOid;
+		}
+		if (relationId == MetadataCache.distAuthinfoRelationId)
+		{
+			MetadataCache.distAuthinfoRelationId = InvalidOid;
+			MetadataCache.distAuthinfoIndexId = InvalidOid;
+		}
+
+		if (relationId == MetadataCache.distPoolinfoRelationId)
+		{
+			MetadataCache.distPoolinfoRelationId = InvalidOid;
+			MetadataCache.distPoolinfoIndexId = InvalidOid;
+		}
+
+		if (relationId == MetadataCache.distClockLogicalSequenceId)
+		{
+			MetadataCache.distClockLogicalSequenceId = InvalidOid;
+		}
+
 		if (relationId == MetadataCache.distObjectRelationId)
 		{
 			InvalidateDistObjectCache();
+			MetadataCache.distObjectRelationId = InvalidOid;
+			MetadataCache.distObjectPrimaryKeyIndexId = InvalidOid;
 		}
 	}
 }