mirror of
https://github.com/HabitRPG/habitica.git
synced 2025-12-16 06:07:21 +01:00
chore: Add migration helper functions
This commit is contained in:
35
migrations/utils/connect.js
Normal file
35
migrations/utils/connect.js
Normal file
@@ -0,0 +1,35 @@
|
||||
'use strict';
|
||||
|
||||
const MongoClient = require('mongodb').MongoClient;
|
||||
const logger = require('./logger');
|
||||
|
||||
let db;
|
||||
|
||||
function connectToDb (dbUri) {
|
||||
return new Promise((resolve, reject) => {
|
||||
MongoClient.connect(dbUri, (err, database) => {
|
||||
if (err) {
|
||||
logger.error(`Uh oh... Problem connecting to the database at ${dbUri}`);
|
||||
return reject(err);
|
||||
}
|
||||
|
||||
db = database;
|
||||
|
||||
logger.success(`Connected to ${dbUri}`);
|
||||
|
||||
resolve(database);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
function closeDb () {
|
||||
if (db) db.close();
|
||||
|
||||
logger.success('CLosed connection to the database');
|
||||
return Promise.resolve();
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
connectToDb,
|
||||
closeDb,
|
||||
}
|
||||
Reference in New Issue
Block a user