Files
NodeBB/src/database/redis.js

152 lines
3.5 KiB
JavaScript
Raw Normal View History

2013-12-02 15:45:15 -05:00
2013-12-02 15:46:25 -05:00
(function(module) {
'use strict';
2013-12-02 16:19:30 -05:00
var redisClient,
2013-12-02 15:46:25 -05:00
redis = require('redis'),
winston = require('winston'),
nconf = require('nconf'),
redis_socket_or_host = nconf.get('redis:host'),
2013-12-02 16:19:30 -05:00
utils = require('./../../public/src/utils.js');
2013-12-02 15:46:25 -05:00
if (redis_socket_or_host && redis_socket_or_host.indexOf('/')>=0) {
/* If redis.host contains a path name character, use the unix dom sock connection. ie, /tmp/redis.sock */
2013-12-02 16:19:30 -05:00
redisClient = redis.createClient(nconf.get('redis:host'));
2013-12-02 15:46:25 -05:00
} else {
/* Else, connect over tcp/ip */
2013-12-02 16:19:30 -05:00
redisClient = redis.createClient(nconf.get('redis:port'), nconf.get('redis:host'));
2013-12-02 15:46:25 -05:00
}
if (nconf.get('redis:password')) {
2013-12-02 16:19:30 -05:00
redisClient.auth(nconf.get('redis:password'));
2013-12-02 15:46:25 -05:00
}
var db = parseInt(nconf.get('redis:database'), 10);
if (db){
2013-12-02 16:19:30 -05:00
redisClient.select(db, function(error) {
if(error) {
2013-12-02 15:46:25 -05:00
winston.error("NodeBB could not connect to your Redis database. Redis returned the following error: " + error.message);
process.exit();
}
});
}
/*
* A possibly more efficient way of doing multiple sismember calls
*/
2013-12-02 16:19:30 -05:00
function sismembers(key, needles, callback) {
2013-12-02 15:46:25 -05:00
var tempkey = key + ':temp:' + utils.generateUUID();
2013-12-02 16:19:30 -05:00
redisClient.sadd(tempkey, needles, function() {
redisClient.sinter(key, tempkey, function(err, data) {
redisClient.del(tempkey);
2013-12-02 15:46:25 -05:00
callback(err, data);
});
});
};
2013-12-02 16:19:30 -05:00
//
// Exported functions
//
2013-12-02 16:23:14 -05:00
module.getFileName = function(callback) {
var multi = redisClient.multi();
multi.config('get', 'dir');
multi.config('get', 'dbfilename');
multi.exec(function (err, results) {
if (err) {
return callback(err);
}
results = results.reduce(function (memo, config) {
memo[config[0]] = config[1];
return memo;
}, {});
var dbFile = path.join(results.dir, results.dbfilename);
callback(null, dbFile);
});
}
2013-12-02 16:19:30 -05:00
module.setObject = function(key, data, callback) {
redisClient.hmset(key, data, callback);
}
2013-12-02 15:46:25 -05:00
2013-12-02 16:19:30 -05:00
module.setObjectField = function(key, field, callback) {
redisClient.hset(key, field, callback)
}
module.getObject = function(key, callback) {
redisClient.hgetall(key, callback)
}
module.getObjectField = function(key, field, callback) {
module.getObjectFields(key, [field], function(err, data) {
if(err) {
return callback(err);
}
callback(null, data[field]);
});
}
module.getObjectFields = function(key, fields, callback) {
redisClient.hmget(key, fields, function(err, data) {
2013-12-02 15:46:25 -05:00
if(err) {
return callback(err, null);
}
var returnData = {};
for (var i = 0, ii = fields.length; i < ii; ++i) {
returnData[fields[i]] = data[i];
}
callback(null, returnData);
});
2013-12-02 16:19:30 -05:00
}
module.deleteObjectField = function(key, field, callback) {
redisClient.hdel(key, field, callback);
}
2013-12-02 17:10:26 -05:00
module.incrObjectField = function(key, field, callback) {
redisClient.hincrby(key, field, 1, callback);
2013-12-02 16:19:30 -05:00
}
2013-12-02 17:10:26 -05:00
module.incrObjectFieldBy = function(key, field, value, callback) {
redisClient.hincrby(key, field, value, callback);
}
2013-12-02 16:19:30 -05:00
module.setAdd = function(key, value, callback) {
redisClient.sadd(key, value, callback);
}
module.setRemove = function(key, value, callback) {
redisClient.srem(key, value, callback);
}
module.isSetMember = function(key, value, callback) {
redisClient.sismember(key, value, callback);
}
module.getSetMembers = function(key, callback) {
redisClient.smembers(key, callback);
}
2013-12-02 15:46:25 -05:00
2013-12-02 17:10:26 -05:00
module.sortedSetAdd = function(key, score, value, callback) {
redisClient.zadd(key, score, value, callback);
}
module.sortedSetRemove = function(key, value, callback) {
redisClient.zrem(key, value, callback);
}
2013-12-02 15:46:25 -05:00
}(exports));
2013-12-02 15:45:15 -05:00