postgresql materialized views #123

plugins/dbgate-plugin-postgres/src/backend/Analyser.js

@@ -56,6 +56,12 @@ class Analyser extends DatabaseAnalyser {
     const pkColumns = await this.driver.query(this.pool, this.createQuery('primaryKeys', ['tables']));
     const fkColumns = await this.driver.query(this.pool, this.createQuery('foreignKeys', ['tables']));
     const views = await this.driver.query(this.pool, this.createQuery('views', ['views']));
+    const matviews = this.driver.dialect.materializedViews
+      ? await this.driver.query(this.pool, this.createQuery('matviews', ['matviews']))
+      : null;
+    const matviewColumns = this.driver.dialect.materializedViews
+      ? await this.driver.query(this.pool, this.createQuery('matviewColumns', ['matviews']))
+      : null;
     const routines = await this.driver.query(this.pool, this.createQuery('routines', ['procedures', 'functions']));
 
     return {
@@ -108,6 +114,18 @@ class Analyser extends DatabaseAnalyser {
           .filter(col => col.pure_name == view.pure_name && col.schema_name == view.schema_name)
           .map(getColumnInfo),
       })),
+      matviews: matviews
+        ? matviews.rows.map(matview => ({
+            objectId: `matviews:${matview.schema_name}.${matview.pure_name}`,
+            pureName: matview.pure_name,
+            schemaName: matview.schema_name,
+            contentHash: matview.hash_code,
+            createSql: `CREATE MATERIALIZED VIEW "${matview.schema_name}"."${matview.pure_name}"\nAS\n${matview.definition}`,
+            columns: matviewColumns.rows
+              .filter(col => col.pure_name == matview.pure_name && col.schema_name == matview.schema_name)
+              .map(getColumnInfo),
+          }))
+        : undefined,
       procedures: routines.rows
         .filter(x => x.object_type == 'PROCEDURE')
         .map(proc => ({
@@ -133,6 +151,9 @@ class Analyser extends DatabaseAnalyser {
       ? await this.driver.query(this.pool, this.createQuery('tableModifications'))
       : null;
     const viewModificationsQueryData = await this.driver.query(this.pool, this.createQuery('viewModifications'));
+    const matviewModificationsQueryData = this.driver.dialect.materializedViews
+      ? await this.driver.query(this.pool, this.createQuery('matviewModifications'))
+      : null;
     const routineModificationsQueryData = await this.driver.query(this.pool, this.createQuery('routineModifications'));
 
     return {
@@ -150,6 +171,14 @@ class Analyser extends DatabaseAnalyser {
         schemaName: x.schema_name,
         contentHash: x.hash_code,
       })),
+      matviews: matviewModificationsQueryData
+        ? matviewModificationsQueryData.rows.map(x => ({
+            objectId: `matviews:${x.schema_name}.${x.pure_name}`,
+            pureName: x.pure_name,
+            schemaName: x.schema_name,
+            contentHash: x.hash_code,
+          }))
+        : undefined,
       procedures: routineModificationsQueryData.rows
         .filter(x => x.object_type == 'PROCEDURE')
         .map(x => ({

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

@@ -2,11 +2,14 @@ const columns = require('./columns');
 const tableModifications = require('./tableModifications');
 const tableList = require('./tableList');
 const viewModifications = require('./viewModifications');
+const matviewModifications = require('./matviewModifications');
 const primaryKeys = require('./primaryKeys');
 const foreignKeys = require('./foreignKeys');
 const views = require('./views');
+const matviews = require('./matviews');
 const routines = require('./routines');
 const routineModifications = require('./routineModifications');
+const matviewColumns = require('./matviewColumns');
 
 module.exports = {
   columns,
@@ -18,4 +21,7 @@ module.exports = {
   views,
   routines,
   routineModifications,
+  matviews,
+  matviewModifications,
+  matviewColumns,
 };

plugins/dbgate-plugin-postgres/src/backend/sql/matviewColumns.js

@@ -0,0 +1,17 @@
+module.exports = `
+SELECT pg_namespace.nspname AS "schema_name"
+    , pg_class.relname AS "pure_name"
+    , pg_attribute.attname AS "column_name"
+    , pg_catalog.format_type(pg_attribute.atttypid, pg_attribute.atttypmod) AS "data_type"
+FROM pg_catalog.pg_class
+    INNER JOIN pg_catalog.pg_namespace
+        ON pg_class.relnamespace = pg_namespace.oid
+    INNER JOIN pg_catalog.pg_attribute
+        ON pg_class.oid = pg_attribute.attrelid
+-- Keeps only materialized views, and non-db/catalog/index columns 
+WHERE pg_class.relkind = 'm'
+    AND pg_attribute.attnum >= 1
+    AND ('matviews:' || pg_namespace.nspname || '.' || pg_class.relname) =OBJECT_ID_CONDITION
+
+ORDER BY pg_attribute.attnum
+`;

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

@@ -0,0 +1,8 @@
+module.exports = `
+select 
+  matviewname as "pure_name",
+  schemaname as "schema_name",
+  md5(definition) as "hash_code"
+from
+  pg_catalog.pg_matviews WHERE schemaname NOT LIKE 'pg_%' 
+`;

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

@@ -0,0 +1,10 @@
+module.exports = `
+select 
+  matviewname as "pure_name",
+  schemaname as "schema_name",
+  definition as "definition",
+  md5(definition) as "hash_code"
+from
+  pg_catalog.pg_matviews WHERE schemaname NOT LIKE 'pg_%' 
+  and ('matviews:' || schemaname || '.' ||  matviewname) =OBJECT_ID_CONDITION
+`;

plugins/dbgate-plugin-postgres/src/frontend/drivers.js

@@ -29,6 +29,10 @@ const postgresDriver = {
   engine: 'postgres@dbgate-plugin-postgres',
   title: 'Postgre SQL',
   defaultPort: 5432,
+  dialect: {
+    ...dialect,
+    materializedViews: true,
+  },
 };
 
 /** @type {import('dbgate-types').EngineDriver} */