summaryrefslogtreecommitdiff
path: root/jstests/sharding/cursor1.js
blob: cf5bfb27c1ebdefd6e31e4a5f3ac2ec79e83f548 (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
// cursor1.js
// checks that cursors survive a chunk's move
(function() {

    var s = new ShardingTest({name: "sharding_cursor1", shards: 2});

    s.config.settings.find().forEach(printjson);

    // create a sharded 'test.foo', for the moment with just one chunk
    s.adminCommand({enablesharding: "test"});
    s.ensurePrimaryShard('test', 'shard0001');
    s.adminCommand({shardcollection: "test.foo", key: {_id: 1}});

    db = s.getDB("test");
    primary = s.getPrimaryShard("test").getDB("test");
    secondary = s.getOther(primary).getDB("test");

    var numObjs = 30;
    var bulk = db.foo.initializeUnorderedBulkOp();
    for (i = 0; i < numObjs; i++) {
        bulk.insert({_id: i});
    }
    assert.writeOK(bulk.execute());
    assert.eq(1,
              s.config.chunks.count({"ns": "test.foo"}),
              "test requires collection to have one chunk initially");

    // we'll split the collection in two and move the second chunk while three cursors are open
    // cursor1 still has more data in the first chunk, the one that didn't move
    // cursor2 buffered the last obj of the first chunk
    // cursor3 buffered data that was moved on the second chunk
    var cursor1 = db.foo.find().batchSize(3);
    assert.eq(3, cursor1.objsLeftInBatch());
    var cursor2 = db.foo.find().batchSize(5);
    assert.eq(5, cursor2.objsLeftInBatch());
    var cursor3 = db.foo.find().batchSize(7);
    assert.eq(7, cursor3.objsLeftInBatch());

    s.adminCommand({split: "test.foo", middle: {_id: 5}});
    s.adminCommand({movechunk: "test.foo", find: {_id: 5}, to: secondary.getMongo().name});
    assert.eq(2, s.config.chunks.count({"ns": "test.foo"}));

    // the cursors should not have been affected
    assert.eq(numObjs, cursor1.itcount(), "c1");
    assert.eq(numObjs, cursor2.itcount(), "c2");
    assert.eq(numObjs, cursor3.itcount(), "c3");

    // Test that a cursor with a 1 second timeout eventually times out.
    var cur = db.foo.find().batchSize(2);
    assert(cur.next(), "T1");
    assert(cur.next(), "T2");
    assert.commandWorked(s.admin.runCommand({
        setParameter: 1,
        cursorTimeoutMillis: 1000  // 1 second.
    }));

    assert.soon(function() {
        try {
            cur.next();
            cur.next();
            print("cursor still alive");
            return false;
        } catch (e) {
            return true;
        }
    }, "cursor failed to time out", /*timeout*/ 30000, /*interval*/ 5000);

    s.stop();

})();