Files
mongo/jstests/sharding/cursor_cleanup.js
2015-09-29 09:37:18 -04:00

64 lines
1.8 KiB
JavaScript

//
// Tests cleanup of sharded and unsharded cursors
//
var st = new ShardingTest({ shards : 2, mongos : 1 });
st.stopBalancer();
var mongos = st.s0;
var admin = mongos.getDB( "admin" );
var config = mongos.getDB( "config" );
var shards = config.shards.find().toArray();
var coll = mongos.getCollection( "foo.bar" );
var collUnsharded = mongos.getCollection( "foo.baz" );
// Shard collection
printjson(admin.runCommand({ enableSharding : coll.getDB() + "" }));
printjson(admin.runCommand({ movePrimary : coll.getDB() + "", to : shards[0]._id }));
printjson(admin.runCommand({ shardCollection : coll + "", key : { _id : 1 } }));
printjson(admin.runCommand({ split : coll + "", middle : { _id : 0 } }));
printjson(admin.runCommand({ moveChunk : coll + "", find : { _id : 0 }, to : shards[1]._id }));
jsTest.log("Collection set up...");
st.printShardingStatus(true);
jsTest.log("Insert enough data to overwhelm a query batch.");
var bulk = coll.initializeUnorderedBulkOp();
var bulk2 = collUnsharded.initializeUnorderedBulkOp();
for (var i = -150; i < 150; i++) {
bulk.insert({ _id : i });
bulk2.insert({ _id : i });
}
assert.writeOK(bulk.execute());
assert.writeOK(bulk2.execute());
jsTest.log("Open a cursor to a sharded and unsharded collection.");
var shardedCursor = coll.find();
assert.neq(null, shardedCursor.next());
var unshardedCursor = collUnsharded.find();
assert.neq(null, unshardedCursor.next());
jsTest.log("Check whether the cursor is registered in the cursor info.");
var cursorInfo = admin.serverStatus().metrics.cursor;
printjson(cursorInfo);
assert.eq(cursorInfo.open.multiTarget, 1);
jsTest.log("End the cursors.");
shardedCursor.itcount();
unshardedCursor.itcount();
var cursorInfo = admin.serverStatus().metrics.cursor;;
printjson(cursorInfo);
assert.eq(cursorInfo.open.multiTarget, 0);
jsTest.log("DONE!");
st.stop();