dennii/dbgate
1
0
Fork 0
mirror of https://github.com/DeNNiiInc/dbgate.git synced 2026-04-18 02:06:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

create script callable from react

Browse Source
This commit is contained in:
Jan Prochazka
2020-02-03 20:34:38 +01:00
parent 680bed549f
commit 8a85cfe687
14 changed files with 147 additions and 95 deletions
Download Patch File Download Diff File Expand all files Collapse all files

77
packages/engines/mssql/MsSqlAnalyser.js
View File

@@ -1,42 +1,49 @@
const fs = require('fs-extra');
const fp = require('lodash/fp');
const path = require('path');
const _ = require('lodash');
const fp = require("lodash/fp");
const _ = require("lodash");
const DatabaseAnalayser = require('../default/DatabaseAnalyser');
const DatabaseAnalayser = require("../default/DatabaseAnalyser");
/** @returns {Promise<string>} */
async function loadQuery(name) {
return await fs.readFile(path.join(__dirname, name), 'utf-8');
async function loadQuery(pool, name) {
return await pool._nativeModules.fs.readFile(
pool._nativeModules.path.join(__dirname, name),
"utf-8"
);
}
const byTableFilter = table => x => x.pureName == table.pureName && x.schemaName == x.schemaName;
const byTableFilter = table => x =>
x.pureName == table.pureName && x.schemaName == x.schemaName;
function extractPrimaryKeys(table, pkColumns) {
const filtered = pkColumns.filter(byTableFilter(table));
if (filtered.length == 0) return undefined;
return {
..._.pick(filtered[0], ['constraintName', 'schemaName', 'pureName']),
constraintType: 'primaryKey',
columns: filtered.map(fp.pick('columnName')),
..._.pick(filtered[0], ["constraintName", "schemaName", "pureName"]),
constraintType: "primaryKey",
columns: filtered.map(fp.pick("columnName"))
};
}
function extractForeignKeys(table, fkColumns) {
const grouped = _.groupBy(fkColumns.filter(byTableFilter(table)), 'constraintName');
const grouped = _.groupBy(
fkColumns.filter(byTableFilter(table)),
"constraintName"
);
return _.keys(grouped).map(constraintName => ({
constraintName,
constraintType: 'foreignKey',
constraintType: "foreignKey",
..._.pick(grouped[constraintName][0], [
'constraintName',
'schemaName',
'pureName',
'refSchemaName',
'refTableName',
'updateAction',
'deleteAction',
"constraintName",
"schemaName",
"pureName",
"refSchemaName",
"refTableName",
"updateAction",
"deleteAction"
]),
columns: grouped[constraintName].map(fp.pick(['columnName', 'refColumnName'])),
columns: grouped[constraintName].map(
fp.pick(["columnName", "refColumnName"])
)
}));
}
@@ -53,15 +60,27 @@ class MsSqlAnalyser extends DatabaseAnalayser {
functions = false,
triggers = false
) {
let res = await loadQuery(resFileName);
res = res.replace('=[OBJECT_ID_CONDITION]', ' is not null');
let res = await loadQuery(this.pool, resFileName);
res = res.replace("=[OBJECT_ID_CONDITION]", " is not null");
return res;
}
async runAnalysis() {
const tables = await this.driver.query(this.pool, await this.createQuery('tables.sql'));
const columns = await this.driver.query(this.pool, await this.createQuery('columns.sql'));
const pkColumns = await this.driver.query(this.pool, await this.createQuery('primary_keys.sql'));
const fkColumns = await this.driver.query(this.pool, await this.createQuery('foreign_keys.sql'));
const tables = await this.driver.query(
this.pool,
await this.createQuery("tables.sql")
);
const columns = await this.driver.query(
this.pool,
await this.createQuery("columns.sql")
);
const pkColumns = await this.driver.query(
this.pool,
await this.createQuery("primary_keys.sql")
);
const fkColumns = await this.driver.query(
this.pool,
await this.createQuery("foreign_keys.sql")
);
this.result.tables = tables.rows.map(table => ({
...table,
@@ -70,10 +89,10 @@ class MsSqlAnalyser extends DatabaseAnalayser {
.map(({ isNullable, isIdentity, ...col }) => ({
...col,
notNull: !isNullable,
autoIncrement: !!isIdentity,
autoIncrement: !!isIdentity
})),
primaryKey: extractPrimaryKeys(table, pkColumns.rows),
foreignKeys: extractForeignKeys(table, fkColumns.rows),
foreignKeys: extractForeignKeys(table, fkColumns.rows)
}));
}
}