mongo/jstests/sharding/sharding_migrate_large_docs.js

62 lines
2.0 KiB
JavaScript

//
// Tests migration behavior of large documents
// @tags: [requires_sharding]
//
import {ShardingTest} from "jstests/libs/shardingtest.js";
let st = new ShardingTest({shards: 2, mongos: 1});
let mongos = st.s0;
let coll = mongos.getCollection("foo.bar");
let admin = mongos.getDB("admin");
let shards = mongos.getCollection("config.shards").find().toArray();
let shardAdmin = st.shard0.getDB("admin");
assert(admin.runCommand({enableSharding: coll.getDB() + "", primaryShard: shards[0]._id}).ok);
assert(admin.runCommand({shardCollection: coll + "", key: {_id: 1}}).ok);
assert(admin.runCommand({split: coll + "", middle: {_id: 0}}).ok);
jsTest.log("Preparing large insert...");
let data1MB = "x";
while (data1MB.length < 1024 * 1024) data1MB += data1MB;
let data15MB = "";
for (var i = 0; i < 15; i++) data15MB += data1MB;
let data15PlusMB = data15MB;
for (var i = 0; i < 1023 * 1024; i++) data15PlusMB += "x";
print("~15MB object size is : " + Object.bsonsize({_id: 0, d: data15PlusMB}));
jsTest.log("Inserting docs of large and small sizes...");
// Two large docs next to each other
coll.insert({_id: -2, d: data15PlusMB});
coll.insert({_id: -1, d: data15PlusMB});
// Docs of assorted sizes
assert.commandWorked(coll.insert({_id: 0, d: "x"}));
assert.commandWorked(coll.insert({_id: 1, d: data15PlusMB}));
assert.commandWorked(coll.insert({_id: 2, d: "x"}));
assert.commandWorked(coll.insert({_id: 3, d: data15MB}));
assert.commandWorked(coll.insert({_id: 4, d: "x"}));
assert.commandWorked(coll.insert({_id: 5, d: data1MB}));
assert.commandWorked(coll.insert({_id: 6, d: "x"}));
assert.eq(9, coll.find().itcount());
jsTest.log("Starting migration...");
assert(admin.runCommand({moveChunk: coll + "", find: {_id: 0}, to: shards[1]._id}).ok);
assert(admin.runCommand({moveChunk: coll + "", find: {_id: -1}, to: shards[1]._id}).ok);
// Ensure that the doc count is correct and that the mongos query path can handle docs near the 16MB
// user BSON size limit.
assert.eq(9, coll.find().itcount());
jsTest.log("DONE!");
st.stop();