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

postgres analyser fixed - broken loading FKs in incremental analysis

Browse Source
This commit is contained in:
Jan Prochazka
2025-11-28 10:24:53 +01:00
parent 0913011120
commit 65d13189b3
5 changed files with 104 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -299,6 +299,23 @@ export class DatabaseAnalyser<TClient = any> {
}
}
objectIdConditionApplied(typeFields) {
if (!this.modifications || !typeFields || this.modifications.length == 0) {
return false;
}
if (this.modifications.some(x => typeFields.includes(x.objectTypeField) && x.action == 'all')) {
// do not filter objects
return false;
}
const filterIds = this.modifications
.filter(x => typeFields.includes(x.objectTypeField) && (x.action == 'add' || x.action == 'change'))
.map(x => x.objectId);
if (filterIds.length == 0) {
return false;
}
return true;
}
async getModifications() {
const snapshot = await this._getFastSnapshot();
if (!snapshot) return null;

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

@@ -99,6 +99,10 @@ class Analyser extends DatabaseAnalyser {
this.feedback({ analysingMessage: 'DBGM-00244 Loading foreign key constraints' });
// const fk_tableConstraints = await this.analyserQuery('fk_tableConstraints', ['tables']);
if (this.objectIdConditionApplied(['tables'])) {
this.feedback({ analysingMessage: 'DBGM-00000 Loading foreign key refs' });
fkColumns = await this.analyserQuery('foreignKeyColumns', ['tables']);
} else {
this.feedback({ analysingMessage: 'DBGM-00245 Loading foreign key refs' });
const foreignKeys = await this.analyserQuery('foreignKeys', ['tables']);
@@ -148,6 +152,7 @@ class Analyser extends DatabaseAnalyser {
}
}
fkColumns = { rows: fkRows };
}
this.feedback({ analysingMessage: 'DBGM-00247 Loading views' });
const views = await this.analyserQuery('views', ['views']);

36
plugins/dbgate-plugin-postgres/src/backend/sql/foreignKeyColumns.js Normal file
View File

@@ -0,0 +1,36 @@
module.exports = `
SELECT
tc.table_schema as schema_name,
tc.table_name as pure_name,
tc.constraint_name,
tc_pk.table_schema AS ref_schema_name,
tc_pk.table_name AS ref_table_name,
rc.unique_constraint_name,
kcu.column_name,
kcu.ordinal_position,
ccu.column_name AS ref_column_name
FROM information_schema.table_constraints AS tc
JOIN information_schema.key_column_usage AS kcu
ON tc.constraint_name = kcu.constraint_name
AND tc.table_schema = kcu.table_schema
JOIN information_schema.referential_constraints AS rc
ON rc.constraint_name = tc.constraint_name
AND rc.constraint_schema = tc.table_schema
JOIN information_schema.table_constraints AS tc_pk
ON tc_pk.constraint_name = rc.unique_constraint_name
AND tc_pk.constraint_schema = rc.unique_constraint_schema
JOIN information_schema.key_column_usage AS ccu
ON ccu.constraint_name = rc.unique_constraint_name
AND ccu.constraint_schema = rc.unique_constraint_schema
AND ccu.ordinal_position = kcu.position_in_unique_constraint
WHERE tc.constraint_type = 'FOREIGN KEY'
AND (('tables:' || tc.table_schema || '.' || tc.table_name) =OBJECT_ID_CONDITION AND tc.table_schema =SCHEMA_NAME_CONDITION)
OR
(('tables:' || tc_pk.table_schema || '.' || tc_pk.table_name) =OBJECT_ID_CONDITION AND tc.table_schema =SCHEMA_NAME_CONDITION)
ORDER BY
tc.table_schema,
tc.table_name,
tc.constraint_name,
kcu.ordinal_position
;
`;

2
plugins/dbgate-plugin-postgres/src/backend/sql/index.js
View File

@@ -16,6 +16,7 @@ const geometryColumns = require('./geometryColumns');
const geographyColumns = require('./geographyColumns');
const proceduresParameters = require('./proceduresParameters');
const foreignKeys = require('./foreignKeys');
const foreignKeyColumns = require('./foreignKeyColumns');
const triggers = require('./triggers');
const listDatabases = require('./listDatabases');
const listVariables = require('./listVariables');
@@ -30,6 +31,7 @@ module.exports = {
viewModifications,
primaryKeys,
fk_keyColumnUsage,
foreignKeyColumns,
foreignKeys,
views,
routines,