summaryrefslogtreecommitdiff
path: root/jstests/core/mr_merge.js
blob: 63e7fd7032b6e8eac0230c7e672270050c70fd81 (plain)
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
// Tests the 'merge' output mode for mapReduce.
// Cannot implicitly shard accessed collections because of following errmsg: Cannot output to a
// non-sharded collection because sharded collection exists already.
// The test runs commands that are not allowed with security token: mapReduce.
// @tags: [
//   not_allowed_with_security_token,
//   assumes_unsharded_collection,
//   # mapReduce does not support afterClusterTime.
//   does_not_support_causal_consistency,
//   does_not_support_stepdowns,
//   uses_map_reduce_with_temp_collections,
// ]
(function() {
"use strict";
(function() {
const source = db.mr_merge;
source.drop();

assert.commandWorked(source.insert({_id: 1, a: [1, 2]}));
assert.commandWorked(source.insert({_id: 2, a: [2, 3]}));
assert.commandWorked(source.insert({_id: 3, a: [3, 4]}));

const out = db.mr_merge_out;
const outName = out.getName();
out.drop();

const map = function() {
    for (let i = 0; i < this.a.length; i++)
        emit(this.a[i], 1);
};
const reduce = function(k, vs) {
    return Array.sum(vs);
};

assert.commandWorked(source.mapReduce(map, reduce, {out: outName}));

let expected = [{_id: 1, value: 1}, {_id: 2, value: 2}, {_id: 3, value: 2}, {_id: 4, value: 1}];
assert.docEq(expected, out.find().sort({_id: 1}).toArray());

assert.commandWorked(source.insert({_id: 4, a: [4, 5]}));
// Insert something that should be unaltered by the mapReduce into the output collection.
assert.commandWorked(out.insert({_id: 10, value: 5}));
assert.commandWorked(
    source.mapReduce(map, reduce, {out: {merge: outName}, query: {_id: {$gt: 3}}}));

expected = [
    {_id: 1, value: 1},
    {_id: 2, value: 2},
    {_id: 3, value: 2},
    {_id: 4, value: 1},
    {_id: 5, value: 1},
    {_id: 10, value: 5}
];
assert.docEq(expected, out.find().sort({_id: 1}).toArray());

assert.commandWorked(source.insert({_id: 5, a: [5, 6]}));
// Insert something that should be unaltered by the mapReduce into the output collection.
assert.commandWorked(out.insert({_id: 20, value: 10}));
assert.commandWorked(
    source.mapReduce(map, reduce, {out: {merge: outName}, query: {_id: {$gt: 4}}}));

expected = [
    {_id: 1, value: 1},
    {_id: 2, value: 2},
    {_id: 3, value: 2},
    {_id: 4, value: 1},
    {_id: 5, value: 1},
    {_id: 6, value: 1},
    {_id: 10, value: 5},
    {_id: 20, value: 10}
];
assert.docEq(expected, out.find().sort({_id: 1}).toArray());
}());
(function() {
const source = db.mr_merge;
source.drop();

assert.commandWorked(source.insert({_id: 1, x: 1}));
assert.commandWorked(source.insert({_id: 2, x: 1}));
assert.commandWorked(source.insert({_id: 3, x: 2}));

const out = db.mr_merge_out;
const outName = out.getName();
out.drop();

const map = function() {
    emit(this.x, 1);
};
const reduce = function(k, v) {
    return Array.sum(v);
};

assert.commandWorked(
    source.mapReduce(map, reduce, {out: {merge: outName}, query: {_id: {$gt: 0}}}));

assert.eq(2, out.findOne({_id: 1}).value);
assert.eq(1, out.findOne({_id: 2}).value);

assert.commandWorked(source.insert({_id: 4, x: 2}));
assert.commandWorked(
    source.mapReduce(map, reduce, {out: {merge: outName}, query: {_id: {$gt: 3}}}));

assert.eq(2, out.findOne({_id: 1}).value);
assert.eq(1, out.findOne({_id: 2}).value);
}());
}());