add materialized view support

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

@@ -1237,13 +1237,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 {
@@ -2842,6 +2843,20 @@ func convertViewStmt(n *pg.ViewStmt) *ast.ViewStmt {
 	}
 }
 
+// func convertMatViewStmt(n *pg.ViewStmt) *ast.MatViewStmt {
+// 	if n == nil {
+// 		return nil
+// 	}
+// 	return &ast.MatViewStmt{
+// 		MatView:            convertRangeVar(n.View),
+// 		Aliases:         convertSlice(n.Aliases),
+// 		Query:           convertNode(n.Query),
+// 		Replace:         n.Replace,
+// 		Options:         convertSlice(n.Options),
+// 		WithCheckOption: ast.ViewCheckOption(n.WithCheckOption),
+// 	}
+// }
+
 func convertWindowClause(n *pg.WindowClause) *ast.WindowClause {
 	if n == nil {
 		return nil

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

@@ -0,0 +1,14 @@
+package ast
+
+type MatViewStmt struct {
+	MatView         *RangeVar
+	Aliases         *List
+	Query           Node
+	Replace         bool
+	Options         *List
+	WithCheckOption ViewCheckOption
+}
+
+func (n *MatViewStmt) Pos() int {
+	return 0
+}

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

@@ -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:
+		return c.createTableAs(n, colGen)
+
 	case *ast.ViewStmt:
 		err = c.createView(n, colGen)
 

internal/endtoend/testdata/materialized_views/postgresql/pgx/go/models.go

@@ -247,3 +247,51 @@ 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)
+
+	// if stmt.Replace && err == nil {
+	// 	schema.Tables[existingIdx] = &tbl
+	// } else {
+	// 	schema.Tables = append(schema.Tables, &tbl)
+	// }
+
+	return nil
+}