NodeBB/src/database/redis/sets.js

"use strict";

module.exports = function(redisClient, module) {
	var helpers = module.helpers.redis;

	module.setAdd = function(key, value, callback) {
		callback = callback || function() {};
		if (!Array.isArray(value)) {
			value = [value];
		}
		if (!value.length) {
			return callback();
		}
		redisClient.sadd(key, value, function(err, res) {
			callback(err);
		});
	};

	module.setsAdd = function(keys, value, callback) {
		callback = callback || function() {};
		helpers.multiKeysValue(redisClient, 'sadd', keys, value, function(err, res) {
			callback(err);
		});
	};

	module.setRemove = function(key, value, callback) {
		callback = callback || function() {};
		redisClient.srem(key, value, function(err, res) {
			callback(err);
		});
	};

	module.setsRemove = function(keys, value, callback) {
		callback = callback || function() {};
		helpers.multiKeysValue(redisClient, 'srem', keys, value, function(err, res) {
			callback(err);
		});
	};

	module.isSetMember = function(key, value, callback) {
		redisClient.sismember(key, value, function(err, result) {
			callback(err, result === 1);
		});
	};

	module.isSetMembers = function(key, values, callback) {
		helpers.multiKeyValues(redisClient, 'sismember', key, values, function(err, results) {
			callback(err, results ? helpers.resultsToBool(results) : null);
		});
	};

	module.isMemberOfSets = function(sets, value, callback) {
		helpers.multiKeysValue(redisClient, 'sismember', sets, value, function(err, results) {
			callback(err, results ? helpers.resultsToBool(results) : null);
		});
	};

	module.getSetMembers = function(key, callback) {
		redisClient.smembers(key, callback);
	};

	module.getSetsMembers = function(keys, callback) {
		helpers.multiKeys(redisClient, 'smembers', keys, callback);
	};

	module.setCount = function(key, callback) {
		redisClient.scard(key, callback);
	};

	module.setsCount = function(keys, callback) {
		helpers.multiKeys(redisClient, 'scard', keys, callback);
	};

	module.setRemoveRandom = function(key, callback) {
		callback = callback || function() {};
		redisClient.spop(key, callback);
	};

	return module;
};