forked from elastic/elasticsearch
-
Notifications
You must be signed in to change notification settings - Fork 2
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Fix
NOT_STARTED statuses appearing inappropirately during node shut…
…down (elastic#75750) This PR fixes two situations where `NOT_STARTED` can appear as the shard migration status inappropriately: 1. When the node is actually shut down after having all the shards migrate away. 2. When a non-data-node is registered for shutdown. It also adds tests to ensure these cases are handled correctly.
- Loading branch information
Showing
8 changed files
with
357 additions
and
28 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
147 changes: 147 additions & 0 deletions
147
...n/src/internalClusterTest/java/org/elasticsearch/xpack/shutdown/NodeShutdownShardsIT.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,147 @@ | ||
| /* | ||
| * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one | ||
| * or more contributor license agreements. Licensed under the Elastic License | ||
| * 2.0; you may not use this file except in compliance with the Elastic License | ||
| * 2.0. | ||
| */ | ||
|
|
||
| package org.elasticsearch.xpack.shutdown; | ||
|
|
||
| import org.elasticsearch.Build; | ||
| import org.elasticsearch.action.admin.cluster.node.info.NodeInfo; | ||
| import org.elasticsearch.action.admin.cluster.node.info.NodesInfoResponse; | ||
| import org.elasticsearch.action.support.master.AcknowledgedResponse; | ||
| import org.elasticsearch.cluster.metadata.SingleNodeShutdownMetadata; | ||
| import org.elasticsearch.cluster.node.DiscoveryNode; | ||
| import org.elasticsearch.common.settings.Settings; | ||
| import org.elasticsearch.plugins.Plugin; | ||
| import org.elasticsearch.test.ESIntegTestCase; | ||
| import org.elasticsearch.test.InternalTestCluster; | ||
|
|
||
| import java.util.Arrays; | ||
| import java.util.Collection; | ||
|
|
||
| import static org.elasticsearch.cluster.metadata.SingleNodeShutdownMetadata.Status.COMPLETE; | ||
| import static org.hamcrest.Matchers.equalTo; | ||
|
|
||
| @ESIntegTestCase.ClusterScope(scope = ESIntegTestCase.Scope.TEST, numDataNodes = 0, numClientNodes = 0) | ||
| public class NodeShutdownShardsIT extends ESIntegTestCase { | ||
|
|
||
| @Override | ||
| protected Collection<Class<? extends Plugin>> nodePlugins() { | ||
| return Arrays.asList(ShutdownPlugin.class); | ||
| } | ||
|
|
||
| /** | ||
| * Verifies that a node that's removed from the cluster with zero shards stays in the `COMPLETE` status after it leaves, rather than | ||
| * reverting to `NOT_STARTED` (this was a bug in the initial implementation). | ||
| */ | ||
| public void testShardStatusStaysCompleteAfterNodeLeaves() throws Exception { | ||
| assumeTrue("must be on a snapshot build of ES to run in order for the feature flag to be set", Build.CURRENT.isSnapshot()); | ||
| final String nodeToRestartName = internalCluster().startNode(); | ||
| final String nodeToRestartId = getNodeId(nodeToRestartName); | ||
| internalCluster().startNode(); | ||
|
|
||
| // Mark the node for shutdown | ||
| PutShutdownNodeAction.Request putShutdownRequest = new PutShutdownNodeAction.Request( | ||
| nodeToRestartId, | ||
| SingleNodeShutdownMetadata.Type.REMOVE, | ||
| this.getTestName(), | ||
| null | ||
| ); | ||
| AcknowledgedResponse putShutdownResponse = client().execute(PutShutdownNodeAction.INSTANCE, putShutdownRequest).get(); | ||
| assertTrue(putShutdownResponse.isAcknowledged()); | ||
|
|
||
| internalCluster().stopNode(nodeToRestartName); | ||
|
|
||
| NodesInfoResponse nodes = client().admin().cluster().prepareNodesInfo().clear().get(); | ||
| assertThat(nodes.getNodes().size(), equalTo(1)); | ||
|
|
||
| GetShutdownStatusAction.Response getResp = client().execute( | ||
| GetShutdownStatusAction.INSTANCE, | ||
| new GetShutdownStatusAction.Request(nodeToRestartId) | ||
| ).get(); | ||
|
|
||
| assertThat(getResp.getShutdownStatuses().get(0).migrationStatus().getStatus(), equalTo(COMPLETE)); | ||
| } | ||
|
|
||
| /** | ||
| * Similar to the previous test, but ensures that the status stays at `COMPLETE` when the node is offline when the shutdown is | ||
| * registered. This may happen if {@link NodeSeenService} isn't working as expected. | ||
| */ | ||
| public void testShardStatusStaysCompleteAfterNodeLeavesIfRegisteredWhileNodeOffline() throws Exception { | ||
| assumeTrue("must be on a snapshot build of ES to run in order for the feature flag to be set", Build.CURRENT.isSnapshot()); | ||
| final String nodeToRestartName = internalCluster().startNode(); | ||
| final String nodeToRestartId = getNodeId(nodeToRestartName); | ||
| internalCluster().startNode(); | ||
|
|
||
| // Stop the node we're going to shut down and mark it as shutting down while it's offline. This checks that the cluster state | ||
| // listener is working correctly. | ||
| internalCluster().restartNode(nodeToRestartName, new InternalTestCluster.RestartCallback() { | ||
| @Override | ||
| public Settings onNodeStopped(String nodeName) throws Exception { | ||
| PutShutdownNodeAction.Request putShutdownRequest = new PutShutdownNodeAction.Request( | ||
| nodeToRestartId, | ||
| SingleNodeShutdownMetadata.Type.REMOVE, | ||
| "testShardStatusStaysCompleteAfterNodeLeavesIfRegisteredWhileNodeOffline", | ||
| null | ||
| ); | ||
| AcknowledgedResponse putShutdownResponse = client().execute(PutShutdownNodeAction.INSTANCE, putShutdownRequest).get(); | ||
| assertTrue(putShutdownResponse.isAcknowledged()); | ||
|
|
||
| return super.onNodeStopped(nodeName); | ||
| } | ||
| }); | ||
|
|
||
| internalCluster().stopNode(nodeToRestartName); | ||
|
|
||
| NodesInfoResponse nodes = client().admin().cluster().prepareNodesInfo().clear().get(); | ||
| assertThat(nodes.getNodes().size(), equalTo(1)); | ||
|
|
||
| GetShutdownStatusAction.Response getResp = client().execute( | ||
| GetShutdownStatusAction.INSTANCE, | ||
| new GetShutdownStatusAction.Request(nodeToRestartId) | ||
| ).get(); | ||
|
|
||
| assertThat(getResp.getShutdownStatuses().get(0).migrationStatus().getStatus(), equalTo(COMPLETE)); | ||
| } | ||
|
|
||
| /** | ||
| * Checks that non-data nodes that are registered for shutdown have a shard migration status of `COMPLETE` rather than `NOT_STARTED`. | ||
| * (this was a bug in the initial implementation). | ||
| */ | ||
| public void testShardStatusIsCompleteOnNonDataNodes() throws Exception { | ||
| assumeTrue("must be on a snapshot build of ES to run in order for the feature flag to be set", Build.CURRENT.isSnapshot()); | ||
| final String nodeToShutDownName = internalCluster().startMasterOnlyNode(); | ||
| internalCluster().startMasterOnlyNode(); // Just to have at least one other node | ||
| final String nodeToRestartId = getNodeId(nodeToShutDownName); | ||
|
|
||
| // Mark the node for shutdown | ||
| PutShutdownNodeAction.Request putShutdownRequest = new PutShutdownNodeAction.Request( | ||
| nodeToRestartId, | ||
| SingleNodeShutdownMetadata.Type.REMOVE, | ||
| this.getTestName(), | ||
| null | ||
| ); | ||
| AcknowledgedResponse putShutdownResponse = client().execute(PutShutdownNodeAction.INSTANCE, putShutdownRequest).get(); | ||
| assertTrue(putShutdownResponse.isAcknowledged()); | ||
|
|
||
| GetShutdownStatusAction.Response getResp = client().execute( | ||
| GetShutdownStatusAction.INSTANCE, | ||
| new GetShutdownStatusAction.Request(nodeToRestartId) | ||
| ).get(); | ||
|
|
||
| assertThat(getResp.getShutdownStatuses().get(0).migrationStatus().getStatus(), equalTo(COMPLETE)); | ||
| } | ||
|
|
||
| private String getNodeId(String nodeName) throws Exception { | ||
| NodesInfoResponse nodes = client().admin().cluster().prepareNodesInfo().clear().get(); | ||
| return nodes.getNodes() | ||
| .stream() | ||
| .map(NodeInfo::getNode) | ||
| .filter(node -> node.getName().equals(nodeName)) | ||
| .map(DiscoveryNode::getId) | ||
| .findFirst() | ||
| .orElseThrow(); | ||
| } | ||
| } |
Oops, something went wrong.