Add support for pg_matviews

BemiHQ · Jan 6, 2025 · b2df4ff · b2df4ff
1 parent e15537c
commit b2df4ff
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 0 deletions.
diff --git a/src/query_handler_test.go b/src/query_handler_test.go
@@ -142,6 +142,11 @@ func TestHandleQuery(t *testing.T) {
 			"types":       {Uint32ToString(pgtype.BoolOID)},
 			"values":      {},
 		},
+		"SELECT schemaname, matviewname AS objectname FROM pg_catalog.pg_matviews": {
+			"description": {"schemaname", "objectname"},
+			"types":       {Uint32ToString(pgtype.BoolOID), Uint32ToString(pgtype.BoolOID)},
+			"values":      {},
+		},
 		// pg_namespace
 		"SELECT DISTINCT(nspname) FROM pg_catalog.pg_namespace WHERE nspname != 'information_schema' AND nspname != 'pg_catalog'": {
 			"description": {"nspname"},

diff --git a/src/query_parser_table.go b/src/query_parser_table.go
@@ -475,6 +475,7 @@ var PG_SYSTEM_TABLES = NewSet([]string{
 	"pg_language",
 	"pg_largeobject",
 	"pg_largeobject_metadata",
+	"pg_matviews",
 	"pg_namespace",
 	"pg_opclass",
 	"pg_operator",

diff --git a/src/select_remapper_table.go b/src/select_remapper_table.go
@@ -23,6 +23,7 @@ const (
 	PG_TABLE_PG_AUTH_MEMBERS       = "pg_auth_members"
 	PG_TABLE_PG_USER               = "pg_user"
 	PG_TABLE_PG_STAT_ACTIVITY      = "pg_stat_activity"
+	PG_TABLE_PG_MATVIEWS           = "pg_matviews"
 
 	PG_TABLE_TABLES = "tables"
 )
@@ -106,6 +107,10 @@ func (remapper *SelectRemapperTable) RemapTable(node *pgQuery.Node) *pgQuery.Nod
 			// pg_stat_activity -> return empty table
 			tableNode := parser.MakeEmptyTableNode(PG_TABLE_PG_STAT_ACTIVITY, PG_STAT_ACTIVITY_COLUMNS, qSchemaTable.Alias)
 			return remapper.overrideTable(node, tableNode)
+		case PG_TABLE_PG_MATVIEWS:
+			// pg_matviews -> return empty table
+			tableNode := parser.MakeEmptyTableNode(PG_TABLE_PG_MATVIEWS, PG_MATVIEWS_COLUMNS, qSchemaTable.Alias)
+			return remapper.overrideTable(node, tableNode)
 		default:
 			// pg_catalog.pg_* other system tables -> return as is
 			return node
@@ -285,3 +290,13 @@ var PG_STAT_ACTIVITY_COLUMNS = []string{
 	"query",
 	"backend_type",
 }
+
+var PG_MATVIEWS_COLUMNS = []string{
+	"schemaname",
+	"matviewname",
+	"matviewowner",
+	"tablespace",
+	"hasindexes",
+	"ispopulated",
+	"definition",
+}