'use strict'; module.exports = function (redisClient, module) { var _ = require('lodash'); var utils = require('../../utils'); var helpers = module.helpers.redis; require('./sorted/add')(redisClient, module); require('./sorted/remove')(redisClient, module); require('./sorted/union')(redisClient, module); require('./sorted/intersect')(redisClient, module); module.getSortedSetRange = function (key, start, stop, callback) { sortedSetRange('zrange', key, start, stop, false, callback); }; module.getSortedSetRevRange = function (key, start, stop, callback) { sortedSetRange('zrevrange', key, start, stop, false, callback); }; module.getSortedSetRangeWithScores = function (key, start, stop, callback) { sortedSetRange('zrange', key, start, stop, true, callback); }; module.getSortedSetRevRangeWithScores = function (key, start, stop, callback) { sortedSetRange('zrevrange', key, start, stop, true, callback); }; function sortedSetRange(method, key, start, stop, withScores, callback) { if (Array.isArray(key)) { if (!key.length) { return setImmediate(callback, null, []); } const batch = redisClient.batch(); key.forEach((key) => { batch[method]([key, start, stop, 'WITHSCORES']); }); batch.exec(function (err, data) { if (err) { return callback(err); } data = _.flatten(data); var objects = []; for (var i = 0; i < data.length; i += 2) { objects.push({ value: data[i], score: parseFloat(data[i + 1]) }); } objects.sort((a, b) => { if (method === 'zrange') { return a.score - b.score; } return b.score - a.score; }); if (withScores) { return callback(null, objects); } objects = objects.map(item => item.value); callback(null, objects); }); return; } var params = [key, start, stop]; if (withScores) { params.push('WITHSCORES'); } redisClient[method](params, function (err, data) { if (err) { return callback(err); } if (!withScores) { return callback(null, data); } var objects = []; for (var i = 0; i < data.length; i += 2) { objects.push({ value: data[i], score: parseFloat(data[i + 1]) }); } callback(null, objects); }); } module.getSortedSetRangeByScore = function (key, start, count, min, max, callback) { redisClient.zrangebyscore([key, min, max, 'LIMIT', start, count], callback); }; module.getSortedSetRevRangeByScore = function (key, start, count, max, min, callback) { redisClient.zrevrangebyscore([key, max, min, 'LIMIT', start, count], callback); }; module.getSortedSetRangeByScoreWithScores = function (key, start, count, min, max, callback) { sortedSetRangeByScoreWithScores('zrangebyscore', key, start, count, min, max, callback); }; module.getSortedSetRevRangeByScoreWithScores = function (key, start, count, max, min, callback) { sortedSetRangeByScoreWithScores('zrevrangebyscore', key, start, count, max, min, callback); }; function sortedSetRangeByScoreWithScores(method, key, start, count, min, max, callback) { redisClient[method]([key, min, max, 'WITHSCORES', 'LIMIT', start, count], function (err, data) { if (err) { return callback(err); } var objects = []; for (var i = 0; i < data.length; i += 2) { objects.push({ value: data[i], score: parseFloat(data[i + 1]) }); } callback(null, objects); }); } module.sortedSetCount = function (key, min, max, callback) { redisClient.zcount(key, min, max, callback); }; module.sortedSetCard = function (key, callback) { redisClient.zcard(key, callback); }; module.sortedSetsCard = function (keys, callback) { if (!Array.isArray(keys) || !keys.length) { return callback(null, []); } var batch = redisClient.batch(); for (var i = 0; i < keys.length; i += 1) { batch.zcard(keys[i]); } batch.exec(callback); }; module.sortedSetRank = function (key, value, callback) { redisClient.zrank(key, value, callback); }; module.sortedSetsRanks = function (keys, values, callback) { var batch = redisClient.batch(); for (var i = 0; i < values.length; i += 1) { batch.zrank(keys[i], values[i]); } batch.exec(callback); }; module.sortedSetRanks = function (key, values, callback) { var batch = redisClient.batch(); for (var i = 0; i < values.length; i += 1) { batch.zrank(key, values[i]); } batch.exec(callback); }; module.sortedSetRevRank = function (key, value, callback) { redisClient.zrevrank(key, value, callback); }; module.sortedSetScore = function (key, value, callback) { if (!key || value === undefined) { return callback(null, null); } redisClient.zscore(key, value, function (err, score) { if (err) { return callback(err); } if (score === null) { return callback(null, score); } callback(null, parseFloat(score)); }); }; module.sortedSetsScore = function (keys, value, callback) { helpers.execKeysValue(redisClient, 'batch', 'zscore', keys, value, function (err, scores) { if (err) { return callback(err); } scores = scores.map(function (d) { return d === null ? d : parseFloat(d); }); callback(null, scores); }); }; module.sortedSetScores = function (key, values, callback) { helpers.execKeyValues(redisClient, 'batch', 'zscore', key, values, function (err, scores) { if (err) { return callback(err); } scores = scores.map(function (d) { return d === null ? d : parseFloat(d); }); callback(null, scores); }); }; module.isSortedSetMember = function (key, value, callback) { module.sortedSetScore(key, value, function (err, score) { callback(err, utils.isNumber(score)); }); }; module.isSortedSetMembers = function (key, values, callback) { helpers.execKeyValues(redisClient, 'batch', 'zscore', key, values, function (err, results) { if (err) { return callback(err); } callback(null, results.map(Boolean)); }); }; module.isMemberOfSortedSets = function (keys, value, callback) { if (!Array.isArray(keys) || !keys.length) { return setImmediate(callback, null, []); } helpers.execKeysValue(redisClient, 'batch', 'zscore', keys, value, function (err, results) { if (err) { return callback(err); } callback(null, results.map(Boolean)); }); }; module.getSortedSetsMembers = function (keys, callback) { if (!Array.isArray(keys) || !keys.length) { return setImmediate(callback, null, []); } var batch = redisClient.batch(); for (var i = 0; i < keys.length; i += 1) { batch.zrange(keys[i], 0, -1); } batch.exec(callback); }; module.sortedSetIncrBy = function (key, increment, value, callback) { callback = callback || helpers.noop; redisClient.zincrby(key, increment, value, function (err, newValue) { callback(err, !err ? parseFloat(newValue) : undefined); }); }; module.getSortedSetRangeByLex = function (key, min, max, start, count, callback) { sortedSetLex('zrangebylex', false, key, min, max, start, count, callback); }; module.getSortedSetRevRangeByLex = function (key, max, min, start, count, callback) { sortedSetLex('zrevrangebylex', true, key, max, min, start, count, callback); }; module.sortedSetRemoveRangeByLex = function (key, min, max, callback) { callback = callback || helpers.noop; sortedSetLex('zremrangebylex', false, key, min, max, function (err) { callback(err); }); }; module.sortedSetLexCount = function (key, min, max, callback) { sortedSetLex('zlexcount', false, key, min, max, callback); }; function sortedSetLex(method, reverse, key, min, max, start, count, callback) { callback = callback || start; var minmin; var maxmax; if (reverse) { minmin = '+'; maxmax = '-'; } else { minmin = '-'; maxmax = '+'; } if (min !== minmin && !min.match(/^[[(]/)) { min = '[' + min; } if (max !== maxmax && !max.match(/^[[(]/)) { max = '[' + max; } if (count) { redisClient[method]([key, min, max, 'LIMIT', start, count], callback); } else { redisClient[method]([key, min, max], callback); } } };