139 lines
3.8 KiB
JavaScript
139 lines
3.8 KiB
JavaScript
/**
|
|
* This file tests the top-chunk optimization logic in splitChunk command.
|
|
* Whenever a chunk is split, the shouldMigrate field should be set if the extreme chunk
|
|
* has only a single document, where extreme chunk is defined as the chunk containing
|
|
* either the upper or lower bound of the entire shard key space.
|
|
*/
|
|
|
|
var st = new ShardingTest({ shards: 1 });
|
|
st.stopBalancer();
|
|
|
|
var testDB = st.s.getDB('test');
|
|
testDB.adminCommand({ enableSharding: 'test' });
|
|
|
|
var callSplit = function(db, minKey, maxKey, splitPoints) {
|
|
var res = st.s.adminCommand({getShardVersion: "test.user"});
|
|
assert.commandWorked(res);
|
|
var shardVersion = [res.version, res.versionEpoch];
|
|
return db.runCommand({
|
|
splitChunk: 'test.user',
|
|
from: 'shard0000',
|
|
min: minKey,
|
|
max: maxKey,
|
|
keyPattern: { x: 1 },
|
|
splitKeys: splitPoints,
|
|
shardVersion: shardVersion,
|
|
});
|
|
};
|
|
|
|
var tests = [
|
|
//
|
|
// Lower extreme chunk tests.
|
|
//
|
|
|
|
// All chunks have 1 doc.
|
|
//
|
|
// Expected doc counts for new chunks:
|
|
// [ MinKey, -2 ): 1
|
|
// [ -2, -1 ): 1
|
|
// [ -1, 0): 1
|
|
//
|
|
function(db) {
|
|
var res = callSplit(db, { x: MinKey }, { x: 0 }, [{ x: -2 }, { x: -1 }]);
|
|
assert.commandWorked(res);
|
|
assert.neq(res.shouldMigrate, null, tojson(res));
|
|
assert(bsonWoCompare(res.shouldMigrate.min, { x: MinKey }) == 0, tojson(res.shouldMigrate.min));
|
|
assert(bsonWoCompare(res.shouldMigrate.max, { x: -2 }) == 0, tojson(res.shouldMigrate.max));
|
|
},
|
|
|
|
// One chunk has single doc, extreme doesn't.
|
|
//
|
|
// Expected doc counts for new chunks:
|
|
// [ MinKey, -1 ): 2
|
|
// [ -1, 0): 1
|
|
//
|
|
function(db) {
|
|
var res = callSplit(db, { x: MinKey }, { x: 0 }, [{ x: -1 }]);
|
|
assert.commandWorked(res);
|
|
assert.eq(res.shouldMigrate, null, tojson(res));
|
|
},
|
|
|
|
// Only extreme has single doc.
|
|
//
|
|
// Expected doc counts for new chunks:
|
|
// [ MinKey, -2 ): 1
|
|
// [ -2, 0): 2
|
|
//
|
|
function(db) {
|
|
var res = callSplit(db, { x: MinKey }, { x: 0 }, [{ x: -2 }]);
|
|
assert.commandWorked(res);
|
|
assert.neq(res.shouldMigrate, null, tojson(res));
|
|
assert(bsonWoCompare(res.shouldMigrate.min, { x: MinKey }) == 0, tojson(res.shouldMigrate.min));
|
|
assert(bsonWoCompare(res.shouldMigrate.max, { x: -2 }) == 0, tojson(res.shouldMigrate.max));
|
|
},
|
|
|
|
//
|
|
// Upper extreme chunk tests.
|
|
//
|
|
|
|
// All chunks have 1 doc.
|
|
//
|
|
// Expected doc counts for new chunks:
|
|
// [ 0, 1 ): 1
|
|
// [ 1, 2 ): 1
|
|
// [ 2, MaxKey): 1
|
|
//
|
|
function(db) {
|
|
var res = callSplit(db, { x: 0 }, { x: MaxKey }, [{ x: 1 }, { x: 2 }]);
|
|
assert.commandWorked(res);
|
|
assert.neq(res.shouldMigrate, null, tojson(res));
|
|
assert(bsonWoCompare(res.shouldMigrate.min, { x: 2 }) == 0, tojson(res.shouldMigrate.min));
|
|
assert(bsonWoCompare(res.shouldMigrate.max, { x: MaxKey }) == 0, tojson(res.shouldMigrate.max));
|
|
},
|
|
|
|
// One chunk has single doc, extreme doesn't.
|
|
//
|
|
// Expected doc counts for new chunks:
|
|
// [ 0, 1 ): 1
|
|
// [ 1, MaxKey): 2
|
|
//
|
|
function(db) {
|
|
var res = callSplit(db, { x: 0 }, { x: MaxKey }, [{ x: 1 }]);
|
|
assert.commandWorked(res);
|
|
assert.eq(res.shouldMigrate, null, tojson(res));
|
|
},
|
|
|
|
// Only extreme has single doc.
|
|
//
|
|
// Expected doc counts for new chunks:
|
|
// [ 0, 2 ): 2
|
|
// [ 2, MaxKey): 1
|
|
//
|
|
function(db) {
|
|
var res = callSplit(db, { x: 0 }, { x: MaxKey }, [{ x: 2 }]);
|
|
assert.commandWorked(res);
|
|
assert.neq(res.shouldMigrate, null, tojson(res));
|
|
assert(bsonWoCompare(res.shouldMigrate.min, { x: 2 }) == 0, tojson(res.shouldMigrate.min));
|
|
assert(bsonWoCompare(res.shouldMigrate.max, { x: MaxKey }) == 0, tojson(res.shouldMigrate.max));
|
|
},
|
|
];
|
|
|
|
tests.forEach(function(test) {
|
|
// setup
|
|
testDB.adminCommand({ shardCollection: 'test.user', key: { x: 1 }});
|
|
testDB.adminCommand({ split: 'test.user', middle: { x: 0 }});
|
|
|
|
for (var x = -3; x < 3; x++) {
|
|
testDB.user.insert({ x: x });
|
|
}
|
|
|
|
// run test
|
|
test(st.d0.getDB('admin'));
|
|
|
|
// teardown
|
|
testDB.user.drop();
|
|
});
|
|
|
|
st.stop();
|
|
|