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

CockroachDB analysis #112

Browse Source
This commit is contained in:
Jan Prochazka
2021-05-14 16:44:48 +02:00
parent 216ef7736b
commit f9545eaf7f
2 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -122,7 +122,13 @@ const driver = {
return { rows: res.rows.map(row => zipDataRow(row, columns)), columns }; return { rows: res.rows.map(row => zipDataRow(row, columns)), columns };
}, },
async stream(client, sql, options) { async stream(client, sql, options) {
const sqlSplitted = identify(sql, { dialect: 'psql', strict: false }); let sqlSplitted;
try {
sqlSplitted = identify(sql, { dialect: 'psql', strict: false });
} catch (e) {
// workaround
sqlSplitted = [{ text: sql }];
}
for (const sqlItem of sqlSplitted) { for (const sqlItem of sqlSplitted) {
await runStreamItem(client, sqlItem.text, options); await runStreamItem(client, sqlItem.text, options);

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

@@ -47,6 +47,8 @@ from
AND n.nspname !~ '^pg_toast' AND n.nspname !~ '^pg_toast'
ORDER BY a.attnum ORDER BY a.attnum
) as tabledefinition ) as tabledefinition
inner join information_schema.tables on tables.table_schema = tabledefinition.nspname and tables.table_name = tabledefinition.relname
and tables.table_type not like '%VIEW%'
where ('tables:' || nspname || '.' || relname) =OBJECT_ID_CONDITION where ('tables:' || nspname || '.' || relname) =OBJECT_ID_CONDITION
group by relname, nspname, oid group by relname, nspname, oid
`; `;