mirrors/haste-server
1
0
Fork 0
mirror of https://github.com/seejohnrun/haste-server.git synced 2024-11-01 11:31:22 +00:00
Code Issues Releases Wiki Activity

Changes to support MongoDB Node.JS Driver v3

Browse source
This commit is contained in:
Andrew Molchanov 2020-11-27 02:23:47 +03:00
parent 5d2965ffc5
commit ddabb1187f
No known key found for this signature in database
GPG key ID: 7C75AA362C5C27FC
1 changed files with 10 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
lib/document_stores/mongo.js → lib/document_stores/mongodb.js
View file

@ -1,21 +1,20 @@
const MongoClient = require('mongodb').MongoClient,
var MongoClient = require('mongodb').MongoClient,
winston = require('winston'); winston = require('winston');
var MongoDocumentStore = function (options) { const MongoDocumentStore = function (options) {
this.expire = options.expire; this.expire = options.expire;
this.connectionUrl = process.env.DATABASE_URl || options.connectionUrl; this.connectionUrl = process.env.DATABASE_URL || options.connectionUrl;
this.connectionName = process.env.DATABASE_NAME || options.connectionName;
}; };
MongoDocumentStore.prototype.set = function (key, data, callback, skipExpire) { MongoDocumentStore.prototype.set = function (key, data, callback, skipExpire) {
var now = Math.floor(new Date().getTime() / 1000), const now = Math.floor(new Date().getTime() / 1000),
that = this; that = this;
this.safeConnect(function (err, db) { this.safeConnect(function (err, db) {
if (err) if (err)
return callback(false); return callback(false);
db.collection('entries').update({ db.collection('entries').update({
'entry_id': key, 'entry_id': key,
$or: [ $or: [
@ -40,13 +39,13 @@ MongoDocumentStore.prototype.set = function (key, data, callback, skipExpire) {
}; };
MongoDocumentStore.prototype.get = function (key, callback, skipExpire) { MongoDocumentStore.prototype.get = function (key, callback, skipExpire) {
var now = Math.floor(new Date().getTime() / 1000), const now = Math.floor(new Date().getTime() / 1000),
that = this; that = this;
this.safeConnect(function (err, db) { this.safeConnect(function (err, db) {
if (err) if (err)
return callback(false); return callback(false);
db.collection('entries').findOne({ db.collection('entries').findOne({
'entry_id': key, 'entry_id': key,
$or: [ $or: [
@ -75,12 +74,12 @@ MongoDocumentStore.prototype.get = function (key, callback, skipExpire) {
}; };
MongoDocumentStore.prototype.safeConnect = function (callback) { MongoDocumentStore.prototype.safeConnect = function (callback) {
MongoClient.connect(this.connectionUrl, function (err, db) { MongoClient.connect(this.connectionUrl, function (err, client) {
if (err) { if (err) {
winston.error('error connecting to mongodb', { error: err }); winston.error('error connecting to mongodb', { error: err });
callback(err); callback(err);
} else { } else {
callback(undefined, db); callback(undefined, client.db(this.connectionDBName));
} }
}); });
}; };