89 lines
2.5 KiB
JavaScript
89 lines
2.5 KiB
JavaScript
|
|
||
|
|
|
||
|
|
var MongoClient = require('mongodb').MongoClient,
|
||
|
|
winston = require('winston');
|
||
|
|
|
||
|
|
var MongoDocumentStore = function (options) {
|
||
|
|
this.expire = options.expire;
|
||
|
|
this.connectionUrl = process.env.DATABASE_URl || options.connectionUrl;
|
||
|
|
};
|
||
|
|
|
||
|
|
MongoDocumentStore.prototype.set = function (key, data, callback, skipExpire) {
|
||
|
|
var now = Math.floor(new Date().getTime() / 1000),
|
||
|
|
that = this;
|
||
|
|
|
||
|
|
this.safeConnect(function (err, db) {
|
||
|
|
if (err)
|
||
|
|
return callback(false);
|
||
|
|
|
||
|
|
db.collection('entries').update({
|
||
|
|
'entry_id': key,
|
||
|
|
$or: [
|
||
|
|
{ expiration: -1 },
|
||
|
|
{ expiration: { $gt: now } }
|
||
|
|
]
|
||
|
|
}, {
|
||
|
|
'entry_id': key,
|
||
|
|
'value': data,
|
||
|
|
'expiration': that.expire && !skipExpire ? that.expire + now : -1
|
||
|
|
}, {
|
||
|
|
upsert: true
|
||
|
|
}, function (err, existing) {
|
||
|
|
if (err) {
|
||
|
|
winston.error('error persisting value to mongodb', { error: err });
|
||
|
|
return callback(false);
|
||
|
|
}
|
||
|
|
|
||
|
|
callback(true);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
MongoDocumentStore.prototype.get = function (key, callback, skipExpire) {
|
||
|
|
var now = Math.floor(new Date().getTime() / 1000),
|
||
|
|
that = this;
|
||
|
|
|
||
|
|
this.safeConnect(function (err, db) {
|
||
|
|
if (err)
|
||
|
|
return callback(false);
|
||
|
|
|
||
|
|
db.collection('entries').findOne({
|
||
|
|
'entry_id': key,
|
||
|
|
$or: [
|
||
|
|
{ expiration: -1 },
|
||
|
|
{ expiration: { $gt: now } }
|
||
|
|
]
|
||
|
|
}, function (err, entry) {
|
||
|
|
if (err) {
|
||
|
|
winston.error('error persisting value to mongodb', { error: err });
|
||
|
|
return callback(false);
|
||
|
|
}
|
||
|
|
|
||
|
|
callback(entry === null ? false : entry.value);
|
||
|
|
|
||
|
|
if (entry !== null && entry.expiration !== -1 && that.expire && !skipExpire) {
|
||
|
|
db.collection('entries').update({
|
||
|
|
'entry_id': key
|
||
|
|
}, {
|
||
|
|
$set: {
|
||
|
|
'expiration': that.expire + now
|
||
|
|
}
|
||
|
|
}, function (err, result) { });
|
||
|
|
}
|
||
|
|
});
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
MongoDocumentStore.prototype.safeConnect = function (callback) {
|
||
|
|
MongoClient.connect(this.connectionUrl, function (err, db) {
|
||
|
|
if (err) {
|
||
|
|
winston.error('error connecting to mongodb', { error: err });
|
||
|
|
callback(err);
|
||
|
|
} else {
|
||
|
|
callback(undefined, db);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
module.exports = MongoDocumentStore;
|