summaryrefslogtreecommitdiff
path: root/jstests/mr_stored.js
blob: 7963d9892e17ddac55025dae820c03b8b3b460d3 (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()