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;
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}));
|