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
72
|
// Tests whether the geospatial search is stable under btree updates
//
// Tests the implementation of the 2d search, not the behavior we promise. MongoDB currently
// promises no isolation, so there is no guarantee that we get the results we expect in this file.
// The old query system, if it saw a 2d query, would never consider a collscan.
//
// The new query system can answer the queries in this file with a collscan and ranks
// the collscan against the indexed result.
//
// In order to expose the specific NON GUARANTEED isolation behavior this file tests
// we disable table scans to ensure that the new query system only looks at the 2d
// scan.
assert.commandWorked( db._adminCommand( { setParameter:1, notablescan:true } ) );
var status = function( msg ){
print( "\n\n###\n" + msg + "\n###\n\n" );
};
var coll = db.getCollection( "jstests_geo_update_btree2" );
coll.drop();
coll.ensureIndex( { loc : '2d' } );
status( "Inserting points..." );
var numPoints = 10;
Random.setRandomSeed();
for ( i = 0; i < numPoints; i++ ) {
coll.insert( { _id : i, loc : [ Random.rand() * 180, Random.rand() * 180 ], i : i % 2 } );
}
status( "Starting long query..." );
var query = coll.find({ loc : { $within : { $box : [[-180, -180], [180, 180]] } } }).batchSize( 2 );
var firstValues = [ query.next()._id, query.next()._id ];
printjson( firstValues );
status( "Removing points not returned by query..." );
var allQuery = coll.find();
var removeIds = [];
while( allQuery.hasNext() ){
var id = allQuery.next()._id;
if( firstValues.indexOf( id ) < 0 ){
removeIds.push( id );
}
}
var updateIds = [];
for( var i = 0, max = removeIds.length / 2; i < max; i++ ) updateIds.push( removeIds.pop() );
printjson( removeIds );
coll.remove({ _id : { $in : removeIds } });
status( "Updating points returned by query..." );
printjson(updateIds);
var big = new Array( 3000 ).toString();
for( var i = 0; i < updateIds.length; i++ )
coll.update({ _id : updateIds[i] }, { $set : { data : big } });
status( "Counting final points..." );
// It's not defined whether or not we return documents that are modified during a query. We
// shouldn't crash, but it's not defined how many results we get back. This test is modifying every
// doc not returned by the query, and since we currently handle the invalidation by removing them,
// we won't return them. But we shouldn't crash.
// assert.eq( ( numPoints - 2 ) / 2, query.itcount() )
query.itcount();
assert.commandWorked( db._adminCommand( { setParameter:1, notablescan:false} ) );
|
