forked from elastic/elasticsearch
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Put a fake allocation id on allocate stale primary command (elastic#3…
…4140) removes fake allocation id after recovery is done Relates to elastic#33432
- Loading branch information
1 parent
b307a4e
commit 7f38c83
Showing
8 changed files
with
352 additions
and
40 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
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
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
238 changes: 238 additions & 0 deletions
238
server/src/test/java/org/elasticsearch/cluster/routing/AllocationIdIT.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,238 @@ | ||
/* | ||
* Licensed to Elasticsearch under one or more contributor | ||
* license agreements. See the NOTICE file distributed with | ||
* this work for additional information regarding copyright | ||
* ownership. Elasticsearch licenses this file to you under | ||
* the Apache License, Version 2.0 (the "License"); you may | ||
* not use this file except in compliance with the License. | ||
* You may obtain a copy of the License at | ||
* | ||
* http://www.apache.org/licenses/LICENSE-2.0 | ||
* | ||
* Unless required by applicable law or agreed to in writing, | ||
* software distributed under the License is distributed on an | ||
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY | ||
* KIND, either express or implied. See the License for the | ||
* specific language governing permissions and limitations | ||
* under the License. | ||
*/ | ||
|
||
package org.elasticsearch.cluster.routing; | ||
|
||
import org.apache.lucene.store.SimpleFSDirectory; | ||
import org.elasticsearch.action.admin.cluster.allocation.ClusterAllocationExplanation; | ||
import org.elasticsearch.action.admin.indices.stats.ShardStats; | ||
import org.elasticsearch.action.index.IndexRequestBuilder; | ||
import org.elasticsearch.client.Requests; | ||
import org.elasticsearch.cluster.ClusterState; | ||
import org.elasticsearch.cluster.health.ClusterHealthStatus; | ||
import org.elasticsearch.cluster.metadata.IndexMetaData; | ||
import org.elasticsearch.cluster.routing.allocation.AllocationDecision; | ||
import org.elasticsearch.cluster.routing.allocation.ShardAllocationDecision; | ||
import org.elasticsearch.cluster.routing.allocation.command.AllocateStalePrimaryAllocationCommand; | ||
import org.elasticsearch.common.settings.Settings; | ||
import org.elasticsearch.index.IndexService; | ||
import org.elasticsearch.index.IndexSettings; | ||
import org.elasticsearch.index.MockEngineFactoryPlugin; | ||
import org.elasticsearch.index.engine.Engine; | ||
import org.elasticsearch.index.shard.RemoveCorruptedShardDataCommandIT; | ||
import org.elasticsearch.index.shard.ShardId; | ||
import org.elasticsearch.index.shard.ShardPath; | ||
import org.elasticsearch.index.store.Store; | ||
import org.elasticsearch.indices.IndicesService; | ||
import org.elasticsearch.plugins.Plugin; | ||
import org.elasticsearch.test.DummyShardLock; | ||
import org.elasticsearch.test.ESIntegTestCase; | ||
import org.elasticsearch.test.InternalSettingsPlugin; | ||
import org.elasticsearch.test.InternalTestCluster; | ||
import org.elasticsearch.test.transport.MockTransportService; | ||
|
||
import java.io.IOException; | ||
import java.nio.file.Path; | ||
import java.util.Arrays; | ||
import java.util.Collection; | ||
import java.util.Set; | ||
import java.util.concurrent.ExecutionException; | ||
import java.util.stream.Collectors; | ||
|
||
import static org.elasticsearch.index.query.QueryBuilders.matchAllQuery; | ||
import static org.elasticsearch.test.hamcrest.ElasticsearchAssertions.assertHitCount; | ||
import static org.hamcrest.Matchers.equalTo; | ||
import static org.hamcrest.Matchers.greaterThan; | ||
import static org.hamcrest.Matchers.hasSize; | ||
import static org.hamcrest.Matchers.is; | ||
import static org.hamcrest.Matchers.not; | ||
|
||
@ESIntegTestCase.ClusterScope(scope = ESIntegTestCase.Scope.SUITE, numDataNodes = 0) | ||
public class AllocationIdIT extends ESIntegTestCase { | ||
|
||
@Override | ||
protected Collection<Class<? extends Plugin>> nodePlugins() { | ||
return Arrays.asList(MockTransportService.TestPlugin.class, MockEngineFactoryPlugin.class, InternalSettingsPlugin.class); | ||
} | ||
|
||
public void testFailedRecoveryOnAllocateStalePrimaryRequiresAnotherAllocateStalePrimary() throws Exception { | ||
/* | ||
* Allocation id is put on start of shard while historyUUID is adjusted after recovery is done. | ||
* | ||
* If during execution of AllocateStalePrimary a proper allocation id is stored in allocation id set and recovery is failed | ||
* shard restart skips the stage where historyUUID is changed. | ||
* | ||
* That leads to situation where allocated stale primary and its replica belongs to the same historyUUID and | ||
* replica will receive operations after local checkpoint while documents before checkpoints could be significant different. | ||
* | ||
* Therefore, on AllocateStalePrimary we put some fake allocation id (no real one could be generated like that) | ||
* and any failure during recovery requires extra AllocateStalePrimary command to be executed. | ||
*/ | ||
|
||
// initial set up | ||
final String indexName = "index42"; | ||
final String master = internalCluster().startMasterOnlyNode(); | ||
String node1 = internalCluster().startNode(); | ||
createIndex(indexName, Settings.builder() | ||
.put(IndexMetaData.SETTING_NUMBER_OF_SHARDS, 1) | ||
.put(IndexMetaData.SETTING_NUMBER_OF_REPLICAS, 1) | ||
.put(IndexSettings.INDEX_CHECK_ON_STARTUP.getKey(), "checksum").build()); | ||
final int numDocs = indexDocs(indexName, "foo", "bar"); | ||
final IndexSettings indexSettings = getIndexSettings(indexName, node1); | ||
final Set<String> allocationIds = getAllocationIds(indexName); | ||
final ShardId shardId = new ShardId(resolveIndex(indexName), 0); | ||
final Path indexPath = getIndexPath(node1, shardId); | ||
assertThat(allocationIds, hasSize(1)); | ||
final String historyUUID = historyUUID(node1, indexName); | ||
String node2 = internalCluster().startNode(); | ||
ensureGreen(indexName); | ||
internalCluster().assertSameDocIdsOnShards(); | ||
// initial set up is done | ||
|
||
internalCluster().stopRandomNode(InternalTestCluster.nameFilter(node1)); | ||
|
||
// index more docs to node2 that marks node1 as stale | ||
int numExtraDocs = indexDocs(indexName, "foo", "bar2"); | ||
assertHitCount(client(node2).prepareSearch(indexName).setQuery(matchAllQuery()).get(), numDocs + numExtraDocs); | ||
|
||
internalCluster().stopRandomNode(InternalTestCluster.nameFilter(node2)); | ||
|
||
// create fake corrupted marker on node1 | ||
putFakeCorruptionMarker(indexSettings, shardId, indexPath); | ||
|
||
// thanks to master node1 is out of sync | ||
node1 = internalCluster().startNode(); | ||
|
||
// there is only _stale_ primary | ||
checkNoValidShardCopy(indexName, shardId); | ||
|
||
// allocate stale primary | ||
client(node1).admin().cluster().prepareReroute() | ||
.add(new AllocateStalePrimaryAllocationCommand(indexName, 0, node1, true)) | ||
.get(); | ||
|
||
// allocation fails due to corruption marker | ||
assertBusy(() -> { | ||
final ClusterState state = client().admin().cluster().prepareState().get().getState(); | ||
final ShardRouting shardRouting = state.routingTable().index(indexName).shard(shardId.id()).primaryShard(); | ||
assertThat(shardRouting.state(), equalTo(ShardRoutingState.UNASSIGNED)); | ||
assertThat(shardRouting.unassignedInfo().getReason(), equalTo(UnassignedInfo.Reason.ALLOCATION_FAILED)); | ||
}); | ||
|
||
try(Store store = new Store(shardId, indexSettings, new SimpleFSDirectory(indexPath), new DummyShardLock(shardId))) { | ||
store.removeCorruptionMarker(); | ||
} | ||
|
||
// index is red: no any shard is allocated (allocation id is a fake id that does not match to anything) | ||
checkHealthStatus(indexName, ClusterHealthStatus.RED); | ||
checkNoValidShardCopy(indexName, shardId); | ||
|
||
internalCluster().restartNode(node1, InternalTestCluster.EMPTY_CALLBACK); | ||
|
||
// index is still red due to mismatch of allocation id | ||
checkHealthStatus(indexName, ClusterHealthStatus.RED); | ||
checkNoValidShardCopy(indexName, shardId); | ||
|
||
// no any valid shard is there; have to invoke AllocateStalePrimary again | ||
client().admin().cluster().prepareReroute() | ||
.add(new AllocateStalePrimaryAllocationCommand(indexName, 0, node1, true)) | ||
.get(); | ||
|
||
ensureYellow(indexName); | ||
|
||
// bring node2 back | ||
node2 = internalCluster().startNode(); | ||
ensureGreen(indexName); | ||
|
||
assertThat(historyUUID(node1, indexName), not(equalTo(historyUUID))); | ||
assertThat(historyUUID(node1, indexName), equalTo(historyUUID(node2, indexName))); | ||
|
||
internalCluster().assertSameDocIdsOnShards(); | ||
} | ||
|
||
public void checkHealthStatus(String indexName, ClusterHealthStatus healthStatus) { | ||
final ClusterHealthStatus indexHealthStatus = client().admin().cluster() | ||
.health(Requests.clusterHealthRequest(indexName)).actionGet().getStatus(); | ||
assertThat(indexHealthStatus, is(healthStatus)); | ||
} | ||
|
||
private int indexDocs(String indexName, Object ... source) throws InterruptedException, ExecutionException { | ||
// index some docs in several segments | ||
int numDocs = 0; | ||
for (int k = 0, attempts = randomIntBetween(5, 10); k < attempts; k++) { | ||
final int numExtraDocs = between(10, 100); | ||
IndexRequestBuilder[] builders = new IndexRequestBuilder[numExtraDocs]; | ||
for (int i = 0; i < builders.length; i++) { | ||
builders[i] = client().prepareIndex(indexName, "type").setSource(source); | ||
} | ||
|
||
indexRandom(true, false, true, Arrays.asList(builders)); | ||
numDocs += numExtraDocs; | ||
} | ||
|
||
return numDocs; | ||
} | ||
|
||
private Path getIndexPath(String nodeName, ShardId shardId) { | ||
final Set<Path> indexDirs = RemoveCorruptedShardDataCommandIT.getDirs(nodeName, shardId, ShardPath.INDEX_FOLDER_NAME); | ||
assertThat(indexDirs, hasSize(1)); | ||
return indexDirs.iterator().next(); | ||
} | ||
|
||
private Set<String> getAllocationIds(String indexName) { | ||
final ClusterState state = client().admin().cluster().prepareState().get().getState(); | ||
final Set<String> allocationIds = state.metaData().index(indexName).inSyncAllocationIds(0); | ||
return allocationIds; | ||
} | ||
|
||
private IndexSettings getIndexSettings(String indexName, String nodeName) { | ||
final IndicesService indicesService = internalCluster().getInstance(IndicesService.class, nodeName); | ||
final IndexService indexService = indicesService.indexService(resolveIndex(indexName)); | ||
return indexService.getIndexSettings(); | ||
} | ||
|
||
private String historyUUID(String node, String indexName) { | ||
final ShardStats[] shards = client(node).admin().indices().prepareStats(indexName).clear().get().getShards(); | ||
assertThat(shards.length, greaterThan(0)); | ||
final Set<String> historyUUIDs = Arrays.stream(shards) | ||
.map(shard -> shard.getCommitStats().getUserData().get(Engine.HISTORY_UUID_KEY)) | ||
.collect(Collectors.toSet()); | ||
assertThat(historyUUIDs, hasSize(1)); | ||
return historyUUIDs.iterator().next(); | ||
} | ||
|
||
private void putFakeCorruptionMarker(IndexSettings indexSettings, ShardId shardId, Path indexPath) throws IOException { | ||
try(Store store = new Store(shardId, indexSettings, new SimpleFSDirectory(indexPath), new DummyShardLock(shardId))) { | ||
store.markStoreCorrupted(new IOException("fake ioexception")); | ||
} | ||
} | ||
|
||
private void checkNoValidShardCopy(String indexName, ShardId shardId) throws Exception { | ||
final ClusterAllocationExplanation explanation = | ||
client().admin().cluster().prepareAllocationExplain() | ||
.setIndex(indexName).setShard(shardId.id()).setPrimary(true) | ||
.get().getExplanation(); | ||
|
||
final ShardAllocationDecision shardAllocationDecision = explanation.getShardAllocationDecision(); | ||
assertThat(shardAllocationDecision.isDecisionTaken(), equalTo(true)); | ||
assertThat(shardAllocationDecision.getAllocateDecision().getAllocationDecision(), | ||
equalTo(AllocationDecision.NO_VALID_SHARD_COPY)); | ||
} | ||
|
||
} |
Oops, something went wrong.