mirror of https://github.com/mongodb/mongo
96 lines
3.7 KiB
JavaScript
96 lines
3.7 KiB
JavaScript
/**
|
|
* Tests for basic functionality of the move collection feature.
|
|
*
|
|
* @tags: [
|
|
* requires_fcv_80,
|
|
* featureFlagMoveCollection,
|
|
* assumes_balancer_off,
|
|
* ]
|
|
*/
|
|
|
|
import {ShardingTest} from "jstests/libs/shardingtest.js";
|
|
|
|
(function () {
|
|
"use strict";
|
|
|
|
let st = new ShardingTest({mongos: 1, shards: 2});
|
|
|
|
const dbName = "db";
|
|
const collName = "foo";
|
|
const ns = dbName + "." + collName;
|
|
let mongos = st.s0;
|
|
let shard0 = st.shard0.shardName;
|
|
let shard1 = st.shard1.shardName;
|
|
|
|
let cmdObj = {moveCollection: ns, toShard: shard0};
|
|
|
|
// Fail if collection is not tracked.
|
|
assert.commandFailedWithCode(mongos.adminCommand(cmdObj), ErrorCodes.NamespaceNotFound);
|
|
|
|
assert.commandWorked(st.s.adminCommand({enableSharding: dbName, primaryShard: shard0}));
|
|
assert.commandWorked(mongos.adminCommand({shardCollection: ns, key: {oldKey: 1}}));
|
|
|
|
// Fail if collection is sharded.
|
|
assert.commandFailedWithCode(mongos.adminCommand(cmdObj), ErrorCodes.NamespaceNotFound);
|
|
|
|
const unsplittableCollName = "foo_unsplittable";
|
|
const unsplittableCollNs = dbName + "." + unsplittableCollName;
|
|
assert.commandWorked(st.s.getDB(dbName).runCommand({create: unsplittableCollName}));
|
|
|
|
// Fail if missing required field toShard.
|
|
assert.commandFailedWithCode(
|
|
mongos.adminCommand({moveCollection: unsplittableCollNs}),
|
|
ErrorCodes.IDLFailedToParse,
|
|
);
|
|
|
|
// Fail if command called on shard.
|
|
assert.commandFailedWithCode(
|
|
st.shard0.adminCommand({moveCollection: unsplittableCollNs, toShard: shard1}),
|
|
ErrorCodes.CommandNotFound,
|
|
);
|
|
|
|
const coll = mongos.getDB(dbName)[unsplittableCollName];
|
|
for (let i = -25; i < 25; ++i) {
|
|
assert.commandWorked(coll.insert({oldKey: i}));
|
|
}
|
|
assert.eq(50, st.rs0.getPrimary().getCollection(unsplittableCollNs).countDocuments({}));
|
|
|
|
// move to non-primary shard.
|
|
assert.commandWorked(mongos.adminCommand({moveCollection: unsplittableCollNs, toShard: shard1}));
|
|
|
|
// Should have unsplittable set to true
|
|
let configDb = mongos.getDB("config");
|
|
let unshardedColl = configDb.collections.findOne({_id: unsplittableCollNs});
|
|
assert.eq(unshardedColl.unsplittable, true);
|
|
let unshardedChunk = configDb.chunks.find({uuid: unshardedColl.uuid}).toArray();
|
|
assert.eq(1, unshardedChunk.length);
|
|
|
|
assert.eq(50, st.rs1.getPrimary().getCollection(unsplittableCollNs).countDocuments({}));
|
|
assert.eq(0, st.rs0.getPrimary().getCollection(unsplittableCollNs).countDocuments({}));
|
|
|
|
const metrics = st.config0.getDB("admin").serverStatus({}).shardingStatistics.moveCollection;
|
|
|
|
assert.eq(metrics.countStarted, 1);
|
|
assert.eq(metrics.countSameKeyStarted, undefined);
|
|
assert.eq(metrics.countSucceeded, 1);
|
|
assert.eq(metrics.countFailed, 0);
|
|
assert.eq(metrics.countCanceled, 0);
|
|
|
|
// move to primary shard.
|
|
assert.commandWorked(mongos.adminCommand({moveCollection: unsplittableCollNs, toShard: shard0}));
|
|
unshardedColl = configDb.collections.findOne({_id: unsplittableCollNs});
|
|
assert.eq(unshardedColl.unsplittable, true);
|
|
unshardedChunk = configDb.chunks.find({uuid: unshardedColl.uuid}).toArray();
|
|
assert.eq(1, unshardedChunk.length);
|
|
|
|
assert.eq(0, st.rs1.getPrimary().getCollection(unsplittableCollNs).countDocuments({}));
|
|
assert.eq(50, st.rs0.getPrimary().getCollection(unsplittableCollNs).countDocuments({}));
|
|
|
|
// Successfully move a collection with a long namespace below 255.
|
|
const longCollName = "a".repeat(250);
|
|
const longNs = dbName + "." + longCollName;
|
|
assert.commandWorked(mongos.getDB(dbName).createCollection(longCollName));
|
|
assert.commandWorked(mongos.adminCommand({moveCollection: longNs, toShard: shard1}));
|
|
st.stop();
|
|
})();
|