diff --git a/api/src/engines/index.js b/api/src/engines/index.js new file mode 100644 index 000000000..8e6ffe170 --- /dev/null +++ b/api/src/engines/index.js @@ -0,0 +1,4 @@ +module.exports = connection => { + const { engine } = connection; + return require(`./${engine}`); +}; diff --git a/api/src/proc/connectProcess.js b/api/src/proc/connectProcess.js index c46799a78..3b803e85e 100644 --- a/api/src/proc/connectProcess.js +++ b/api/src/proc/connectProcess.js @@ -1,6 +1,8 @@ +const engines = require('../engines'); + process.on('message', async connection => { try { - const driver = require(`../engines/${connection.engine}/index`); + const driver = engines(connection); const conn = await driver.connect(connection); const res = await driver.getVersion(conn); process.send(res); diff --git a/api/src/proc/serverConnectionProcess.js b/api/src/proc/serverConnectionProcess.js index 567d078b0..0250d43da 100644 --- a/api/src/proc/serverConnectionProcess.js +++ b/api/src/proc/serverConnectionProcess.js @@ -1,8 +1,10 @@ +const engines = require('../engines'); + let systemConnection; let storedConnection; async function handleRefreshDatabases() { - const driver = require(`../engines/${storedConnection.engine}/index`); + const driver = engines(storedConnection); const databases = await driver.listDatabases(systemConnection); process.send({ msgtype: 'databases', databases }); }