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

mysql - analyse primary and foreign keys

Browse Source
This commit is contained in:
Jan Prochazka
2020-05-16 08:45:18 +02:00
parent 70ab03ad73
commit 83e3f8b1d8
7 changed files with 81 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -2,37 +2,7 @@ const fp = require('lodash/fp');
const _ = require('lodash');
const sql = require('./sql');
const DatabaseAnalayser = require('../default/DatabaseAnalyser');
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')),
};
}
function extractForeignKeys(table, fkColumns) {
const grouped = _.groupBy(fkColumns.filter(byTableFilter(table)), 'constraintName');
return _.keys(grouped).map((constraintName) => ({
constraintName,
constraintType: 'foreignKey',
..._.pick(grouped[constraintName][0], [
'constraintName',
'schemaName',
'pureName',
'refSchemaName',
'refTableName',
'updateAction',
'deleteAction',
]),
columns: grouped[constraintName].map(fp.pick(['columnName', 'refColumnName'])),
}));
}
const DatabaseAnalyser = require('../default/DatabaseAnalyser');
function objectTypeToField(type) {
switch (type.trim()) {
@@ -186,7 +156,7 @@ function detectType(col) {
};
}
class MsSqlAnalyser extends DatabaseAnalayser {
class MsSqlAnalyser extends DatabaseAnalyser {
constructor(pool, driver) {
super(pool, driver);
}
@@ -239,8 +209,8 @@ class MsSqlAnalyser extends DatabaseAnalayser {
autoIncrement: !!isIdentity,
commonType: detectType(col),
})),
primaryKey: extractPrimaryKeys(row, pkColumnsRows.rows),
foreignKeys: extractForeignKeys(row, fkColumnsRows.rows),
primaryKey: DatabaseAnalyser.extractPrimaryKeys(row, pkColumnsRows.rows),
foreignKeys: DatabaseAnalyser.extractForeignKeys(row, fkColumnsRows.rows),
}));
const views = viewsRows.rows.map((row) => ({