hyperzlib
/
nodebb
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

change batch.js style

Browse Source
v1.18.x
barisusakli 8 years ago
parent 264e7ee303
commit 01e7b864b8
1 changed files with 98 additions and 102 deletions
Show Stats Download Patch File Download Diff File

200
src/batch.js
Unescape Escape View File

@ -2,113 +2,109 @@
'use strict'; 'use strict';
var async = require('async'), var async = require('async');
db = require('./database'), var db = require('./database');
utils = require('../public/src/utils'); var utils = require('../public/src/utils');
(function (Batch) { var DEFAULT_BATCH_SIZE = 100;
var DEFAULT_BATCH_SIZE = 100; exports.processSortedSet = function (setKey, process, options, callback) {
if (typeof options === 'function') {
Batch.processSortedSet = function (setKey, process, options, callback) { callback = options;
if (typeof options === 'function') { options = {};
callback = options; }
options = {};
} callback = typeof callback === 'function' ? callback : function () {};
options = options || {};
callback = typeof callback === 'function' ? callback : function () {};
options = options || {}; if (typeof process !== 'function') {
return callback(new Error('[[error:process-not-a-function]]'));
if (typeof process !== 'function') { }
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)) {
// use the fast path if possible return db.processSortedSet(setKey, process, options.batch || DEFAULT_BATCH_SIZE, callback);
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 () {};
// custom done condition
options.doneIf = typeof options.doneIf === 'function' ? options.doneIf : function () {}; var batch = options.batch || DEFAULT_BATCH_SIZE;
var start = 0;
var batch = options.batch || DEFAULT_BATCH_SIZE; var stop = batch;
var start = 0; var done = false;
var stop = batch;
var done = false; async.whilst(
function () {
async.whilst( return !done;
function () { },
return !done; function (next) {
}, db.getSortedSetRange(setKey, start, stop, function (err, ids) {
function (next) { if (err) {
db.getSortedSetRange(setKey, start, stop, function (err, ids) { return next(err);
if (err) { }
return next(err); if (!ids.length || options.doneIf(start, stop, ids)) {
}
if (!ids.length || options.doneIf(start, stop, ids)) {
done = true;
return next();
}
process(ids, function (err) {
if (err) {
return next(err);
}
start += utils.isNumber(options.alwaysStartAt) ? options.alwaysStartAt : batch + 1;
stop = start + batch;
next();
});
});
},
callback
);
};
Batch.processArray = function (array, process, options, callback) {
if (typeof options === 'function') {
callback = options;
options = {};
}
callback = typeof callback === 'function' ? callback : function () {};
options = options || {};
if (!Array.isArray(array) || !array.length) {
return callback();
}
if (typeof process !== 'function') {
return callback(new Error('[[error:process-not-a-function]]'));
}
var batch = options.batch || DEFAULT_BATCH_SIZE;
var start = 0;
var done = false;
async.whilst(
function () {
return !done;
},
function (next) {
var currentBatch = array.slice(start, start + batch);
if (!currentBatch.length) {
done = true; done = true;
return next(); return next();
} }
process(currentBatch, function (err) { process(ids, function (err) {
if (err) { if (err) {
return next(err); return next(err);
} }
start = start + batch; start += utils.isNumber(options.alwaysStartAt) ? options.alwaysStartAt : batch + 1;
if (options.interval) { stop = start + batch;
setTimeout(next, options.interval); next();
} else {
next();
}
}); });
}, });
function (err) { },
callback(err); callback
);
};
exports.processArray = function (array, process, options, callback) {
if (typeof options === 'function') {
callback = options;
options = {};
}
callback = typeof callback === 'function' ? callback : function () {};
options = options || {};
if (!Array.isArray(array) || !array.length) {
return callback();
}
if (typeof process !== 'function') {
return callback(new Error('[[error:process-not-a-function]]'));
}
var batch = options.batch || DEFAULT_BATCH_SIZE;
var start = 0;
var done = false;
async.whilst(
function () {
return !done;
},
function (next) {
var currentBatch = array.slice(start, start + batch);
if (!currentBatch.length) {
done = true;
return next();
} }
); process(currentBatch, function (err) {
}; if (err) {
return next(err);
}(exports)); }
start = start + batch;
if (options.interval) {
setTimeout(next, options.interval);
} else {
next();
}
});
},
function (err) {
callback(err);
}
);
};

Write Preview
Loading…
Cancel
Save