v1.18.x
parent
aae7f2434d
commit
b48d81310c
@ -0,0 +1,61 @@
|
||||
|
||||
|
||||
'use strict';
|
||||
|
||||
var async = require('async'),
|
||||
db = require('./database'),
|
||||
utils = require('../public/src/utils');
|
||||
|
||||
(function(Batch) {
|
||||
|
||||
var DEFAULT_BATCH_SIZE = 100;
|
||||
|
||||
Batch.processSortedSet = function(setKey, process, options, callback) {
|
||||
if (typeof options === 'function') {
|
||||
callback = options;
|
||||
options = {};
|
||||
}
|
||||
|
||||
callback = typeof callback === 'function' ? callback : function(){};
|
||||
options = options || {};
|
||||
|
||||
if (typeof process !== 'function') {
|
||||
return callback(new Error('[[error:process-not-a-function]]'));
|
||||
}
|
||||
|
||||
// custom done condition
|
||||
options.doneIf = typeof options.doneIf === 'function' ? options.doneIf : function(){};
|
||||
|
||||
var batch = options.batch || DEFAULT_BATCH_SIZE;
|
||||
var start = 0;
|
||||
var end = batch;
|
||||
var done = false;
|
||||
|
||||
async.whilst(
|
||||
function() {
|
||||
return !done;
|
||||
},
|
||||
function(next) {
|
||||
db.getSortedSetRange(setKey, start, end, function(err, ids) {
|
||||
if (err) {
|
||||
return next(err);
|
||||
}
|
||||
if (!ids.length || options.doneIf(start, end, ids)) {
|
||||
done = true;
|
||||
return next();
|
||||
}
|
||||
process(err, ids, function(err) {
|
||||
if (err) {
|
||||
return next(err);
|
||||
}
|
||||
start += utils.isNumber(options.alwaysStartAt) ? options.alwaysStartAt : batch + 1;
|
||||
end = start + batch;
|
||||
next();
|
||||
});
|
||||
});
|
||||
},
|
||||
callback
|
||||
);
|
||||
};
|
||||
|
||||
}(exports));
|
Loading…
Reference in New Issue