external
/
citus
mirror of https://github.com/citusdata/citus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

improve & comment

tenant-schema-isolation-complete-view
Onur Tirtir 2023-10-16 16:42:59 +03:00
parent ac97d54515
commit db43b6fdce
1 changed files with 21 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
src/backend/distributed/operations/rebalancer_placement_separation.c
View File

@ -54,9 +54,20 @@ typedef struct NodeToPlacementGroupHashEntry
bool shouldHaveShards; bool shouldHaveShards;
/* /*
* Whether given node is allowed to separate any shardgroup placements. * Whether given node has some shard placements that cannot be moved away.
*
* For the nodes that this rebalancer-run is not allowed to move the
* placements away from, InitRebalancerPlacementSeparationContext() sets
* this to true if the node has some shard placements already. And if the
* node has a single shard placement group that needs a separate node, it
* also sets assignedPlacementGroup.
*
* We do so to prevent TryAssignPlacementGroupsToNodeGroups() making
* incorrect assignments later on.
*
* See InitRebalancerPlacementSeparationContext() for more details.
*/ */
bool allowedToSeparateAnyPlacementGroup; bool hasPlacementsThatCannotBeMovedAway;
/* /*
* Shardgroup placement that is assigned to this node to be separated * Shardgroup placement that is assigned to this node to be separated
@ -150,8 +161,7 @@ InitRebalancerPlacementSeparationContext(RebalancerPlacementSeparationContext *c
nodePlacementGroupHashEntry->shouldHaveShards = nodePlacementGroupHashEntry->shouldHaveShards =
workerNode->shouldHaveShards; workerNode->shouldHaveShards;
nodePlacementGroupHashEntry->allowedToSeparateAnyPlacementGroup = nodePlacementGroupHashEntry->hasPlacementsThatCannotBeMovedAway = false;
workerNode->shouldHaveShards;
nodePlacementGroupHashEntry->assignedPlacementGroup = NULL; nodePlacementGroupHashEntry->assignedPlacementGroup = NULL;
/* /*
@ -168,8 +178,8 @@ InitRebalancerPlacementSeparationContext(RebalancerPlacementSeparationContext *c
* For this reason, below we find out the assigned placement groups for * For this reason, below we find out the assigned placement groups for
* nodes of type S because we want to avoid from moving the placements * nodes of type S because we want to avoid from moving the placements
* (if any) from a node of type D to a node that is used to separate a * (if any) from a node of type D to a node that is used to separate a
* placement group within S. We also set allowedToSeparateAnyPlacementGroup * placement group within S. We also set hasPlacementsThatCannotBeMovedAway
* to false for the nodes that already have some shard placements within S * to true for the nodes that already have some shard placements within S
* because we want to avoid from moving the placements that need a separate * because we want to avoid from moving the placements that need a separate
* node (if any) from node D to node S. * node (if any) from node D to node S.
* *
@ -191,20 +201,12 @@ InitRebalancerPlacementSeparationContext(RebalancerPlacementSeparationContext *c
continue; continue;
} }
ShardgroupPlacement *separatedShardgroupPlacement = nodePlacementGroupHashEntry->hasPlacementsThatCannotBeMovedAway =
NodeGroupHasDistributedTableShardPlacements(
nodePlacementGroupHashEntry->nodeGroupId);
nodePlacementGroupHashEntry->assignedPlacementGroup =
NodeGroupGetSeparatedShardgroupPlacement( NodeGroupGetSeparatedShardgroupPlacement(
nodePlacementGroupHashEntry->nodeGroupId); nodePlacementGroupHashEntry->nodeGroupId);
if (separatedShardgroupPlacement)
{
nodePlacementGroupHashEntry->assignedPlacementGroup =
separatedShardgroupPlacement;
}
else
{
nodePlacementGroupHashEntry->allowedToSeparateAnyPlacementGroup =
!NodeGroupHasDistributedTableShardPlacements(
nodePlacementGroupHashEntry->nodeGroupId);
}
} }
} }
@ -331,7 +333,7 @@ TryAssignPlacementGroupToNodeGroup(RebalancerPlacementSeparationContext *context
return false; return false;
} }
if (!nodePlacementGroupHashEntry->allowedToSeparateAnyPlacementGroup) if (nodePlacementGroupHashEntry->hasPlacementsThatCannotBeMovedAway)
{ {
return false; return false;
} }