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

change union to match intersect

Browse Source 
accept aggregate and withscores
This commit is contained in:
barisusakli
2016-09-08 17:59:59 +03:00
parent 8d897d8dcf
commit eabb745116
3 changed files with 46 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/database/redis/sorted.js
View File

@@ -232,32 +232,34 @@ module.exports = function(redisClient, module) {
multi.exec(callback);
};
module.getSortedSetUnion = function(sets, start, stop, callback) {
sortedSetUnion('zrange', sets, start, stop, false, callback);
module.getSortedSetUnion = function(params, callback) {
params.method = 'zrange';
sortedSetUnion(params, callback);
};
module.getSortedSetRevUnion = function(sets, start, stop, callback) {
sortedSetUnion('zrevrange', sets, start, stop, false, callback);
module.getSortedSetRevUnion = function(params, callback) {
params.method = 'zrevrange';
sortedSetUnion(params, callback);
};
function sortedSetUnion(method, sets, start, stop, withScores, callback) {
function sortedSetUnion(params, callback) {
var tempSetName = 'temp_' + Date.now();
var params = [tempSetName, start, stop];
if (withScores) {
var rangeParams = [tempSetName, params.start, params.stop];
if (params.withScores) {
params.push('WITHSCORES');
}
var multi = redisClient.multi();
multi.zunionstore([tempSetName, sets.length].concat(sets));
multi[method](params);
multi.zunionstore([tempSetName, params.sets.length].concat(params.sets));
multi[params.method](rangeParams);
multi.del(tempSetName);
multi.exec(function(err, results) {
if (err) {
return callback(err);
}
if (!withScores) {
if (!params.withScores) {
return callback(null, results ? results[1] : null);
}
results = results[1] || [];