mirror of
https://github.com/NodeBB/NodeBB.git
synced 2025-11-03 12:36:02 +01:00
fast path for mongodb batches (otherwise it's O(n^2) memory, which gets ugly fast)
This commit is contained in:
@@ -23,6 +23,11 @@ var async = require('async'),
|
||||
return callback(new Error('[[error:process-not-a-function]]'));
|
||||
}
|
||||
|
||||
// use the fast path if possible
|
||||
if (db.processSortedSet && typeof options.doneIf !== 'function' && !utils.isNumber(options.alwaysStartAt)) {
|
||||
return db.processSortedSet(setKey, process, options.batch || DEFAULT_BATCH_SIZE, callback);
|
||||
}
|
||||
|
||||
// custom done condition
|
||||
options.doneIf = typeof options.doneIf === 'function' ? options.doneIf : function(){};
|
||||
|
||||
|
||||
@@ -533,4 +533,40 @@ module.exports = function(db, module) {
|
||||
callback(err, data);
|
||||
});
|
||||
};
|
||||
|
||||
module.processSortedSet = function(setKey, process, batch, callback) {
|
||||
var done = false, ids = [], cursor = db.collection('objects').
|
||||
find({_key: setKey}).
|
||||
sort({score: 1}).
|
||||
project({_id: 0, value: 1}).
|
||||
batchSize(batch);
|
||||
|
||||
async.whilst(
|
||||
function() {
|
||||
return !done;
|
||||
},
|
||||
function(next) {
|
||||
cursor.next(function(err, item) {
|
||||
if (err) {
|
||||
return next(err);
|
||||
}
|
||||
if (item === null) {
|
||||
done = true;
|
||||
} else {
|
||||
ids.push(item.value);
|
||||
}
|
||||
|
||||
if (ids.length < batch && (!done || ids.length === 0)) {
|
||||
return next(null);
|
||||
}
|
||||
|
||||
process(ids, function(err) {
|
||||
ids = [];
|
||||
return next(err);
|
||||
});
|
||||
});
|
||||
},
|
||||
callback
|
||||
);
|
||||
};
|
||||
};
|
||||
Reference in New Issue
Block a user