diff options
Diffstat (limited to 'jstests/sharding/sort1.js')
| -rw-r--r-- | jstests/sharding/sort1.js | 190 |
1 files changed, 97 insertions, 93 deletions
diff --git a/jstests/sharding/sort1.js b/jstests/sharding/sort1.js index 2ff8e1a4daf..57bae9dc390 100644 --- a/jstests/sharding/sort1.js +++ b/jstests/sharding/sort1.js @@ -1,106 +1,110 @@ (function() { -var s = new ShardingTest({ name: "sort1", - shards: 2, - mongos: 2 }); - -s.adminCommand( { enablesharding : "test" } ); -s.ensurePrimaryShard('test', 'shard0001'); -s.adminCommand( { shardcollection : "test.data" , key : { 'sub.num' : 1 } } ); - -db = s.getDB( "test" ); - -N = 100; - -forward = []; -backward = []; -for ( i=0; i<N; i++ ){ - db.data.insert( { _id : i , sub: {num : i , x : N - i }} ); - forward.push( i ); - backward.push( ( N - 1 ) - i ); -} - -s.adminCommand( { split : "test.data" , middle : { 'sub.num' : 33 } } ); -s.adminCommand( { split : "test.data" , middle : { 'sub.num' : 66 } } ); - -s.adminCommand({ movechunk : "test.data", - find : { 'sub.num' : 50 }, - to : s.getOther( s.getPrimaryShard( "test" ) ).name, - waitForDelete : true }); - -assert.lte( 3 , s.config.chunks.find().itcount() , "A1" ); - -temp = s.config.chunks.find().sort( { min : 1 } ).toArray(); -temp.forEach( printjsononeline ); - -z = 0; -for ( ; z<temp.length; z++ ) - if ( temp[z].min["sub.num"] <= 50 && temp[z].max["sub.num"] > 50 ) - break; - -assert.eq( temp[z-1].shard , temp[z+1].shard , "A2" ); -assert.neq( temp[z-1].shard , temp[z].shard , "A3" ); - -temp = db.data.find().sort( { 'sub.num' : 1 } ).toArray(); -assert.eq( N , temp.length , "B1" ); -for ( i=0; i<100; i++ ){ - assert.eq( i , temp[i].sub.num , "B2" ); -} - - -db.data.find().sort( { 'sub.num' : 1 } ).toArray(); -s.getPrimaryShard("test").getDB( "test" ).data.find().sort( { 'sub.num' : 1 } ).toArray(); - -a = Date.timeFunc( function(){ z = db.data.find().sort( { 'sub.num' : 1 } ).toArray(); } , 200 ); -assert.eq( 100 , z.length , "C1" ); -b = 1.5 * Date.timeFunc( function(){ - z = s.getPrimaryShard("test").getDB("test").data.find().sort({'sub.num' : 1}).toArray(); - }, 200 ); -assert.eq( 67 , z.length , "C2" ); - -print( "a: " + a + " b:" + b + " mongos slow down: " + Math.ceil( 100 * ( ( a - b ) / b ) ) + "%" ); - -// -- secondary index sorting - -function getSorted( by , dir , proj ){ - var s = {}; - s[by] = dir || 1; - printjson( s ); - var cur = db.data.find( {} , proj || {} ).sort( s ); - return terse( cur.map( function(z){ return z.sub.num; } ) ); -} - -function terse( a ){ - var s = ""; - for ( var i=0; i<a.length; i++ ){ - if ( i > 0 ) - s += ","; - s += a[i]; + var s = new ShardingTest({name: "sort1", shards: 2, mongos: 2}); + + s.adminCommand({enablesharding: "test"}); + s.ensurePrimaryShard('test', 'shard0001'); + s.adminCommand({shardcollection: "test.data", key: {'sub.num': 1}}); + + db = s.getDB("test"); + + N = 100; + + forward = []; + backward = []; + for (i = 0; i < N; i++) { + db.data.insert({_id: i, sub: {num: i, x: N - i}}); + forward.push(i); + backward.push((N - 1) - i); + } + + s.adminCommand({split: "test.data", middle: {'sub.num': 33}}); + s.adminCommand({split: "test.data", middle: {'sub.num': 66}}); + + s.adminCommand({ + movechunk: "test.data", + find: {'sub.num': 50}, + to: s.getOther(s.getPrimaryShard("test")).name, + waitForDelete: true + }); + + assert.lte(3, s.config.chunks.find().itcount(), "A1"); + + temp = s.config.chunks.find().sort({min: 1}).toArray(); + temp.forEach(printjsononeline); + + z = 0; + for (; z < temp.length; z++) + if (temp[z].min["sub.num"] <= 50 && temp[z].max["sub.num"] > 50) + break; + + assert.eq(temp[z - 1].shard, temp[z + 1].shard, "A2"); + assert.neq(temp[z - 1].shard, temp[z].shard, "A3"); + + temp = db.data.find().sort({'sub.num': 1}).toArray(); + assert.eq(N, temp.length, "B1"); + for (i = 0; i < 100; i++) { + assert.eq(i, temp[i].sub.num, "B2"); + } + + db.data.find().sort({'sub.num': 1}).toArray(); + s.getPrimaryShard("test").getDB("test").data.find().sort({'sub.num': 1}).toArray(); + + a = Date.timeFunc(function() { + z = db.data.find().sort({'sub.num': 1}).toArray(); + }, 200); + assert.eq(100, z.length, "C1"); + b = 1.5 * + Date.timeFunc(function() { + z = s.getPrimaryShard("test").getDB("test").data.find().sort({'sub.num': 1}).toArray(); + }, 200); + assert.eq(67, z.length, "C2"); + + print("a: " + a + " b:" + b + " mongos slow down: " + Math.ceil(100 * ((a - b) / b)) + "%"); + + // -- secondary index sorting + + function getSorted(by, dir, proj) { + var s = {}; + s[by] = dir || 1; + printjson(s); + var cur = db.data.find({}, proj || {}).sort(s); + return terse(cur.map(function(z) { + return z.sub.num; + })); + } + + function terse(a) { + var s = ""; + for (var i = 0; i < a.length; i++) { + if (i > 0) + s += ","; + s += a[i]; + } + return s; } - return s; -} -forward = terse(forward); -backward = terse(backward); + forward = terse(forward); + backward = terse(backward); -assert.eq( forward , getSorted( "sub.num" , 1 ) , "D1" ); -assert.eq( backward , getSorted( "sub.num" , -1 ) , "D2" ); + assert.eq(forward, getSorted("sub.num", 1), "D1"); + assert.eq(backward, getSorted("sub.num", -1), "D2"); -assert.eq( backward , getSorted( "sub.x" , 1 ) , "D3" ); -assert.eq( forward , getSorted( "sub.x" , -1 ) , "D4" ); + assert.eq(backward, getSorted("sub.x", 1), "D3"); + assert.eq(forward, getSorted("sub.x", -1), "D4"); -assert.eq( backward , getSorted( "sub.x" , 1 , { 'sub.num' : 1 } ) , "D5" ); -assert.eq( forward , getSorted( "sub.x" , -1 , { 'sub.num' : 1 } ) , "D6" ); + assert.eq(backward, getSorted("sub.x", 1, {'sub.num': 1}), "D5"); + assert.eq(forward, getSorted("sub.x", -1, {'sub.num': 1}), "D6"); -assert.eq( backward , getSorted( "sub.x" , 1 , { 'sub' : 1 } ) , "D7" ); -assert.eq( forward , getSorted( "sub.x" , -1 , { 'sub' : 1 } ) , "D8" ); + assert.eq(backward, getSorted("sub.x", 1, {'sub': 1}), "D7"); + assert.eq(forward, getSorted("sub.x", -1, {'sub': 1}), "D8"); -assert.eq( backward , getSorted( "sub.x" , 1 , { '_id' : 0 } ) , "D9" ); -assert.eq( forward , getSorted( "sub.x" , -1 , { '_id' : 0 } ) , "D10" ); + assert.eq(backward, getSorted("sub.x", 1, {'_id': 0}), "D9"); + assert.eq(forward, getSorted("sub.x", -1, {'_id': 0}), "D10"); -assert.eq( backward , getSorted( "sub.x" , 1 , { '_id' : 0, 'sub.num':1 } ) , "D11" ); -assert.eq( forward , getSorted( "sub.x" , -1 , { '_id' : 0, 'sub.num':1 } ) , "D12" ); + assert.eq(backward, getSorted("sub.x", 1, {'_id': 0, 'sub.num': 1}), "D11"); + assert.eq(forward, getSorted("sub.x", -1, {'_id': 0, 'sub.num': 1}), "D12"); -s.stop(); + s.stop(); })(); |