Support for parsing REFRESH MATERIALIZED VIEW

The engine already understood it, just had to teach the compiler. closes sqlc-dev#2264
Jille · Jun 21, 2023 · f62cb26 · f62cb26
1 parent f086636
commit f62cb26
Show file tree

Hide file tree

Showing 12 changed files with 165 additions and 1 deletion.
diff --git a/internal/compiler/output_columns.go b/internal/compiler/output_columns.go
@@ -115,7 +115,7 @@ func (c *Compiler) outputColumns(qc *QueryCatalog, node ast.Node) ([]*Column, er
 		}
 	case *ast.CallStmt:
 		targets = &ast.List{}
-	case *ast.TruncateStmt:
+	case *ast.TruncateStmt, *ast.RefreshMatViewStmt:
 		targets = &ast.List{}
 	case *ast.UpdateStmt:
 		targets = n.ReturningList
@@ -480,6 +480,11 @@ func (c *Compiler) sourceTables(qc *QueryCatalog, node ast.Node) ([]*Table, erro
 			_, ok := node.(*ast.RangeVar)
 			return ok
 		})
+	case *ast.RefreshMatViewStmt:
+		list = astutils.Search(n.Relation, func(node ast.Node) bool {
+			_, ok := node.(*ast.RangeVar)
+			return ok
+		})
 	case *ast.UpdateStmt:
 		list = &ast.List{
 			Items: append(n.FromClause.Items, n.Relations.Items...),

diff --git a/internal/compiler/parse.go b/internal/compiler/parse.go
@@ -50,6 +50,7 @@ func (c *Compiler) parseQuery(stmt ast.Node, src string, o opts.Parser) (*Query,
 		}
 	case *ast.TruncateStmt:
 	case *ast.UpdateStmt:
+	case *ast.RefreshMatViewStmt:
 	default:
 		return nil, ErrUnsupportedStatementType
 	}

diff --git a/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v4/go/db.go b/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v4/go/db.go
diff --git a/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v4/go/models.go b/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v4/go/models.go
diff --git a/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v4/go/query.sql.go b/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v4/go/query.sql.go
diff --git a/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v4/query.sql b/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v4/query.sql
@@ -0,0 +1,4 @@
+CREATE MATERIALIZED VIEW myview AS (SELECT 1);
+
+-- name: Refresh :exec
+REFRESH MATERIALIZED VIEW myview;
diff --git a/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v4/sqlc.json b/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v4/sqlc.json
@@ -0,0 +1,13 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "go",
+      "engine": "postgresql",
+      "sql_package": "pgx/v4",
+      "name": "querytest",
+      "schema": "query.sql",
+      "queries": "query.sql",
+    }
+  ]
+}
diff --git a/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v5/go/db.go b/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v5/go/db.go
diff --git a/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v5/go/models.go b/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v5/go/models.go
diff --git a/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v5/go/query.sql.go b/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v5/go/query.sql.go
diff --git a/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v5/query.sql b/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v5/query.sql
@@ -0,0 +1,4 @@
+CREATE MATERIALIZED VIEW myview AS (SELECT 1);
+
+-- name: Refresh :exec
+REFRESH MATERIALIZED VIEW myview;
diff --git a/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v5/sqlc.json b/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v5/sqlc.json
@@ -0,0 +1,13 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "go",
+      "engine": "postgresql",
+      "sql_package": "pgx/v5",
+      "name": "querytest",
+      "schema": "query.sql",
+      "queries": "query.sql",
+    }
+  ]
+}