feat(postgresql): Add materialized view support (#1509)

Author: ludusrusso

internal/endtoend/testdata/create_materialized_view/postgresql/go/db.go

@@ -0,0 +1,3 @@
+-- name: GetFirst :many
+SELECT * FROM mat_first_view;
+

internal/endtoend/testdata/create_materialized_view/postgresql/go/models.go

@@ -0,0 +1,3 @@
+CREATE TABLE foo (val text not null);
+
+CREATE MATERIALIZED VIEW mat_first_view AS SELECT * FROM foo;

internal/endtoend/testdata/create_materialized_view/postgresql/go/query.sql.go

@@ -0,0 +1,12 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "engine": "postgresql",
+      "path": "go",
+      "name": "querytest",
+      "schema": "schema.sql",
+      "queries": "query.sql"
+    }
+  ]
+}

internal/endtoend/testdata/create_materialized_view/postgresql/query.sql

@@ -0,0 +1,3 @@
+-- name: GetFirst :many
+SELECT * FROM second_table;
+

internal/endtoend/testdata/create_materialized_view/postgresql/schema.sql

@@ -0,0 +1,3 @@
+CREATE TABLE foo (val text not null);
+
+CREATE TABLE second_table AS SELECT * FROM foo;

internal/endtoend/testdata/create_materialized_view/postgresql/sqlc.json

@@ -0,0 +1,12 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "engine": "postgresql",
+      "path": "go",
+      "name": "querytest",
+      "schema": "schema.sql",
+      "queries": "query.sql"
+    }
+  ]
+}

internal/endtoend/testdata/create_table_as/postgresql/go/db.go

@@ -1264,13 +1264,14 @@ func convertCreateTableAsStmt(n *pg.CreateTableAsStmt) *ast.CreateTableAsStmt {
 	if n == nil {
 		return nil
 	}
-	return &ast.CreateTableAsStmt{
+	res := &ast.CreateTableAsStmt{
 		Query:        convertNode(n.Query),
 		Into:         convertIntoClause(n.Into),
 		Relkind:      ast.ObjectType(n.Relkind),
 		IsSelectInto: n.IsSelectInto,
 		IfNotExists:  n.IfNotExists,
 	}
+	return res
 }
 
 func convertCreateTableSpaceStmt(n *pg.CreateTableSpaceStmt) *ast.CreateTableSpaceStmt {

internal/endtoend/testdata/create_table_as/postgresql/go/models.go

@@ -288,6 +288,7 @@ func (c *Catalog) Update(stmt ast.Statement, colGen columnGenerator) error {
 	}
 	var err error
 	switch n := stmt.Raw.Stmt.(type) {
+
 	case *ast.AlterTableStmt:
 		err = c.alterTable(n)
 
@@ -330,6 +331,9 @@ func (c *Catalog) Update(stmt ast.Statement, colGen columnGenerator) error {
 	case *ast.CreateTableStmt:
 		err = c.createTable(n)
 
+	case *ast.CreateTableAsStmt:
+		err = c.createTableAs(n, colGen)
+
 	case *ast.ViewStmt:
 		err = c.createView(n, colGen)
 

internal/endtoend/testdata/create_table_as/postgresql/go/query.sql.go

@@ -247,3 +247,45 @@ func (c *Catalog) renameTable(stmt *ast.RenameTableStmt) error {
 	}
 	return nil
 }
+
+func (c *Catalog) createTableAs(stmt *ast.CreateTableAsStmt, colGen columnGenerator) error {
+	cols, err := colGen.OutputColumns(stmt.Query)
+	if err != nil {
+		return err
+	}
+
+	catName := ""
+	if stmt.Into.Rel.Catalogname != nil {
+		catName = *stmt.Into.Rel.Catalogname
+	}
+	schemaName := ""
+	if stmt.Into.Rel.Schemaname != nil {
+		schemaName = *stmt.Into.Rel.Schemaname
+	}
+
+	tbl := Table{
+		Rel: &ast.TableName{
+			Catalog: catName,
+			Schema:  schemaName,
+			Name:    *stmt.Into.Rel.Relname,
+		},
+		Columns: cols,
+	}
+
+	ns := tbl.Rel.Schema
+	if ns == "" {
+		ns = c.DefaultSchema
+	}
+	schema, err := c.getSchema(ns)
+	if err != nil {
+		return err
+	}
+	_, _, err = schema.getTable(tbl.Rel)
+	if err == nil {
+		return sqlerr.RelationExists(tbl.Rel.Name)
+	}
+
+	schema.Tables = append(schema.Tables, &tbl)
+
+	return nil
+}