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

fixed load postgres schema on Azure #906

Browse Source
This commit is contained in:
SPRINX0\prochazka
2024-10-01 16:36:26 +02:00
parent 67f58a8dfe
commit 9d8dd558e2
10 changed files with 24 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -56,7 +56,12 @@ class Analyser extends DatabaseAnalyser {
}
createQuery(resFileName, typeFields, replacements = {}) {
const query = super.createQuery(sql[resFileName], typeFields, replacements);
const query = super.createQuery(sql[resFileName], typeFields, {
...replacements,
$typeAggFunc: this.driver.dialect.stringAgg ? 'string_agg' : 'max',
$typeAggParam: this.driver.dialect.stringAgg ? ", '|'" : '',
$md5Function: this.dialect?.isFipsComplianceOn ? 'LENGTH' : 'MD5',
});
return query;
}
@@ -138,10 +143,7 @@ class Analyser extends DatabaseAnalyser {
: null;
this.feedback({ analysingMessage: 'Loading routines' });
const routines = await this.analyserQuery('routines', ['procedures', 'functions'], {
$typeAggFunc: this.driver.dialect.stringAgg ? 'string_agg' : 'max',
$typeAggParam: this.driver.dialect.stringAgg ? ", '|'" : '',
});
const routines = await this.analyserQuery('routines', ['procedures', 'functions']);
this.feedback({ analysingMessage: 'Loading indexes' });
const indexes = this.driver.__analyserInternals.skipIndexes

8
plugins/dbgate-plugin-postgres/src/backend/drivers.js
View File

@@ -182,6 +182,13 @@ const drivers = driverBases.map(driverBase => ({
const { rows } = await this.query(dbhan, 'SELECT version()');
const { version } = rows[0];
let isFipsComplianceOn = false;
try {
await this.query(dbhan, "SELECT MD5('test')");
} catch (err) {
isFipsComplianceOn = true;
}
const isCockroach = version.toLowerCase().includes('cockroachdb');
const isRedshift = version.toLowerCase().includes('redshift');
const isPostgres = !isCockroach && !isRedshift;
@@ -207,6 +214,7 @@ const drivers = driverBases.map(driverBase => ({
isRedshift,
versionMajor,
versionMinor,
isFipsComplianceOn,
};
},
async readQuery(dbhan, sql, structure) {

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

@@ -2,7 +2,7 @@ module.exports = `
select
matviewname as "pure_name",
schemaname as "schema_name",
md5(definition) as "hash_code"
$md5Function(definition) as "hash_code"
from
pg_catalog.pg_matviews WHERE schemaname NOT LIKE 'pg_%' AND schemaname =SCHEMA_NAME_CONDITION
`;

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

@@ -3,7 +3,7 @@ select
matviewname as "pure_name",
schemaname as "schema_name",
definition as "definition",
md5(definition) as "hash_code"
$md5Function(definition) as "hash_code"
from
pg_catalog.pg_matviews WHERE schemaname NOT LIKE 'pg_%'
and ('matviews:' || schemaname || '.' || matviewname) =OBJECT_ID_CONDITION

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

@@ -2,7 +2,7 @@ module.exports = `
select
routine_name as "pure_name",
routine_schema as "schema_name",
md5(routine_definition) as "hash_code",
$md5Function(routine_definition) as "hash_code",
routine_type as "object_type"
from
information_schema.routines where routine_schema !~ '^_timescaledb_'

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

@@ -3,7 +3,7 @@ select
routine_name as "pure_name",
routine_schema as "schema_name",
max(routine_definition) as "definition",
max(md5(routine_definition)) as "hash_code",
max($md5Function(routine_definition)) as "hash_code",
routine_type as "object_type",
$typeAggFunc(data_type $typeAggParam) as "data_type",
max(external_language) as "language"

4
plugins/dbgate-plugin-postgres/src/backend/sql/tableModifications.js
View File

@@ -1,7 +1,7 @@
module.exports = `
select infoTables.table_schema as "schema_name", infoTables.table_name as "pure_name",
(
select md5(string_agg(
select $md5Function(string_agg(
infoColumns.column_name || '|' || infoColumns.data_type || '|' || infoColumns.is_nullable::varchar(255) || '|' || coalesce(infoColumns.character_maximum_length, -1)::varchar(255)
|| '|' || coalesce(infoColumns.numeric_precision, -1)::varchar(255) ,
',' order by infoColumns.ordinal_position
@@ -10,7 +10,7 @@ select infoTables.table_schema as "schema_name", infoTables.table_name as "pure_
where infoColumns.table_schema = infoTables.table_schema and infoColumns.table_name = infoTables.table_name
),
(
select md5(string_agg(
select $md5Function(string_agg(
infoConstraints.constraint_name || '|' || infoConstraints.constraint_type ,
',' order by infoConstraints.constraint_name
)) as "hash_code_constraints"

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

@@ -2,7 +2,7 @@ module.exports = `
select
table_name as "pure_name",
table_schema as "schema_name",
md5(view_definition) as "hash_code"
$md5Function(view_definition) as "hash_code"
from
information_schema.views where table_schema != 'information_schema' and table_schema != 'pg_catalog' and table_schema !~ '^_timescaledb_' and table_schema =SCHEMA_NAME_CONDITION
`;

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

@@ -3,7 +3,7 @@ select
table_name as "pure_name",
table_schema as "schema_name",
view_definition as "create_sql",
md5(view_definition) as "hash_code"
$md5Function(view_definition) as "hash_code"
from
information_schema.views
where table_schema !~ '^_timescaledb_' and table_schema =SCHEMA_NAME_CONDITION