postgre analyser

api/src/engines/postgres/index.js

@@ -1,4 +1,6 @@
 const { Client } = require('pg');
+const PostgreAnalyser = require('./PostgreAnalyser');
+const PostgreDumper = require('./PostgreDumper');
 
 /** @type {import('dbgate').SqlDialect} */
 const dialect = {
@@ -8,7 +10,8 @@ const dialect = {
   },
 };
 
-module.exports = {
+/** @type {import('dbgate').EngineDriver} */
+const driver = {
   async connect({ server, port, user, password, database }) {
     const client = new Client({ host: server, port, user, password, database: database || 'postgres' });
     await client.connect();
@@ -16,16 +19,26 @@ module.exports = {
   },
   async query(client, sql) {
     const res = await client.query(sql);
-    return { rows: res.rows };
+    return { rows: res.rows, columns: res.fields };
   },
   async getVersion(client) {
     const { rows } = await this.query(client, 'SELECT version()');
     const { version } = rows[0];
     return { version };
   },
+  async analyseFull(pool) {
+    const analyser = new PostgreAnalyser(pool, this);
+    await analyser.runAnalysis();
+    return analyser.result;
+  },
+  createDumper() {
+    return new PostgreDumper(this);
+  },
   async listDatabases(client) {
     const { rows } = await this.query(client, 'SELECT datname AS name FROM pg_database WHERE datistemplate = false');
     return rows;
   },
   dialect,
 };
+
+module.exports = driver;