mirror of https://github.com/mongodb/mongo
83 lines
2.9 KiB
JavaScript
83 lines
2.9 KiB
JavaScript
/**
|
|
* Verifies that resharding honors the critical section timeout.
|
|
*
|
|
* @tags: [
|
|
* uses_atclustertime
|
|
* ]
|
|
*/
|
|
|
|
import {DiscoverTopology} from "jstests/libs/discover_topology.js";
|
|
import {ReshardingTest} from "jstests/sharding/libs/resharding_test_fixture.js";
|
|
|
|
function setupTest(reshardingTest, namespace, timeout) {
|
|
reshardingTest.setup();
|
|
|
|
jsTest.log(`Running test for criticalSectionTimeoutMillis = ${timeout}`);
|
|
|
|
const donorShardNames = reshardingTest.donorShardNames;
|
|
const inputCollection = reshardingTest.createShardedCollection({
|
|
ns: namespace,
|
|
shardKeyPattern: {oldKey: 1},
|
|
chunks: [
|
|
{min: {oldKey: MinKey}, max: {oldKey: 0}, shard: donorShardNames[0]},
|
|
{min: {oldKey: 0}, max: {oldKey: MaxKey}, shard: donorShardNames[1]},
|
|
],
|
|
});
|
|
|
|
const mongos = inputCollection.getMongo();
|
|
const topology = DiscoverTopology.findConnectedNodes(mongos);
|
|
const coordinator = new Mongo(topology.configsvr.nodes[0]);
|
|
assert.commandWorked(
|
|
coordinator.getDB("admin").adminCommand({setParameter: 1, reshardingCriticalSectionTimeoutMillis: timeout}),
|
|
);
|
|
|
|
assert.commandWorked(
|
|
inputCollection.insert([
|
|
{_id: "stays on shard0", oldKey: -10, newKey: -10},
|
|
{_id: "moves to shard0", oldKey: 10, newKey: -10},
|
|
{_id: "moves to shard1", oldKey: -10, newKey: 10},
|
|
{_id: "stays on shard1", oldKey: 10, newKey: 10},
|
|
]),
|
|
);
|
|
}
|
|
|
|
// This test will not timeout.
|
|
const successReshardingTest = new ReshardingTest({numDonors: 2, numRecipients: 2, reshardInPlace: true});
|
|
const noTimeoutMillis = 8000;
|
|
let namespace = `reshardingDb.coll${noTimeoutMillis}`;
|
|
|
|
setupTest(successReshardingTest, namespace, noTimeoutMillis);
|
|
|
|
let recipientShardNames = successReshardingTest.recipientShardNames;
|
|
successReshardingTest.withReshardingInBackground({
|
|
newShardKeyPattern: {newKey: 1},
|
|
newChunks: [
|
|
{min: {newKey: MinKey}, max: {newKey: 0}, shard: recipientShardNames[0]},
|
|
{min: {newKey: 0}, max: {newKey: MaxKey}, shard: recipientShardNames[1]},
|
|
],
|
|
});
|
|
|
|
successReshardingTest.teardown();
|
|
|
|
// This test will timeout.
|
|
const failureReshardingTest = new ReshardingTest({numDonors: 2, numRecipients: 2, reshardInPlace: true});
|
|
const shouldTimeoutMillis = 0;
|
|
namespace = `reshardingDb.coll${shouldTimeoutMillis}`;
|
|
|
|
setupTest(failureReshardingTest, namespace, shouldTimeoutMillis);
|
|
|
|
recipientShardNames = failureReshardingTest.recipientShardNames;
|
|
failureReshardingTest.withReshardingInBackground(
|
|
{
|
|
newShardKeyPattern: {newKey: 1},
|
|
newChunks: [
|
|
{min: {newKey: MinKey}, max: {newKey: 0}, shard: recipientShardNames[0]},
|
|
{min: {newKey: 0}, max: {newKey: MaxKey}, shard: recipientShardNames[1]},
|
|
],
|
|
},
|
|
() => {},
|
|
{expectedErrorCode: ErrorCodes.ReshardingCriticalSectionTimeout},
|
|
);
|
|
|
|
failureReshardingTest.teardown();
|