summaryrefslogtreecommitdiff
path: root/jstests/sharding/autosplit.js
blob: af67d7820cb5968e11772129bbb3820ac20d759d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
/**
 * This test confirms that chunks get split as they grow due to data insertion.
 */
(function() {
'use strict';
load('jstests/sharding/autosplit_include.js');
load("jstests/sharding/libs/find_chunks_util.js");

var s = new ShardingTest({
    name: "auto1",
    shards: 2,
    mongos: 1,
    other: {enableAutoSplit: true, chunkSize: 10},
});

assert.commandWorked(s.s0.adminCommand({enablesharding: "test"}));
s.ensurePrimaryShard('test', s.shard1.shardName);
assert.commandWorked(s.s0.adminCommand({shardcollection: "test.foo", key: {num: 1}}));

var bigString = "";
while (bigString.length < 1024 * 50)
    bigString += "asocsancdnsjfnsdnfsjdhfasdfasdfasdfnsadofnsadlkfnsaldknfsad";

var db = s.getDB("test");
var primary = s.getPrimaryShard("test").getDB("test");
var coll = db.foo;
var counts = [];

var i = 0;

// Inserts numDocs documents into the collection, waits for any ongoing
// splits to finish, and then prints some information about the
// collection's chunks
function insertDocsAndWaitForSplit(numDocs) {
    var bulk = coll.initializeUnorderedBulkOp();
    var curMaxKey = i;
    // Increment the global 'i' variable to keep 'num' unique across all
    // documents
    for (; i < curMaxKey + numDocs; i++) {
        bulk.insert({num: i, s: bigString});
    }
    assert.commandWorked(bulk.execute());

    waitForOngoingChunkSplits(s);

    s.printChunks();
    s.printChangeLog();
}

insertDocsAndWaitForSplit(100);

counts.push(findChunksUtil.countChunksForNs(s.config, "test.foo"));
assert.eq(100, db.foo.find().itcount());

print("datasize: " +
      tojson(s.getPrimaryShard("test").getDB("admin").runCommand({datasize: "test.foo"})));

insertDocsAndWaitForSplit(100);
counts.push(findChunksUtil.countChunksForNs(s.config, "test.foo"));

insertDocsAndWaitForSplit(200);
counts.push(findChunksUtil.countChunksForNs(s.config, "test.foo"));

insertDocsAndWaitForSplit(300);
counts.push(findChunksUtil.countChunksForNs(s.config, "test.foo"));

assert(counts[counts.length - 1] > counts[0], "counts 1 : " + tojson(counts));
var sorted = counts.slice(0);
// Sort doesn't sort numbers correctly by default, resulting in fail
sorted.sort(function(a, b) {
    return a - b;
});
assert.eq(counts, sorted, "counts 2 : " + tojson(counts));

print(counts);

printjson(db.stats());

s.stop();
})();