-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathindex.js
62 lines (50 loc) · 1.61 KB
/
index.js
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
var trigger = require('level-trigger')
var liveStream = require('level-live-stream')
var viewStream = require('level-view-stream')
var Bucket = require('range-bucket')
var map = require('map-stream')
var through = require('through')
module.exports = function (db) {
if(db.reduce) return
trigger(db)
liveStream(db)
var views = {}
db.reduce = {views: views}
db.reduce.view = viewStream(db, db.reduce)
db.reduce.add = function (view) {
if(!view.name) throw new Error('reduce must have name')
views[view.name] = view
view.bucket = Bucket('mapr', view.name)
view.depth = (view.depth && view.depth > 0) ? view.depth : 0
var range = view.bucket.range()
function doReduce (key, cb) {
var collection = view.initial, values = []
key = JSON.parse(key)
key.push(true)
db.readStream(view.bucket.range(key))
.on('data', function (data) {
collection = view.reduce(collection, data.value, data.key)
})
.on('end', function () {
//save the collection
//get the parent group
var _key = key.slice(); _key.pop()
db.put(view.bucket(_key), collection, cb)
db.emit('reduce', view.name, _key, collection)
db.emit('reduce:'+view.name, _key, collection)
})
}
db.trigger.add({
start: range.start,
end : range.end,
name : 'RED-'+range.name,
map : function (data) {
var key = view.bucket.parse(data.key).key
if(key.length <= view.depth) return
key.pop()
return JSON.stringify(key)
},
job : doReduce
})
}
}