mirror of
https://github.com/ether/etherpad-lite.git
synced 2025-04-25 09:56:15 -04:00
86 lines
2.6 KiB
JavaScript
86 lines
2.6 KiB
JavaScript
'use strict';
|
|
|
|
/**
|
|
* The DB Module provides a database initialized with the settings
|
|
* provided by the settings module
|
|
*/
|
|
|
|
/*
|
|
* 2011 Peter 'Pita' Martischka (Primary Technology Ltd)
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS-IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
const ueberDB = require('ueberdb2');
|
|
const settings = require('../utils/Settings');
|
|
const log4js = require('log4js');
|
|
const stats = require('../stats');
|
|
const util = require('util');
|
|
|
|
// set database settings
|
|
const db =
|
|
new ueberDB.Database(settings.dbType, settings.dbSettings, null, log4js.getLogger('ueberDB'));
|
|
|
|
/**
|
|
* The UeberDB Object that provides the database functions
|
|
*/
|
|
exports.db = null;
|
|
|
|
/**
|
|
* Initializes the database with the settings provided by the settings module
|
|
* @param {Function} callback
|
|
*/
|
|
exports.init = async () => await new Promise((resolve, reject) => {
|
|
db.init((err) => {
|
|
if (err) {
|
|
// there was an error while initializing the database, output it and stop
|
|
console.error('ERROR: Problem while initalizing the database');
|
|
console.error(err.stack ? err.stack : err);
|
|
process.exit(1);
|
|
}
|
|
|
|
if (db.metrics != null) {
|
|
for (const [metric, value] of Object.entries(db.metrics)) {
|
|
if (typeof value !== 'number') continue;
|
|
stats.gauge(`ueberdb_${metric}`, () => db.metrics[metric]);
|
|
}
|
|
}
|
|
|
|
// everything ok, set up Promise-based methods
|
|
['get', 'set', 'findKeys', 'getSub', 'setSub', 'remove'].forEach((fn) => {
|
|
exports[fn] = util.promisify(db[fn].bind(db));
|
|
});
|
|
|
|
// set up wrappers for get and getSub that can't return "undefined"
|
|
const get = exports.get;
|
|
exports.get = async (key) => {
|
|
const result = await get(key);
|
|
return (result === undefined) ? null : result;
|
|
};
|
|
|
|
const getSub = exports.getSub;
|
|
exports.getSub = async (key, sub) => {
|
|
const result = await getSub(key, sub);
|
|
return (result === undefined) ? null : result;
|
|
};
|
|
|
|
// exposed for those callers that need the underlying raw API
|
|
exports.db = db;
|
|
resolve();
|
|
});
|
|
});
|
|
|
|
exports.shutdown = async (hookName, context) => {
|
|
await util.promisify(db.close.bind(db))();
|
|
console.log('Database closed');
|
|
};
|