-
Notifications
You must be signed in to change notification settings - Fork 1.8k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[Segment Replication] Allocation and rebalancing based on average pri…
…mary shard count per index (#6422) * [Segment Replication] Move primary shard first during rebalancing Signed-off-by: Suraj Singh <surajrider@gmail.com> * Address review comments Signed-off-by: Suraj Singh <surajrider@gmail.com> * Add primary weight constraint Signed-off-by: Suraj Singh <surajrider@gmail.com> * Add average primary shard count constraint for allocation and rebalancing operation Signed-off-by: Suraj Singh <surajrider@gmail.com> * Spotless fix and javadocs Signed-off-by: Suraj Singh <surajrider@gmail.com> * Add failing tests Signed-off-by: Suraj Singh <surajrider@gmail.com> * Fix unit tests Signed-off-by: Suraj Singh <surajrider@gmail.com> * Add unit test for nodes breaching multiple constraints Signed-off-by: Suraj Singh <surajrider@gmail.com> * Add comments and update unit test Signed-off-by: Suraj Singh <surajrider@gmail.com> * Address review comments Signed-off-by: Suraj Singh <surajrider@gmail.com> * Remove auto-expand replicas integration test Signed-off-by: Suraj Singh <surajrider@gmail.com> * Address review comments Signed-off-by: Suraj Singh <surajrider@gmail.com> * Update comments for tests Signed-off-by: Suraj Singh <surajrider@gmail.com> * Add changelog entry Signed-off-by: Suraj Singh <surajrider@gmail.com> * Remove extra changelog entry Signed-off-by: Suraj Singh <surajrider@gmail.com> --------- Signed-off-by: Suraj Singh <surajrider@gmail.com> (cherry picked from commit f61402a) Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
- Loading branch information
1 parent
f4773b4
commit d2fce87
Showing
13 changed files
with
1,162 additions
and
76 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
216 changes: 216 additions & 0 deletions
216
...alClusterTest/java/org/opensearch/indices/replication/SegmentReplicationAllocationIT.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,216 @@ | ||
/* | ||
* SPDX-License-Identifier: Apache-2.0 | ||
* | ||
* The OpenSearch Contributors require contributions made to | ||
* this file be licensed under the Apache-2.0 license or a | ||
* compatible open source license. | ||
*/ | ||
|
||
package org.opensearch.indices.replication; | ||
|
||
import com.carrotsearch.hppc.cursors.ObjectObjectCursor; | ||
import org.opensearch.cluster.ClusterState; | ||
import org.opensearch.cluster.metadata.IndexMetadata; | ||
import org.opensearch.cluster.routing.IndexRoutingTable; | ||
import org.opensearch.cluster.routing.RoutingNode; | ||
import org.opensearch.cluster.routing.RoutingNodes; | ||
import org.opensearch.cluster.routing.ShardRouting; | ||
import org.opensearch.cluster.routing.allocation.allocator.BalancedShardsAllocator; | ||
import org.opensearch.common.settings.Settings; | ||
import org.opensearch.common.unit.TimeValue; | ||
import org.opensearch.index.IndexModule; | ||
import org.opensearch.indices.replication.common.ReplicationType; | ||
import org.opensearch.test.InternalTestCluster; | ||
import org.opensearch.test.OpenSearchIntegTestCase; | ||
|
||
import java.util.ArrayList; | ||
import java.util.List; | ||
import java.util.concurrent.TimeUnit; | ||
import java.util.stream.Collectors; | ||
|
||
import static org.opensearch.cluster.routing.ShardRoutingState.STARTED; | ||
import static org.opensearch.test.hamcrest.OpenSearchAssertions.assertAcked; | ||
|
||
import org.opensearch.cluster.OpenSearchAllocationTestCase.ShardAllocations; | ||
|
||
@OpenSearchIntegTestCase.ClusterScope(scope = OpenSearchIntegTestCase.Scope.TEST, numDataNodes = 0) | ||
public class SegmentReplicationAllocationIT extends SegmentReplicationBaseIT { | ||
|
||
private void createIndex(String idxName, int shardCount, int replicaCount, boolean isSegRep) { | ||
Settings.Builder builder = Settings.builder() | ||
.put(IndexMetadata.SETTING_NUMBER_OF_SHARDS, shardCount) | ||
.put(IndexModule.INDEX_QUERY_CACHE_ENABLED_SETTING.getKey(), false) | ||
.put(IndexMetadata.SETTING_NUMBER_OF_REPLICAS, replicaCount); | ||
if (isSegRep) { | ||
builder = builder.put(IndexMetadata.SETTING_REPLICATION_TYPE, ReplicationType.SEGMENT); | ||
} else { | ||
builder = builder.put(IndexMetadata.SETTING_REPLICATION_TYPE, ReplicationType.DOCUMENT); | ||
} | ||
prepareCreate(idxName, builder).get(); | ||
} | ||
|
||
public void enablePreferPrimaryBalance() { | ||
assertAcked( | ||
client().admin() | ||
.cluster() | ||
.prepareUpdateSettings() | ||
.setPersistentSettings( | ||
Settings.builder().put(BalancedShardsAllocator.PREFER_PER_INDEX_PRIMARY_SHARD_BALANCE.getKey(), "true") | ||
) | ||
); | ||
} | ||
|
||
/** | ||
* This test verifies the happy path where primary shard allocation is balanced when multiple indices are created. | ||
* | ||
* This test in general passes without primary shard balance as well due to nature of allocation algorithm which | ||
* assigns all primary shards first followed by replica copies. | ||
*/ | ||
public void testBalancedPrimaryAllocation() throws Exception { | ||
internalCluster().startClusterManagerOnlyNode(); | ||
final int maxReplicaCount = 2; | ||
final int maxShardCount = 5; | ||
final int nodeCount = randomIntBetween(maxReplicaCount + 1, 10); | ||
final int numberOfIndices = randomIntBetween(5, 10); | ||
|
||
final List<String> nodeNames = new ArrayList<>(); | ||
logger.info("--> Creating {} nodes", nodeCount); | ||
for (int i = 0; i < nodeCount; i++) { | ||
nodeNames.add(internalCluster().startNode()); | ||
} | ||
enablePreferPrimaryBalance(); | ||
int shardCount, replicaCount; | ||
ClusterState state; | ||
for (int i = 0; i < numberOfIndices; i++) { | ||
shardCount = randomIntBetween(1, maxShardCount); | ||
replicaCount = randomIntBetween(0, maxReplicaCount); | ||
createIndex("test" + i, shardCount, replicaCount, i % 2 == 0); | ||
logger.info("--> Creating index {} with shard count {} and replica count {}", "test" + i, shardCount, replicaCount); | ||
ensureGreen(TimeValue.timeValueSeconds(60)); | ||
state = client().admin().cluster().prepareState().execute().actionGet().getState(); | ||
logger.info(ShardAllocations.printShardDistribution(state)); | ||
} | ||
verifyPerIndexPrimaryBalance(); | ||
} | ||
|
||
/** | ||
* This test verifies balanced primary shard allocation for a single index with large shard count in event of node | ||
* going down and a new node joining the cluster. The results in shard distribution skewness and re-balancing logic | ||
* ensures the primary shard distribution is balanced. | ||
* | ||
*/ | ||
public void testSingleIndexShardAllocation() throws Exception { | ||
internalCluster().startClusterManagerOnlyNode(); | ||
final int maxReplicaCount = 1; | ||
final int maxShardCount = 50; | ||
final int nodeCount = 5; | ||
|
||
final List<String> nodeNames = new ArrayList<>(); | ||
logger.info("--> Creating {} nodes", nodeCount); | ||
for (int i = 0; i < nodeCount; i++) { | ||
nodeNames.add(internalCluster().startNode()); | ||
} | ||
enablePreferPrimaryBalance(); | ||
|
||
ClusterState state; | ||
createIndex("test", maxShardCount, maxReplicaCount, true); | ||
ensureGreen(TimeValue.timeValueSeconds(60)); | ||
state = client().admin().cluster().prepareState().execute().actionGet().getState(); | ||
logger.info(ShardAllocations.printShardDistribution(state)); | ||
verifyPerIndexPrimaryBalance(); | ||
|
||
// Remove a node | ||
internalCluster().stopRandomNode(InternalTestCluster.nameFilter(nodeNames.get(0))); | ||
ensureGreen(TimeValue.timeValueSeconds(60)); | ||
state = client().admin().cluster().prepareState().execute().actionGet().getState(); | ||
logger.info(ShardAllocations.printShardDistribution(state)); | ||
verifyPerIndexPrimaryBalance(); | ||
|
||
// Add a new node | ||
internalCluster().startDataOnlyNode(); | ||
ensureGreen(TimeValue.timeValueSeconds(60)); | ||
state = client().admin().cluster().prepareState().execute().actionGet().getState(); | ||
logger.info(ShardAllocations.printShardDistribution(state)); | ||
verifyPerIndexPrimaryBalance(); | ||
} | ||
|
||
/** | ||
* Similar to testSingleIndexShardAllocation test but creates multiple indices, multiple node adding in and getting | ||
* removed. The test asserts post each such event that primary shard distribution is balanced across single index. | ||
*/ | ||
public void testAllocationWithDisruption() throws Exception { | ||
internalCluster().startClusterManagerOnlyNode(); | ||
final int maxReplicaCount = 2; | ||
final int maxShardCount = 5; | ||
final int nodeCount = randomIntBetween(maxReplicaCount + 1, 10); | ||
final int numberOfIndices = randomIntBetween(1, 10); | ||
|
||
logger.info("--> Creating {} nodes", nodeCount); | ||
final List<String> nodeNames = new ArrayList<>(); | ||
for (int i = 0; i < nodeCount; i++) { | ||
nodeNames.add(internalCluster().startNode()); | ||
} | ||
enablePreferPrimaryBalance(); | ||
|
||
int shardCount, replicaCount, totalShardCount = 0, totalReplicaCount = 0; | ||
ClusterState state; | ||
for (int i = 0; i < numberOfIndices; i++) { | ||
shardCount = randomIntBetween(1, maxShardCount); | ||
totalShardCount += shardCount; | ||
replicaCount = randomIntBetween(1, maxReplicaCount); | ||
totalReplicaCount += replicaCount; | ||
logger.info("--> Creating index test{} with primary {} and replica {}", i, shardCount, replicaCount); | ||
createIndex("test" + i, shardCount, replicaCount, i % 2 == 0); | ||
ensureGreen(TimeValue.timeValueSeconds(60)); | ||
if (logger.isTraceEnabled()) { | ||
state = client().admin().cluster().prepareState().execute().actionGet().getState(); | ||
logger.info(ShardAllocations.printShardDistribution(state)); | ||
} | ||
} | ||
state = client().admin().cluster().prepareState().execute().actionGet().getState(); | ||
logger.info(ShardAllocations.printShardDistribution(state)); | ||
verifyPerIndexPrimaryBalance(); | ||
|
||
final int additionalNodeCount = randomIntBetween(1, 5); | ||
logger.info("--> Adding {} nodes", additionalNodeCount); | ||
|
||
internalCluster().startNodes(additionalNodeCount); | ||
ensureGreen(TimeValue.timeValueSeconds(60)); | ||
state = client().admin().cluster().prepareState().execute().actionGet().getState(); | ||
logger.info(ShardAllocations.printShardDistribution(state)); | ||
verifyPerIndexPrimaryBalance(); | ||
|
||
logger.info("--> Stop one third nodes"); | ||
for (int i = 0; i < nodeCount; i += 3) { | ||
internalCluster().stopRandomNode(InternalTestCluster.nameFilter(nodeNames.get(i))); | ||
// give replica a chance to promote as primary before terminating node containing the replica | ||
ensureGreen(TimeValue.timeValueSeconds(60)); | ||
} | ||
state = client().admin().cluster().prepareState().execute().actionGet().getState(); | ||
logger.info(ShardAllocations.printShardDistribution(state)); | ||
verifyPerIndexPrimaryBalance(); | ||
} | ||
|
||
/** | ||
* Utility method which ensures cluster has balanced primary shard distribution across a single index. | ||
* @throws Exception | ||
*/ | ||
private void verifyPerIndexPrimaryBalance() throws Exception { | ||
assertBusy(() -> { | ||
final ClusterState currentState = client().admin().cluster().prepareState().execute().actionGet().getState(); | ||
RoutingNodes nodes = currentState.getRoutingNodes(); | ||
for (ObjectObjectCursor<String, IndexRoutingTable> index : currentState.getRoutingTable().indicesRouting()) { | ||
final int totalPrimaryShards = index.value.primaryShardsActive(); | ||
final int avgPrimaryShardsPerNode = (int) Math.ceil(totalPrimaryShards * 1f / currentState.getRoutingNodes().size()); | ||
for (RoutingNode node : nodes) { | ||
final int primaryCount = node.shardsWithState(index.key, STARTED) | ||
.stream() | ||
.filter(ShardRouting::primary) | ||
.collect(Collectors.toList()) | ||
.size(); | ||
assertTrue(primaryCount <= avgPrimaryShardsPerNode); | ||
} | ||
} | ||
}, 60, TimeUnit.SECONDS); | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.