summaryrefslogtreecommitdiff
path: root/jstests/sharding/cursor1.js
blob: 8b799ce1aa67e255f170d2435a72e5579fb55006 (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
// 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() , "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() );

// 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.
gc(); gc();
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);

gc(); gc();

s.stop();

})();