30 lines
718 B
JavaScript
30 lines
718 B
JavaScript
|
// service wrapper for the database drivers in ../database/
|
||
|
|
const { BaseServiceWorker } = require("eris-fleet");
|
||
|
|
|
||
|
|
const database = require(`../database/${process.env.DB ? process.env.DB.split("://")[0] : "dummy"}.js`);
|
||
|
|
|
||
|
|
class DatabaseWorker extends BaseServiceWorker {
|
||
|
|
constructor(setup) {
|
||
|
|
super(setup);
|
||
|
|
this.serviceReady();
|
||
|
|
}
|
||
|
|
|
||
|
|
async handleCommand(data) {
|
||
|
|
try {
|
||
|
|
if (database[data.name]) {
|
||
|
|
return await database[data.name](...data.args);
|
||
|
|
} else {
|
||
|
|
throw "Unknown query";
|
||
|
|
}
|
||
|
|
} catch (err) {
|
||
|
|
return { err: typeof err === "string" ? err : err.message };
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
shutdown(done) {
|
||
|
|
database.stop().then(() => done);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
module.exports = DatabaseWorker;
|