From e2a03ce6a03d18ff26b1409ef1d5f96a0ba9ebf6 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 653c7ce23..a4c7364be 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. */