From 5a941814fd9ff22162e582abbb666f4e5317ad8b Mon Sep 17 00:00:00 2001 From: SaitTalhaNisanci Date: Mon, 10 May 2021 16:57:19 +0300 Subject: [PATCH] Close connection after each shard move (#4967) --- src/backend/distributed/operations/shard_rebalancer.c | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/backend/distributed/operations/shard_rebalancer.c b/src/backend/distributed/operations/shard_rebalancer.c index 64426d4bd..8fd3860ac 100644 --- a/src/backend/distributed/operations/shard_rebalancer.c +++ b/src/backend/distributed/operations/shard_rebalancer.c @@ -1117,7 +1117,6 @@ UpdateShardPlacement(PlacementUpdateEvent *placementUpdateEvent, WorkerNode *sourceNode = placementUpdateEvent->sourceNode; WorkerNode *targetNode = placementUpdateEvent->targetNode; const char *doRepair = "false"; - int connectionFlag = FORCE_NEW_CONNECTION; Datum shardTranferModeLabelDatum = DirectFunctionCall1(enum_out, shardReplicationModeOid); @@ -1189,6 +1188,7 @@ UpdateShardPlacement(PlacementUpdateEvent *placementUpdateEvent, sourceNode->workerPort, REBALANCE_PROGRESS_MOVING); + int connectionFlag = FORCE_NEW_CONNECTION; MultiConnection *connection = GetNodeConnection(connectionFlag, LOCAL_HOST_NAME, PostPortNumber); @@ -1202,6 +1202,7 @@ UpdateShardPlacement(PlacementUpdateEvent *placementUpdateEvent, sourceNode->workerName, sourceNode->workerPort, REBALANCE_PROGRESS_MOVED); + CloseConnection(connection); return true; }