summaryrefslogtreecommitdiff
path: root/jstests/core/mr2.js
blob: c13ff44797033157aae6c03d89c9dc4f148ea362 (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


t = db.mr2;
t.drop();

t.save({comments: [{who: "a", txt: "asdasdasd"}, {who: "b", txt: "asdasdasdasdasdasdas"}]});

t.save({comments: [{who: "b", txt: "asdasdasdaaa"}, {who: "c", txt: "asdasdasdaasdasdas"}]});

function m() {
    for (var i = 0; i < this.comments.length; i++) {
        var c = this.comments[i];
        emit(c.who, {totalSize: c.txt.length, num: 1});
    }
}

function r(who, values) {
    var n = {
        totalSize: 0,
        num: 0
    };
    for (var i = 0; i < values.length; i++) {
        n.totalSize += values[i].totalSize;
        n.num += values[i].num;
    }
    return n;
}

function reformat(r) {
    var x = {};
    var cursor;
    if (r.results)
        cursor = r.results;
    else
        cursor = r.find();
    cursor.forEach(function(z) {
        x[z._id] = z.value;
    });
    return x;
}

function f(who, res) {
    res.avg = res.totalSize / res.num;
    return res;
}

res = t.mapReduce(m, r, {finalize: f, out: "mr2_out"});
printjson(res);
x = reformat(res);
assert.eq(9, x.a.avg, "A1");
assert.eq(16, x.b.avg, "A2");
assert.eq(18, x.c.avg, "A3");
res.drop();

// inline does needs to exist - so set it to false to make sure the code is just checking for
// existence
res = t.mapReduce(m, r, {finalize: f, out: {inline: 0}});
printjson(res);
x = reformat(res);
assert.eq(9, x.a.avg, "B1");
assert.eq(16, x.b.avg, "B2");
assert.eq(18, x.c.avg, "B3");
res.drop();
assert(!("result" in res), "B4");

res = t.mapReduce(m, r, {finalize: f, out: "mr2_out", jsMode: true});
printjson(res);
x = reformat(res);
assert.eq(9, x.a.avg, "A1");
assert.eq(16, x.b.avg, "A2");
assert.eq(18, x.c.avg, "A3");
res.drop();

res = t.mapReduce(m, r, {finalize: f, out: {inline: 5}, jsMode: true});
printjson(res);
x = reformat(res);
assert.eq(9, x.a.avg, "B1");
assert.eq(16, x.b.avg, "B2");
assert.eq(18, x.c.avg, "B3");
res.drop();
assert(!("result" in res), "B4");