From 34883d6cc62393ff932f4d14fc519a8294fcdd5d Mon Sep 17 00:00:00 2001
From: eaydingol <60466783+eaydingol@users.noreply.github.com>
Date: Wed, 6 Mar 2024 14:46:49 +0300
Subject: [PATCH] Fix: store the previous shard cost for order verification
 (#7550)

Store the previous shard cost so that the invariant checking performs as
expected.
---
 src/backend/distributed/operations/shard_rebalancer.c | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/backend/distributed/operations/shard_rebalancer.c b/src/backend/distributed/operations/shard_rebalancer.c
index b1780b669..1d1119f74 100644
--- a/src/backend/distributed/operations/shard_rebalancer.c
+++ b/src/backend/distributed/operations/shard_rebalancer.c
@@ -384,6 +384,7 @@ CheckRebalanceStateInvariants(const RebalanceState *state)
 				Assert(shardCost->cost <= prevShardCost->cost);
 			}
 			totalCost += shardCost->cost;
+			prevShardCost = shardCost;
 		}
 
 		/* Check that utilization field is up to date. */