summaryrefslogtreecommitdiff
path: root/jstests/core/mr_stored.js
blob: 3403411ea70fff78a92749ae68277b6a6e010170 (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

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

t.save( { "partner" : 1, "visits" : 9 } );
t.save( { "partner" : 2, "visits" : 9 } );
t.save( { "partner" : 1, "visits" : 11 } );
t.save( { "partner" : 1, "visits" : 30 } );
t.save( { "partner" : 2, "visits" : 41 } );
t.save( { "partner" : 2, "visits" : 41 } );

m = function(obj){
    emit( obj.partner , { stats : [ obj.visits ] } );
};

r = function( k , v ){
    var stats = [];
    var total = 0;
    for ( var i=0; i<v.length; i++ ){
        for ( var j in v[i].stats ) {
            stats.push( v[i].stats[j] );
            total += v[i].stats[j];
        }
    }
    return { stats : stats , total : total };
};

// Test that map reduce works with stored javascript
db.system.js.save( { _id : "mr_stored_map" , value : m } );
db.system.js.save( { _id : "mr_stored_reduce" , value : r } );

res = t.mapReduce( function () { mr_stored_map(this); } ,
                   function ( k , v ) { return mr_stored_reduce( k , v ); } ,
                   { out : "mr_stored_out" , scope : { xx : 1 } } );
//res.find().forEach( printjson )

z = res.convertToSingleObject();
assert.eq( 2 , Object.keySet( z ).length , "A1" );
assert.eq( [ 9 , 11 , 30 ] , z["1"].stats , "A2" );
assert.eq( [ 9 , 41 , 41 ] , z["2"].stats , "A3" );


res.drop();

m = function(obj){
    var x = "partner";
    var y = "visits";
    emit( obj[x] , { stats : [ obj[y] ] } );
};

db.system.js.save( { _id : "mr_stored_map" , value : m } );

res = t.mapReduce( function () { mr_stored_map(this); } ,
                   function ( k , v ) { return mr_stored_reduce( k , v ); } ,
                   { out : "mr_stored_out" , scope : { xx : 1 } } );
//res.find().forEach( printjson )

z = res.convertToSingleObject();
assert.eq( 2 , Object.keySet( z ).length , "B1" );
assert.eq( [ 9 , 11 , 30 ] , z["1"].stats , "B2" );
assert.eq( [ 9 , 41 , 41 ] , z["2"].stats , "B3" );

db.system.js.remove( { _id : "mr_stored_map" } );
db.system.js.remove( { _id : "mr_stored_reduce" } );

res.drop();