1
0
Fork 0
mirror of https://github.com/seejohnrun/haste-server.git synced 2024-11-01 19:41:21 +00:00
haste-server/lib/document_stores/mongodb.js

88 lines
2.6 KiB
JavaScript
Raw Normal View History

const MongoClient = require('mongodb').MongoClient,
2016-06-10 20:43:43 +00:00
winston = require('winston');
const MongoDocumentStore = function (options) {
2016-06-10 20:43:43 +00:00
this.expire = options.expire;
this.connectionUrl = process.env.DATABASE_URL || options.connectionUrl;
this.connectionName = process.env.DATABASE_NAME || options.connectionName;
2016-06-10 20:43:43 +00:00
};
MongoDocumentStore.prototype.set = function (key, data, callback, skipExpire) {
const now = Math.floor(new Date().getTime() / 1000),
2016-06-10 20:43:43 +00:00
that = this;
this.safeConnect(function (err, db) {
if (err)
return callback(false);
2016-06-10 20:43:43 +00:00
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) {
const now = Math.floor(new Date().getTime() / 1000),
2016-06-10 20:43:43 +00:00
that = this;
this.safeConnect(function (err, db) {
if (err)
return callback(false);
2016-06-10 20:43:43 +00:00
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, client) {
2016-06-10 20:43:43 +00:00
if (err) {
winston.error('error connecting to mongodb', { error: err });
callback(err);
} else {
callback(undefined, client.db(this.connectionDBName));
2016-06-10 20:43:43 +00:00
}
});
};
module.exports = MongoDocumentStore;