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


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 = {};
    r.find().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 } );
x = reformat( res );
assert.eq( 9 , x.a.avg , "A" );
assert.eq( 16 , x.b.avg , "B" );
assert.eq( 18 , x.c.avg , "C" );
res.drop();