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

feat: shorter setsCard

Browse Source
This commit is contained in:
Barış Soner Uşaklı
2019-06-22 23:43:17 -04:00
parent b3ed89f5fd
commit 3780a58ffb
2 changed files with 4 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -159,32 +159,7 @@ module.exports = function (db, module) {
if (!Array.isArray(keys) || !keys.length) { if (!Array.isArray(keys) || !keys.length) {
return callback(null, []); return callback(null, []);
} }
var pipeline = [ async.map(keys, module.sortedSetCard, callback);
{ $match: { _key: { $in: keys } } },
{ $group: { _id: { _key: '$_key' }, count: { $sum: 1 } } },
{ $project: { _id: 1, count: '$count' } },
];
db.collection('objects').aggregate(pipeline).toArray(function (err, results) {
if (err) {
return callback(err);
}
if (!Array.isArray(results)) {
results = [];
}
var map = {};
results.forEach(function (item) {
if (item && item._id._key) {
map[item._id._key] = item.count;
}
});
results = keys.map(function (key) {
return map[key] || 0;
});
callback(null, results);
});
}; };
module.sortedSetRank = function (key, value, callback) { module.sortedSetRank = function (key, value, callback) {

10
src/database/mongo/sorted/union.js
View File

@@ -6,16 +6,12 @@ module.exports = function (db, module) {
return callback(null, 0); return callback(null, 0);
} }
var pipeline = [ db.collection('objects').aggregate([
{ $match: { _key: { $in: keys } } }, { $match: { _key: { $in: keys } } },
{ $group: { _id: { value: '$value' } } }, { $group: { _id: { value: '$value' } } },
{ $group: { _id: null, count: { $sum: 1 } } }, { $group: { _id: null, count: { $sum: 1 } } },
]; { $project: { _id: 0, count: '$count' } },
]).toArray(function (err, data) {
var project = { _id: 0, count: '$count' };
pipeline.push({ $project: project });
db.collection('objects').aggregate(pipeline).toArray(function (err, data) {
callback(err, Array.isArray(data) && data.length ? data[0].count : 0); callback(err, Array.isArray(data) && data.length ? data[0].count : 0);
}); });
}; };