jstests/sortg.js


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

// Test that a memory exception is triggered for in memory sorts, but not for indexed sorts.

t = db.jstests_sortg;
t.drop();

big = new Array( 1000000 ).toString()

for( i = 0; i < 100; ++i ) {
    t.save( {b:0} );
}

for( i = 0; i < 40; ++i ) {
    t.save( {a:big} );
}

function memoryException( sortSpec, querySpec ) {
    querySpec = querySpec || {};
    assert.throws( function() {
                  t.find( querySpec ).sort( sortSpec ).batchSize( 1000 ).itcount()
                  } );
    assert( db.getLastError().match( /too much data for sort\(\) with no index/ ) );
    assert.throws( function() {
                  t.find( querySpec ).sort( sortSpec ).batchSize( 1000 ).explain( true )
                  } );
    assert( db.getLastError().match( /too much data for sort\(\) with no index/ ) );
}

function noMemoryException( sortSpec, querySpec ) {
    querySpec = querySpec || {};
    t.find( querySpec ).sort( sortSpec ).batchSize( 1000 ).itcount();
    if ( 0 ) { // SERVER-5016
    assert( !db.getLastError() );
    }
    t.find( querySpec ).sort( sortSpec ).batchSize( 1000 ).explain( true );
    if ( 0 ) { // SERVER-5016
    assert( !db.getLastError() );
    }
}

// Unindexed sorts.
memoryException( {a:1} );
memoryException( {b:1} );

// Indexed sorts.
noMemoryException( {_id:1} );
noMemoryException( {$natural:1} );

t.ensureIndex( {a:1} );
t.ensureIndex( {b:1} );
t.ensureIndex( {c:1} );

// These sorts are now indexed.
noMemoryException( {a:1} );
noMemoryException( {b:1} );

// A memory exception is triggered for an unindexed sort involving multiple plans.
memoryException( {d:1}, {b:null,c:null} );

// With an indexed plan on _id:1 and an unindexed plan on b:1, the indexed plan
// should succeed even if the unindexed one would exhaust its memory limit.
noMemoryException( {_id:1}, {b:null} );

// With an unindexed plan on b:1 recorded for a query, the query should be
// retried when the unindexed plan exhausts its memory limit.
assert.eq( 'BtreeCursor b_1', t.find( {b:0} ).sort( {_id:1} ).explain().cursor ); // Record b:1 plan
noMemoryException( {_id:1}, {b:null} );