Nemcio/NodeBB
1
0
Fork 0
mirror of https://github.com/NodeBB/NodeBB.git synced 2025-10-26 08:36:12 +01:00
Code Issues Packages Projects Releases Wiki Activity

refactor: dont sleep if iteration is done on first one

Browse Source
This commit is contained in:
Barış Soner Uşaklı
2023-07-24 15:23:59 -04:00
parent 0495b863a9
commit 0136e9248c
1 changed files with 11 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/batch.js
View File

@@ -40,20 +40,20 @@ exports.processSortedSet = async function (setKey, process, options) {
process = util.promisify(process);
}
const method = options.reverse ? 'getSortedSetRevRange' : 'getSortedSetRange';
let iteration = 1;
while (true) {
/* eslint-disable no-await-in-loop */
const ids = await db[`${method}${options.withScores ? 'WithScores' : ''}`](setKey, start, stop);
if (!ids.length || options.doneIf(start, stop, ids)) {
return;
}
await process(ids);
start += utils.isNumber(options.alwaysStartAt) ? options.alwaysStartAt : options.batch;
stop = start + options.batch - 1;
if (options.interval) {
if (iteration > 1 && options.interval) {
await sleep(options.interval);
}
await process(ids);
iteration += 1;
start += utils.isNumber(options.alwaysStartAt) ? options.alwaysStartAt : options.batch;
stop = start + options.batch - 1;
}
};
@@ -72,21 +72,20 @@ exports.processArray = async function (array, process, options) {
if (process && process.constructor && process.constructor.name !== 'AsyncFunction') {
process = util.promisify(process);
}
let iteration = 1;
while (true) {
const currentBatch = array.slice(start, start + batch);
if (!currentBatch.length) {
return;
}
if (iteration > 1 && options.interval) {
await sleep(options.interval);
}
await process(currentBatch);
start += batch;
if (options.interval) {
await sleep(options.interval);
}
iteration += 1;
}
};