From d9f872058fdee5a9b2ee87525d76d2802bdc7c89 Mon Sep 17 00:00:00 2001
From: naisila <nicypp@gmail.com>
Date: Wed, 17 Sep 2025 01:35:39 +0300
Subject: [PATCH] Fix multi_partitioning crash

---
 src/backend/distributed/commands/index.c | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/backend/distributed/commands/index.c b/src/backend/distributed/commands/index.c
index 1401da0a6..d95c53fb5 100644
--- a/src/backend/distributed/commands/index.c
+++ b/src/backend/distributed/commands/index.c
@@ -854,8 +854,11 @@ PostprocessIndexStmt(Node *node, const char *queryString)
 	table_close(relation, NoLock);
 	index_close(indexRelation, NoLock);
 
+	PushActiveSnapshot(GetTransactionSnapshot());
+
 	/* mark index as invalid, in-place (cannot be rolled back) */
 	index_set_state_flags(indexRelationId, INDEX_DROP_CLEAR_VALID);
+	PopActiveSnapshot();
 
 	/* re-open a transaction command from here on out */
 	CommitTransactionCommand();
@@ -1370,8 +1373,11 @@ MarkIndexValid(IndexStmt *indexStmt)
 											schemaId);
 	Relation indexRelation = index_open(indexRelationId, RowExclusiveLock);
 
+	PushActiveSnapshot(GetTransactionSnapshot());
+
 	/* mark index as valid, in-place (cannot be rolled back) */
 	index_set_state_flags(indexRelationId, INDEX_CREATE_SET_VALID);
+	PopActiveSnapshot();
 
 	table_close(relation, NoLock);
 	index_close(indexRelation, NoLock);