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

#451 loading fks on postgres cleanup & fix

Browse Source
This commit is contained in:
Jan Prochazka
2023-01-06 18:40:47 +01:00
parent 37eeaf0cce
commit b06d747399
2 changed files with 53 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
packages/tools/src/DatabaseAnalyser.ts
View File

@@ -170,9 +170,9 @@ export class DatabaseAnalyser {
// return this.structure.tables.find((x) => x.objectId == id); // return this.structure.tables.find((x) => x.objectId == id);
// } // }
containsObjectIdCondition(typeFields) { // containsObjectIdCondition(typeFields) {
return this.createQueryCore('=OBJECT_ID_CONDITION', typeFields) != ' is not null'; // return this.createQueryCore('=OBJECT_ID_CONDITION', typeFields) != ' is not null';
} // }
createQuery(template, typeFields) { createQuery(template, typeFields) {
return this.createQueryCore(template, typeFields); return this.createQueryCore(template, typeFields);

103
plugins/dbgate-plugin-postgres/src/backend/Analyser.js
View File

@@ -57,8 +57,8 @@ class Analyser extends DatabaseAnalyser {
createQuery(resFileName, typeFields) { createQuery(resFileName, typeFields) {
const query = super.createQuery(sql[resFileName], typeFields); const query = super.createQuery(sql[resFileName], typeFields);
if (query) return query.replace('#REFTABLECOND#', this.driver.__analyserInternals.refTableCond); // if (query) return query.replace('#REFTABLECOND#', this.driver.__analyserInternals.refTableCond);
return null; return query;
} }
async _computeSingleObjectId() { async _computeSingleObjectId() {
@@ -80,63 +80,60 @@ class Analyser extends DatabaseAnalyser {
let fkColumns = null; let fkColumns = null;
// if (true) { // if (true) {
if (this.containsObjectIdCondition(['tables']) || this.driver.__analyserInternals.refTableCond) { // if (this.containsObjectIdCondition(['tables']) || this.driver.__analyserInternals.refTableCond) {
this.feedback({ analysingMessage: 'Loading foreign keys' }); // this.feedback({ analysingMessage: 'Loading foreign keys' });
fkColumns = await this.driver.query(this.pool, this.createQuery('foreignKeys', ['tables'])); // fkColumns = await this.driver.query(this.pool, this.createQuery('foreignKeys', ['tables']));
} else { // } else {
this.feedback({ analysingMessage: 'Loading foreign key constraints' }); this.feedback({ analysingMessage: 'Loading foreign key constraints' });
const fk_tableConstraints = await this.driver.query( const fk_tableConstraints = await this.driver.query(this.pool, this.createQuery('fk_tableConstraints', ['tables']));
this.pool,
this.createQuery('fk_tableConstraints', ['tables']) this.feedback({ analysingMessage: 'Loading foreign key refs' });
const fk_referentialConstraints = await this.driver.query(
this.pool,
this.createQuery('fk_referentialConstraints', ['tables'])
);
this.feedback({ analysingMessage: 'Loading foreign key columns' });
const fk_keyColumnUsage = await this.driver.query(this.pool, this.createQuery('fk_keyColumnUsage', ['tables']));
const cntKey = x => `${x.constraint_name}|${x.constraint_schema}`;
const fkRows = [];
const fkConstraintDct = _.keyBy(fk_tableConstraints.rows, cntKey);
for (const fkRef of fk_referentialConstraints.rows) {
const cntBase = fkConstraintDct[cntKey(fkRef)];
const cntRef = fkConstraintDct[`${fkRef.unique_constraint_name}|${fkRef.unique_constraint_schema}`];
if (!cntBase || !cntRef) continue;
const baseCols = _.sortBy(
fk_keyColumnUsage.rows.filter(
x => x.table_name == cntBase.table_name && x.constraint_name == cntBase.constraint_name
),
'ordinal_position'
); );
const refCols = _.sortBy(
this.feedback({ analysingMessage: 'Loading foreign key refs' }); fk_keyColumnUsage.rows.filter(
const fk_referentialConstraints = await this.driver.query( x => x.table_name == cntRef.table_name && x.constraint_name == cntRef.constraint_name
this.pool, ),
this.createQuery('fk_referentialConstraints', ['tables']) 'ordinal_position'
); );
if (baseCols.length != refCols.length) continue;
this.feedback({ analysingMessage: 'Loading foreign key columns' }); for (let i = 0; i < baseCols.length; i++) {
const fk_keyColumnUsage = await this.driver.query(this.pool, this.createQuery('fk_keyColumnUsage', ['tables'])); const baseCol = baseCols[i];
const refCol = refCols[i];
const cntKey = x => `${x.constraint_name}|${x.constraint_schema}`; fkRows.push({
const rows = []; ...fkRef,
const constraintDct = _.keyBy(fk_tableConstraints.rows, cntKey); pure_name: cntBase.table_name,
for (const fkRef of fk_referentialConstraints.rows) { schema_name: cntBase.table_schema,
const cntBase = constraintDct[cntKey(fkRef)]; ref_table_name: cntRef.table_name,
const cntRef = constraintDct[`${fkRef.unique_constraint_name}|${fkRef.unique_constraint_schema}`]; ref_schema_name: cntRef.table_schema,
if (!cntBase || !cntRef) continue; column_name: baseCol.column_name,
const baseCols = _.sortBy( ref_column_name: refCol.column_name,
fk_keyColumnUsage.rows.filter( });
x => x.table_name == cntBase.table_name && x.constraint_name == cntBase.constraint_name
),
'ordinal_position'
);
const refCols = _.sortBy(
fk_keyColumnUsage.rows.filter(
x => x.table_name == cntRef.table_name && x.constraint_name == cntRef.constraint_name
),
'ordinal_position'
);
if (baseCols.length != refCols.length) continue;
for (let i = 0; i < baseCols.length; i++) {
const baseCol = baseCols[i];
const refCol = refCols[i];
rows.push({
...fkRef,
pure_name: cntBase.table_name,
schema_name: cntBase.table_schema,
ref_table_name: cntRef.table_name,
ref_schema_name: cntRef.table_schema,
column_name: baseCol.column_name,
ref_column_name: refCol.column_name,
});
}
} }
fkColumns = { rows };
} }
fkColumns = { rows: fkRows };
// }
this.feedback({ analysingMessage: 'Loading views' }); this.feedback({ analysingMessage: 'Loading views' });
const views = await this.driver.query(this.pool, this.createQuery('views', ['views'])); const views = await this.driver.query(this.pool, this.createQuery('views', ['views']));