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
|
// SERVER-6179: support for two $groups in sharded agg
(function() {
'use strict';
var s = new ShardingTest({shards: 2});
assert.commandWorked(s.s0.adminCommand({enablesharding: "test"}));
s.ensurePrimaryShard('test', 'shard0001');
assert.commandWorked(s.s0.adminCommand({shardcollection: "test.data", key: {_id: 1}}));
var d = s.getDB("test");
// Insert _id values 0 - 99
var N = 100;
var bulkOp = d.data.initializeOrderedBulkOp();
for (var i = 0; i < N; ++i) {
bulkOp.insert({_id: i, i: i % 10});
}
bulkOp.execute();
// Split the data into 3 chunks
assert.commandWorked(s.s0.adminCommand({split: "test.data", middle: {_id: 33}}));
assert.commandWorked(s.s0.adminCommand({split: "test.data", middle: {_id: 66}}));
// Migrate the middle chunk to another shard
assert.commandWorked(s.s0.adminCommand(
{movechunk: "test.data", find: {_id: 50}, to: s.getOther(s.getPrimaryShard("test")).name}));
// Check that we get results rather than an error
var result = d.data
.aggregate({$group: {_id: '$_id', i: {$first: '$i'}}},
{$group: {_id: '$i', avg_id: {$avg: '$_id'}}},
{$sort: {_id: 1}})
.toArray();
var expected = [
{"_id": 0, "avg_id": 45},
{"_id": 1, "avg_id": 46},
{"_id": 2, "avg_id": 47},
{"_id": 3, "avg_id": 48},
{"_id": 4, "avg_id": 49},
{"_id": 5, "avg_id": 50},
{"_id": 6, "avg_id": 51},
{"_id": 7, "avg_id": 52},
{"_id": 8, "avg_id": 53},
{"_id": 9, "avg_id": 54}
];
assert.eq(result, expected);
s.stop();
})();
|
