Nemcio/NodeBB
1
0
Fork 0
mirror of https://github.com/NodeBB/NodeBB.git synced 2026-01-03 14:20:33 +01:00
Code Issues Packages Projects Releases Wiki Activity

sortedSetIntersectionCard

Browse Source
This commit is contained in:
barisusakli
2016-09-16 13:01:11 +03:00
parent 814c4103ad
commit 990ecc8cd2
4 changed files with 78 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/database/mongo/sorted.js
View File

@@ -637,6 +637,24 @@ module.exports = function(db, module) {
};
module.sortedSetIntersectCard = function(keys, callback) {
if (!Array.isArray(keys) || !keys.length) {
return callback(null, 0);
}
var pipeline = [
{ $match: { _key: {$in: keys}} },
{ $group: { _id: {value: '$value'}, count: {$sum: 1}} },
{ $match: { count: keys.length} },
{ $group: { _id: null, count: { $sum: 1 } } },
{ $project: { _id: 0, count: '$count' } }
];
db.collection('objects').aggregate(pipeline, function(err, data) {
callback(err, Array.isArray(data) && data.length ? data[0].count : 0);
});
};
module.getSortedSetIntersect = function(params, callback) {
params.sort = 1;
getSortedSetRevIntersect(params, callback);

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

@@ -93,7 +93,7 @@ module.exports = function(redisClient, module) {
function sortedSetRange(method, key, start, stop, withScores, callback) {
if (Array.isArray(key)) {
return sortedSetUnion(method, key, start, stop, withScores, callback);
return sortedSetUnion({method: method, sets: key, start: start, stop: stop, withScores: withScores}, callback);
}
var params = [key, start, stop];
@@ -265,7 +265,7 @@ module.exports = function(redisClient, module) {
var rangeParams = [tempSetName, params.start, params.stop];
if (params.withScores) {
params.push('WITHSCORES');
rangeParams.push('WITHSCORES');
}
var multi = redisClient.multi();
@@ -302,6 +302,27 @@ module.exports = function(redisClient, module) {
redisClient.zrangebylex([key, min, max, 'LIMIT', start, count], callback);
};
module.sortedSetIntersectCard = function(keys, callback) {
if (!Array.isArray(keys) || !keys.length) {
return callback(null, 0);
}
var tempSetName = 'temp_' + Date.now();
var interParams = [tempSetName, keys.length].concat(keys);
var multi = redisClient.multi();
multi.zinterstore(interParams);
multi.zcard(tempSetName);
multi.del(tempSetName);
multi.exec(function(err, results) {
if (err) {
return callback(err);
}
callback(null, results[1] || 0);
});
};
module.getSortedSetIntersect = function(params, callback) {
params.method = 'zrange';
getSortedSetRevIntersect(params, callback);
@@ -312,7 +333,7 @@ module.exports = function(redisClient, module) {
getSortedSetRevIntersect(params, callback);
};
function getSortedSetRevIntersect (params, callback) {
function getSortedSetRevIntersect(params, callback) {
var sets = params.sets;
var start = params.hasOwnProperty('start') ? params.start : 0;
var stop = params.hasOwnProperty('stop') ? params.stop : -1;

2
src/topics.js
View File

@@ -346,7 +346,7 @@ var social = require('./social');
};
Topics.getTopicBookmarks = function( tid, callback ){
db.getSortedSetRangeWithScores(['tid:' + tid + ':bookmarks'], 0, -1, callback );
db.getSortedSetRangeWithScores(['tid:' + tid + ':bookmarks'], 0, -1, callback);
};
Topics.updateTopicBookmarks = function(tid, pids, callback) {

35
tests/database/sorted.js
View File

@@ -659,6 +659,41 @@ describe('Sorted Set methods', function() {
});
describe('sortedSetIntersectCard', function() {
before(function(done) {
async.parallel([
function(next) {
db.sortedSetAdd('interCard1', [0, 0, 0], ['value1', 'value2', 'value3'], next);
},
function(next) {
db.sortedSetAdd('interCard2', [0, 0, 0], ['value2', 'value3', 'value4'], next);
},
function(next) {
db.sortedSetAdd('interCard3', [0, 0, 0], ['value3', 'value4', 'value5'], next);
},
function(next) {
db.sortedSetAdd('interCard4', [0, 0, 0], ['value4', 'value5', 'value6'], next);
}
], done);
});
it('should return # of elements in intersection', function(done) {
db.sortedSetIntersectCard(['interCard1', 'interCard2', 'interCard3'], function(err, count) {
assert.ifError(err);
assert.strictEqual(count, 1);
done();
});
});
it('should return 0 if intersection is empty', function(done) {
db.sortedSetIntersectCard(['interCard1', 'interCard4'], function(err, count) {
assert.ifError(err);
assert.strictEqual(count, 0);
done();
});
});
});
after(function(done) {
db.flushdb(done);