From 5222edb6243a103f2efc0cfad695464299a767e6 Mon Sep 17 00:00:00 2001
From: barisusakli <barisusakli@gmail.com>
Date: Mon, 8 Sep 2014 16:13:48 -0400
Subject: [PATCH] sortedSetsAdd

using mullti/bulk for redis/mongo
lets you add a score/value pair to multiple sorted sets at once
---
 src/database/level/sorted.js |  8 ++++++++
 src/database/mongo/sorted.js | 21 +++++++++++++++++++++
 src/database/redis/sorted.js | 13 +++++++++++++
 3 files changed, 42 insertions(+)

diff --git a/src/database/level/sorted.js b/src/database/level/sorted.js
index 369c820d53..ac681bb65b 100644
--- a/src/database/level/sorted.js
+++ b/src/database/level/sorted.js
@@ -20,6 +20,14 @@ module.exports = function(db, module) {
 		});
 	};
 
+	module.sortedSetsAdd = function(keys, score, value, callback) {
+		async.each(keys, function(key, next) {
+			module.sortedSetAdd(key, score, value, next);
+		}, function(err) {
+			callback(err);
+		});
+	};
+
 	module.sortedSetRemove = function(key, value, callback) {
 		module.getListRange(key, 0, -1, function(err, set) {
 			set = set.filter(function(a) {return a.value !== value.toString();});
diff --git a/src/database/mongo/sorted.js b/src/database/mongo/sorted.js
index c290b7000a..9e31335308 100644
--- a/src/database/mongo/sorted.js
+++ b/src/database/mongo/sorted.js
@@ -18,6 +18,26 @@ module.exports = function(db, module) {
 		});
 	};
 
+	module.sortedSetsAdd = function(keys, score, value, callback) {
+		callback = callback || helpers.noop;
+
+		value = helpers.valueToString(value);
+		var data = {
+			score: parseInt(score, 10),
+			value: value
+		};
+
+		var bulk = db.collection('objects').initializeUnorderedBulkOp();
+
+		for(var i=0; i<keys.length; ++i) {
+			bulk.find({_key: keys[i], value: value}).upsert().updateOne({$set: data});
+		}
+
+		bulk.execute(function(err, result) {
+			callback(err);
+		});
+	};
+
 	module.sortedSetRemove = function(key, value, callback) {
 		callback = callback || helpers.noop;
 		value = helpers.valueToString(value);
@@ -34,6 +54,7 @@ module.exports = function(db, module) {
 		db.collection('objects').remove({_key: {$in: keys}, value: value}, callback);
 	};
 
+
 	function getSortedSetRange(key, start, stop, sort, withScores, callback) {
 		db.collection('objects').find({_key:key}, {fields: {_id: 0, value: 1, score: 1}})
 			.limit(stop - start + 1)
diff --git a/src/database/redis/sorted.js b/src/database/redis/sorted.js
index 8dfc560d2b..1830e350ac 100644
--- a/src/database/redis/sorted.js
+++ b/src/database/redis/sorted.js
@@ -8,6 +8,19 @@ module.exports = function(redisClient, module) {
 		});
 	};
 
+	module.sortedSetsAdd = function(keys, score, value, callback) {
+		callback = callback || function() {};
+		var multi = redisClient.multi();
+
+		for(var i=0; i<keys.length; ++i) {
+			multi.zadd(keys[i], score, value);
+		}
+
+		multi.exec(function(err, res) {
+			callback(err);
+		});
+	};
+
 	module.sortedSetRemove = function(key, value, callback) {
 		callback = callback || function() {};
 		redisClient.zrem(key, value, function(err) {