136 lines
6.1 KiB
JavaScript
136 lines
6.1 KiB
JavaScript
/**
|
|
* Perform basic tests for the split command against mongos.
|
|
*/
|
|
import {ShardingTest} from "jstests/libs/shardingtest.js";
|
|
import {findChunksUtil} from "jstests/sharding/libs/find_chunks_util.js";
|
|
|
|
var st = new ShardingTest({mongos: 2, shards: 2, other: {chunkSize: 1}});
|
|
var configDB = st.s0.getDB('config');
|
|
|
|
var shard0 = st.shard0.shardName;
|
|
var shard1 = st.shard1.shardName;
|
|
|
|
// split on invalid ns.
|
|
assert.commandFailed(configDB.adminCommand({split: 'user', key: {_id: 1}}));
|
|
|
|
// split on unsharded collection (db is not sharding enabled).
|
|
assert.commandFailed(configDB.adminCommand({split: 'test.user', key: {_id: 1}}));
|
|
|
|
assert.commandWorked(configDB.adminCommand({enableSharding: 'test', primaryShard: shard0}));
|
|
|
|
// split on unsharded collection (db is sharding enabled).
|
|
assert.commandFailed(configDB.adminCommand({split: 'test.user', key: {_id: 1}}));
|
|
|
|
assert.commandWorked(configDB.adminCommand({shardCollection: 'test.user', key: {_id: 1}}));
|
|
|
|
assert.eq(null, findChunksUtil.findOneChunkByNs(configDB, 'test.user', {min: {_id: 0}}));
|
|
|
|
assert.commandWorked(configDB.adminCommand({split: 'test.user', middle: {_id: 0}}));
|
|
assert.neq(null, findChunksUtil.findOneChunkByNs(configDB, 'test.user', {min: {_id: 0}}));
|
|
|
|
// TODO(SERVER-97588): Remove version check from tests when 8.1 becomes last LTS.
|
|
const fcvDoc = configDB.adminCommand({getParameter: 1, featureCompatibilityVersion: 1});
|
|
if (MongoRunner.compareBinVersions(fcvDoc.featureCompatibilityVersion.version, "8.1") >= 0) {
|
|
// It should not fail on boundaries that have already been split.
|
|
assert.commandWorked(configDB.adminCommand({split: 'test.user', middle: {_id: 0}}));
|
|
} else {
|
|
// Cannot split on existing chunk boundary.
|
|
assert.commandFailed(configDB.adminCommand({split: 'test.user', middle: {_id: 0}}));
|
|
}
|
|
|
|
// Attempt to split on a value that is not the shard key.
|
|
assert.commandFailed(configDB.adminCommand({split: 'test.user', middle: {x: 100}}));
|
|
assert.commandFailed(configDB.adminCommand({split: 'test.user', find: {x: 100}}));
|
|
assert.commandFailed(
|
|
configDB.adminCommand({split: 'test.user', bounds: [{x: MinKey}, {x: MaxKey}]}));
|
|
|
|
// Insert documents large enough to fill up a chunk
|
|
var kiloDoc = new Array(1024).join('x');
|
|
var testDB = st.s.getDB('test');
|
|
var bulk = testDB.user.initializeUnorderedBulkOp();
|
|
for (var x = -1200; x < 1200; x++) {
|
|
bulk.insert({_id: x, val: kiloDoc});
|
|
}
|
|
assert.commandWorked(bulk.execute());
|
|
|
|
assert.eq(1,
|
|
findChunksUtil.findChunksByNs(configDB, 'test.user', {min: {$gte: {_id: 0}}}).itcount());
|
|
|
|
// Errors if bounds do not correspond to existing chunk boundaries.
|
|
assert.commandFailed(configDB.adminCommand({split: 'test.user', bounds: [{_id: 0}, {_id: 1000}]}));
|
|
assert.eq(1,
|
|
findChunksUtil.findChunksByNs(configDB, 'test.user', {min: {$gte: {_id: 0}}}).itcount());
|
|
|
|
assert.commandWorked(
|
|
configDB.adminCommand({split: 'test.user', bounds: [{_id: 0}, {_id: MaxKey}]}));
|
|
assert.gt(findChunksUtil.findChunksByNs(configDB, 'test.user', {min: {$gte: {_id: 0}}}).itcount(),
|
|
1);
|
|
|
|
assert.eq(1,
|
|
findChunksUtil.findChunksByNs(configDB, 'test.user', {min: {$lt: {_id: 0}}}).itcount());
|
|
assert.commandWorked(configDB.adminCommand({split: 'test.user', middle: {_id: -600}}));
|
|
assert.gt(findChunksUtil.findChunksByNs(configDB, 'test.user', {min: {$lt: {_id: 0}}}).itcount(),
|
|
1);
|
|
|
|
// Mongos must refresh metadata if the chunk version does not match
|
|
assert.commandWorked(st.s0.adminCommand(
|
|
{moveChunk: 'test.user', find: {_id: -900}, to: shard1, _waitForDelete: true}));
|
|
assert.commandWorked(st.s1.adminCommand({split: 'test.user', middle: {_id: -900}}));
|
|
assert.commandWorked(st.s1.adminCommand(
|
|
{moveChunk: 'test.user', find: {_id: -900}, to: shard0, _waitForDelete: true}));
|
|
assert.commandWorked(st.s1.adminCommand(
|
|
{moveChunk: 'test.user', find: {_id: -901}, to: shard0, _waitForDelete: true}));
|
|
assert.eq(0, findChunksUtil.findChunksByNs(configDB, 'test.user', {shard: shard1}).itcount());
|
|
|
|
//
|
|
// Compound Key
|
|
//
|
|
|
|
assert.commandWorked(configDB.adminCommand({shardCollection: 'test.compound', key: {x: 1, y: 1}}));
|
|
|
|
assert.eq(null, findChunksUtil.findOneChunkByNs(configDB, 'test.compound', {min: {x: 0, y: 0}}));
|
|
assert.commandWorked(configDB.adminCommand({split: 'test.compound', middle: {x: 0, y: 0}}));
|
|
assert.neq(null, findChunksUtil.findOneChunkByNs(configDB, 'test.compound', {min: {x: 0, y: 0}}));
|
|
|
|
// TODO(SERVER-97588): Remove version check from tests when 8.1 becomes last LTS.
|
|
if (MongoRunner.compareBinVersions(fcvDoc.featureCompatibilityVersion.version, "8.1") >= 0) {
|
|
// It should not fail on boundaries that have already been split.
|
|
assert.commandWorked(configDB.adminCommand({split: 'test.compound', middle: {x: 0, y: 0}}));
|
|
assert.commandWorked(
|
|
configDB.adminCommand({split: 'test.compound', middle: {x: MinKey, y: MinKey}}));
|
|
assert.commandWorked(
|
|
configDB.adminCommand({split: 'test.compound', middle: {x: MaxKey, y: MaxKey}}));
|
|
} else {
|
|
// cannot split on existing chunk boundary.
|
|
assert.commandFailed(configDB.adminCommand({split: 'test.compound', middle: {x: 0, y: 0}}));
|
|
assert.commandFailed(
|
|
configDB.adminCommand({split: 'test.compound', middle: {x: MinKey, y: MinKey}}));
|
|
assert.commandFailed(
|
|
configDB.adminCommand({split: 'test.compound', middle: {x: MaxKey, y: MaxKey}}));
|
|
}
|
|
|
|
bulk = testDB.compound.initializeUnorderedBulkOp();
|
|
for (x = -1200; x < 1200; x++) {
|
|
bulk.insert({x: x, y: x, val: kiloDoc});
|
|
}
|
|
assert.commandWorked(bulk.execute());
|
|
|
|
assert.eq(1,
|
|
findChunksUtil.findChunksByNs(configDB, 'test.compound', {min: {$gte: {x: 0, y: 0}}})
|
|
.itcount());
|
|
assert.commandWorked(configDB.adminCommand(
|
|
{split: 'test.compound', bounds: [{x: 0, y: 0}, {x: MaxKey, y: MaxKey}]}));
|
|
assert.gt(
|
|
findChunksUtil.findChunksByNs(configDB, 'test.compound', {min: {$gte: {x: 0, y: 0}}}).itcount(),
|
|
1);
|
|
|
|
assert.eq(
|
|
1,
|
|
findChunksUtil.findChunksByNs(configDB, 'test.compound', {min: {$lt: {x: 0, y: 0}}}).itcount());
|
|
assert.commandWorked(configDB.adminCommand({split: 'test.compound', find: {x: -1, y: -1}}));
|
|
assert.gt(
|
|
findChunksUtil.findChunksByNs(configDB, 'test.compound', {min: {$lt: {x: 0, y: 0}}}).itcount(),
|
|
1);
|
|
|
|
st.stop();
|