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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
// Test various cursor behaviors
//
// @tags: [
// # This test attempts to enable profiling on a server and then get profiling data by reading
// # nodes the "system.profile" collection. The former operation must be routed to the primary in
// # a replica set, whereas the latter may be routed to a secondary.
// assumes_read_preference_unchanged,
// does_not_support_stepdowns,
// requires_capped,
// requires_getmore,
// requires_profiling,
// sbe_incompatible,
// ]
var testDB = db.getSiblingDB("geo_s2cursorlimitskip");
var t = testDB.geo_s2getmmm;
t.drop();
t.createIndex({geo: "2dsphere"});
Random.setRandomSeed();
var random = Random.rand;
/*
* To test that getmore is working within 2dsphere index.
* We insert a bunch of points and get a cursor. We then
* fetch some of the points and verify that the number of
* query and getmore operations are correct. Finally, we
* fetch the rest of the points and again verify that the
* number of query and getmore operations are correct.
*/
function sign() {
return random() > 0.5 ? 1 : -1;
}
function insertRandomPoints(num, minDist, maxDist) {
for (var i = 0; i < num; i++) {
var lat = sign() * (minDist + random() * (maxDist - minDist));
var lng = sign() * (minDist + random() * (maxDist - minDist));
var point = {geo: {type: "Point", coordinates: [lng, lat]}};
assert.commandWorked(t.insert(point));
}
}
var totalPointCount = 200;
var initialAdvance = 10;
var batchSize = 4;
// Insert points between 0.01 and 1.0 away.
insertRandomPoints(totalPointCount, 0.01, 1.0);
var cursor = t.find({
geo: {$geoNear: {$geometry: {type: "Point", coordinates: [0.0, 0.0]}}}
}).batchSize(batchSize);
assert.eq(cursor.count(), totalPointCount);
// Disable profiling in order to drop the system.profile collection.
// Then enable profiling for all operations. This is acceptable because
// our test is blacklisted from the parallel suite.
testDB.setProfilingLevel(0);
testDB.system.profile.drop();
// Create 4MB system.profile collection to prevent the 'getmore' operations from overwriting the
// original query.
assert.commandWorked(
testDB.createCollection("system.profile", {capped: true, size: 4 * 1024 * 1024}));
testDB.setProfilingLevel(2);
for (var j = 0; j < initialAdvance; j++) {
assert(cursor.hasNext());
cursor.next();
}
// We looked at (totalPointCount - initialAdvance) points, there should still be more.
assert(cursor.hasNext());
// Cursor was advanced 10 times, batchSize=4 => 1 query + 2 getmore.
assert.eq(1, testDB.system.profile.count({op: "query", ns: t.getFullName()}));
assert.eq(2, testDB.system.profile.count({op: "getmore", ns: t.getFullName()}));
for (var k = initialAdvance; k < totalPointCount; k++) {
assert(cursor.hasNext());
var tmpPoint = cursor.next();
}
// Cursor was advanced 200 times, batchSize=4 => 1 query + 49 getmore.
assert.eq(1, testDB.system.profile.count({op: "query", ns: t.getFullName()}));
assert.eq(49, testDB.system.profile.count({op: "getmore", ns: t.getFullName()}));
// Disable profiling again - no longer needed for remainder of test
testDB.setProfilingLevel(0);
testDB.system.profile.drop();
// Shouldn't be any more points to look at now.
assert(!cursor.hasNext());
var someLimit = 23;
// Make sure limit does something.
cursor = t.find({
geo: {$geoNear: {$geometry: {type: "Point", coordinates: [0.0, 0.0]}}}
}).limit(someLimit);
// Count doesn't work here -- ignores limit/skip, so we use itcount.
assert.eq(cursor.itcount(), someLimit);
// Make sure skip works by skipping some stuff ourselves.
var someSkip = 3;
cursor = t.find({
geo: {$geoNear: {$geometry: {type: "Point", coordinates: [0.0, 0.0]}}}
}).limit(someLimit + someSkip);
for (var i = 0; i < someSkip; ++i) {
cursor.next();
}
var cursor2 = t.find({geo: {$geoNear: {$geometry: {type: "Point", coordinates: [0.0, 0.0]}}}})
.skip(someSkip)
.limit(someLimit);
while (cursor.hasNext()) {
assert(cursor2.hasNext());
assert.eq(cursor.next(), cursor2.next());
}
|
