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


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

res = t.mapReduce( m , r , { finalize : f , out : { inline : 1 } } );
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" )