/
geo_update_btree2.js
52 lines (35 loc) · 1.34 KB
/
geo_update_btree2.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
// Tests whether the geospatial search is stable under btree updates
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
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..." )
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..." )
assert.eq( ( numPoints - 2 ) / 2, query.itcount() )