mirrors/dbgate
1
0
Fork 0
mirror of https://github.com/dbgate/dbgate synced 2024-11-22 16:27:18 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

backend - using engine driver from plugin

Browse Source
This commit is contained in:
Jan Prochazka 2020-11-24 20:42:02 +01:00
parent 424aff5d93
commit c96cb08cfd
9 changed files with 58 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
packages/api/src/proc/connectProcess.js
View File

@ -1,13 +1,12 @@
const engines = require('dbgate-engines');
const driverConnect = require('../utility/driverConnect');
const childProcessChecker = require('../utility/childProcessChecker');
const requireEngineDriver = require('../utility/requireEngineDriver');
function start() {
childProcessChecker();
process.on('message', async (connection) => {
try {
const driver = engines(connection);
const conn = await driverConnect(driver, connection);
const driver = requireEngineDriver(connection);
const conn = await driver.connect(connection);
const res = await driver.getVersion(conn);
process.send({ msgtype: 'connected', ...res });
} catch (e) {

13
packages/api/src/proc/databaseConnectionProcess.js
View File

@ -1,7 +1,6 @@
const engines = require('dbgate-engines');
const stableStringify = require('json-stable-stringify');
const driverConnect = require('../utility/driverConnect');
const childProcessChecker = require('../utility/childProcessChecker');
const requireEngineDriver = require('../utility/requireEngineDriver');
let systemConnection;
let storedConnection;
@ -26,14 +25,14 @@ async function checkedAsyncCall(promise) {
}
async function handleFullRefresh() {
const driver = engines(storedConnection);
const driver = requireEngineDriver(storedConnection);
analysedStructure = await checkedAsyncCall(driver.analyseFull(systemConnection));
process.send({ msgtype: 'structure', structure: analysedStructure });
setStatusName('ok');
}
async function handleIncrementalRefresh() {
const driver = engines(storedConnection);
const driver = requireEngineDriver(storedConnection);
const newStructure = await checkedAsyncCall(driver.analyseIncremental(systemConnection, analysedStructure));
if (newStructure != null) {
analysedStructure = newStructure;
@ -58,8 +57,8 @@ async function handleConnect({ connection, structure }) {
lastPing = new Date().getTime();
if (!structure) setStatusName('pending');
const driver = engines(storedConnection);
systemConnection = await checkedAsyncCall(driverConnect(driver, storedConnection));
const driver = requireEngineDriver(storedConnection);
systemConnection = await checkedAsyncCall(driver.connect(storedConnection));
if (structure) {
analysedStructure = structure;
handleIncrementalRefresh();
@ -82,7 +81,7 @@ function waitConnected() {
async function handleQueryData({ msgid, sql }) {
await waitConnected();
const driver = engines(storedConnection);
const driver = requireEngineDriver(storedConnection);
try {
const res = await driver.query(systemConnection, sql);
process.send({ msgtype: 'response', msgid, ...res });

13
packages/api/src/proc/serverConnectionProcess.js
View File

@ -1,7 +1,6 @@
const engines = require('dbgate-engines');
const stableStringify = require('json-stable-stringify');
const driverConnect = require('../utility/driverConnect');
const childProcessChecker = require('../utility/childProcessChecker');
const requireEngineDriver = require('../utility/requireEngineDriver');
let systemConnection;
let storedConnection;
@ -10,7 +9,7 @@ let lastStatus = null;
let lastPing = null;
async function handleRefresh() {
const driver = engines(storedConnection);
const driver = requireEngineDriver(storedConnection);
try {
const databases = await driver.listDatabases(systemConnection);
setStatusName('ok');
@ -46,9 +45,9 @@ async function handleConnect(connection) {
setStatusName('pending');
lastPing = new Date().getTime();
const driver = engines(storedConnection);
const driver = requireEngineDriver(storedConnection);
try {
systemConnection = await driverConnect(driver, storedConnection);
systemConnection = await driver.connect(storedConnection);
handleRefresh();
setInterval(handleRefresh, 30 * 1000);
} catch (err) {
@ -66,8 +65,8 @@ function handlePing() {
}
async function handleCreateDatabase({ name }) {
const driver = engines(storedConnection);
systemConnection = await driverConnect(driver, storedConnection);
const driver = requireEngineDriver(storedConnection);
systemConnection = await driver.connect(storedConnection);
console.log(`RUNNING SCRIPT: CREATE DATABASE ${driver.dialect.quoteIdentifier(name)}`);
await driver.query(systemConnection, `CREATE DATABASE ${driver.dialect.quoteIdentifier(name)}`);
await handleRefresh();

9
packages/api/src/proc/sessionProcess.js
View File

@ -1,4 +1,3 @@
const engines = require('dbgate-engines');
const uuidv1 = require('uuid/v1');
const path = require('path');
const fs = require('fs');
@ -6,8 +5,8 @@ const _ = require('lodash');
const childProcessChecker = require('../utility/childProcessChecker');
const goSplit = require('../utility/goSplit');
const driverConnect = require('../utility/driverConnect');
const { jsldir } = require('../utility/directories');
const requireEngineDriver = require('../utility/requireEngineDriver');
let systemConnection;
let storedConnection;
@ -119,8 +118,8 @@ class StreamHandler {
async function handleConnect(connection) {
storedConnection = connection;
const driver = engines(storedConnection);
systemConnection = await driverConnect(driver, storedConnection);
const driver = requireEngineDriver(storedConnection);
systemConnection = await driver.connect(storedConnection);
for (const [resolve] of afterConnectCallbacks) {
resolve();
}
@ -142,7 +141,7 @@ function waitConnected() {
async function handleExecuteQuery({ sql }) {
await waitConnected();
const driver = engines(storedConnection);
const driver = requireEngineDriver(storedConnection);
let resultIndex = 0;
for (const sqlItem of goSplit(sql)) {

8
packages/api/src/shell/queryReader.js
View File

@ -1,12 +1,10 @@
const driverConnect = require('../utility/driverConnect');
const engines = require('dbgate-engines');
const requireEngineDriver = require("../utility/requireEngineDriver");
async function queryReader({ connection, sql }) {
console.log(`Reading query ${sql}`);
const driver = engines(connection);
const pool = await driverConnect(driver, connection);
const driver = requireEngineDriver(connection);
const pool = await driver.connect(connection);
console.log(`Connected.`);
return await driver.readQuery(pool, sql);
}

10
packages/api/src/shell/tableReader.js
View File

@ -1,11 +1,9 @@
const { quoteFullName } = require('dbgate-tools');
const driverConnect = require('../utility/driverConnect');
const engines = require('dbgate-engines');
const requireEngineDriver = require('../utility/requireEngineDriver');
async function tableReader({ connection, pureName, schemaName }) {
const driver = engines(connection);
const pool = await driverConnect(driver, connection);
const driver = requireEngineDriver(connection);
const pool = await driver.connect(connection);
console.log(`Connected.`);
const fullName = { pureName, schemaName };
@ -14,11 +12,13 @@ async function tableReader({ connection, pureName, schemaName }) {
const query = `select * from ${quoteFullName(driver.dialect, fullName)}`;
if (table) {
console.log(`Reading table ${table.pureName}`);
// @ts-ignore
return await driver.readQuery(pool, query, table);
}
const view = await driver.analyseSingleObject(pool, fullName, 'views');
if (view) {
console.log(`Reading view ${view.pureName}`);
// @ts-ignore
return await driver.readQuery(pool, query, view);
}

8
packages/api/src/shell/tableWriter.js
View File

@ -1,12 +1,10 @@
const driverConnect = require('../utility/driverConnect');
const engines = require('dbgate-engines');
const requireEngineDriver = require("../utility/requireEngineDriver");
async function tableWriter({ connection, schemaName, pureName, ...options }) {
console.log(`Write table ${schemaName}.${pureName}`);
const driver = engines(connection);
const pool = await driverConnect(driver, connection);
const driver = requireEngineDriver(connection);
const pool = await driver.connect(connection);
console.log(`Connected.`);
return await driver.writeTable(pool, { schemaName, pureName }, options);
}

24
packages/api/src/utility/requireEngineDriver.js Normal file
View File

@ -0,0 +1,24 @@
const _ = require('lodash');
const requirePlugin = require('../shell/requirePlugin');
/** @returns {import('dbgate-types').EngineDriver} */
function requireEngineDriver(connection) {
let engine = null;
if (_.isString(connection)) {
engine = connection;
} else if (_.isPlainObject(connection)) {
engine = connection.engine;
}
if (!engine) {
throw new Error('Could not get driver from connection');
}
if (engine.includes('@')) {
const [shortName, packageName] = engine.split('@');
const plugin = requirePlugin(packageName);
return plugin.driver;
}
throw new Error(`Could not found engine driver ${engine}`);
}
module.exports = requireEngineDriver;

8
packages/types/engines.d.ts vendored
View File

@ -7,9 +7,9 @@ import { DatabaseInfo, NamedObjectInfo, TableInfo, ViewInfo, ProcedureInfo, Func
export interface StreamOptions {
recordset: (columns) => void;
row: (row) => void;
error: (error) => void;
done: (result) => void;
info: (info) => void;
error?: (error) => void;
done?: (result) => void;
info?: (info) => void;
}
export interface WriteTableOptions {
@ -21,7 +21,7 @@ export interface WriteTableOptions {
export interface EngineDriver {
engine: string;
title: string;
connect(nativeModules, { server, port, user, password, database }): any;
connect({ server, port, user, password, database }): any;
query(pool: any, sql: string): Promise<QueryResult>;
stream(pool: any, sql: string, options: StreamOptions);
readQuery(pool: any, sql: string, structure?: TableInfo): Promise<stream.Readable>;