haste-server/lib/document_stores/redis.js

81 lines
2.2 KiB
JavaScript
Raw Normal View History

2011-11-19 01:52:18 +01:00
var redis = require('redis');
var winston = require('winston');
2011-11-22 04:03:50 +01:00
// For storing in redis
// options[type] = redis
// options[host] - The host to connect to (default localhost)
// options[port] - The port to connect to (default 5379)
// options[db] - The db to use (default 0)
// options[expire] - The time to live for each key set (default never)
2012-09-27 17:46:53 +02:00
var RedisDocumentStore = function(options, client) {
2011-11-22 04:03:50 +01:00
this.expire = options.expire;
2012-09-27 17:46:53 +02:00
if (client) {
2012-09-27 17:47:23 +02:00
winston.info('using predefined redis client');
2012-09-27 17:46:53 +02:00
RedisDocumentStore.client = client;
} else if (!RedisDocumentStore.client) {
2012-09-27 17:50:56 +02:00
winston.info('configuring redis');
2011-11-19 01:59:41 +01:00
RedisDocumentStore.connect(options);
2011-11-19 01:52:18 +01:00
}
};
2011-11-19 01:59:41 +01:00
// Create a connection according to config
RedisDocumentStore.connect = function(options) {
var host = options.host || '127.0.0.1';
var port = options.port || 6379;
var index = options.db || 0;
RedisDocumentStore.client = redis.createClient(port, host);
RedisDocumentStore.client.select(index, function(err, reply) {
if (err) {
winston.error(
'error connecting to redis index ' + index,
{ error: err.message }
);
2011-11-19 01:59:41 +01:00
process.exit(1);
}
else {
winston.info('connected to redis on ' + host + ':' + port + '/' + index);
}
});
};
2011-11-19 01:52:18 +01:00
// Save file in a key
2011-11-22 15:22:37 +01:00
RedisDocumentStore.prototype.set = function(key, data, callback, skipExpire) {
2011-11-22 04:03:50 +01:00
var _this = this;
2011-11-19 01:52:18 +01:00
RedisDocumentStore.client.set(key, data, function(err, reply) {
2011-11-22 04:03:50 +01:00
if (err) {
callback(false);
}
else {
2011-11-22 15:22:37 +01:00
if (!skipExpire) {
_this.setExpiration(key);
}
2011-11-22 04:03:50 +01:00
callback(true);
}
2011-11-19 01:52:18 +01:00
});
};
2011-11-22 04:03:50 +01:00
// Expire a key in expire time if set
RedisDocumentStore.prototype.setExpiration = function(key) {
if (this.expire) {
RedisDocumentStore.client.expire(key, this.expire, function(err, reply) {
2011-11-28 07:13:14 +01:00
if (err) {
2011-11-22 04:03:50 +01:00
winston.error('failed to set expiry on key: ' + key);
}
});
}
};
2011-11-19 01:52:18 +01:00
// Get a file from a key
2011-11-28 07:13:14 +01:00
RedisDocumentStore.prototype.get = function(key, callback, skipExpire) {
var _this = this;
2011-11-19 01:52:18 +01:00
RedisDocumentStore.client.get(key, function(err, reply) {
2011-11-28 07:13:14 +01:00
if (!err && !skipExpire) {
_this.setExpiration(key);
}
2011-11-19 01:52:18 +01:00
callback(err ? false : reply);
});
2011-11-19 01:52:18 +01:00
};
module.exports = RedisDocumentStore;