#2095

src/batch.js

@@ -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));