diff --git a/internal/compiler/output_columns.go b/internal/compiler/output_columns.go index d12e7f9cbb..3cec9c6ae7 100644 --- 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 index c7b6f7f3f8..269783403e 100644 --- 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 new file mode 100644 index 0000000000..5a2cde8cda --- /dev/null +++ b/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v4/go/db.go @@ -0,0 +1,32 @@ +// Code generated by sqlc. DO NOT EDIT. +// versions: +// sqlc v1.18.0 + +package querytest + +import ( + "context" + + "github.com/jackc/pgconn" + "github.com/jackc/pgx/v4" +) + +type DBTX interface { + Exec(context.Context, string, ...interface{}) (pgconn.CommandTag, error) + Query(context.Context, string, ...interface{}) (pgx.Rows, error) + QueryRow(context.Context, string, ...interface{}) pgx.Row +} + +func New(db DBTX) *Queries { + return &Queries{db: db} +} + +type Queries struct { + db DBTX +} + +func (q *Queries) WithTx(tx pgx.Tx) *Queries { + return &Queries{ + db: tx, + } +} diff --git a/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v4/go/models.go b/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v4/go/models.go new file mode 100644 index 0000000000..8f770b5e6d --- /dev/null +++ b/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v4/go/models.go @@ -0,0 +1,11 @@ +// Code generated by sqlc. DO NOT EDIT. +// versions: +// sqlc v1.18.0 + +package querytest + +import () + +type Myview struct { + int32 +} 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 new file mode 100644 index 0000000000..51db328f32 --- /dev/null +++ b/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v4/go/query.sql.go @@ -0,0 +1,19 @@ +// Code generated by sqlc. DO NOT EDIT. +// versions: +// sqlc v1.18.0 +// source: query.sql + +package querytest + +import ( + "context" +) + +const refresh = `-- name: Refresh :exec +REFRESH MATERIALIZED VIEW myview +` + +func (q *Queries) Refresh(ctx context.Context) error { + _, err := q.db.Exec(ctx, refresh) + return err +} diff --git a/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v4/query.sql b/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v4/query.sql new file mode 100644 index 0000000000..db4a722fd4 --- /dev/null +++ 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 new file mode 100644 index 0000000000..4d8ec80dfb --- /dev/null +++ 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 new file mode 100644 index 0000000000..7bc02a1d7d --- /dev/null +++ b/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v5/go/db.go @@ -0,0 +1,32 @@ +// Code generated by sqlc. DO NOT EDIT. +// versions: +// sqlc v1.18.0 + +package querytest + +import ( + "context" + + "github.com/jackc/pgx/v5" + "github.com/jackc/pgx/v5/pgconn" +) + +type DBTX interface { + Exec(context.Context, string, ...interface{}) (pgconn.CommandTag, error) + Query(context.Context, string, ...interface{}) (pgx.Rows, error) + QueryRow(context.Context, string, ...interface{}) pgx.Row +} + +func New(db DBTX) *Queries { + return &Queries{db: db} +} + +type Queries struct { + db DBTX +} + +func (q *Queries) WithTx(tx pgx.Tx) *Queries { + return &Queries{ + db: tx, + } +} diff --git a/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v5/go/models.go b/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v5/go/models.go new file mode 100644 index 0000000000..8f770b5e6d --- /dev/null +++ b/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v5/go/models.go @@ -0,0 +1,11 @@ +// Code generated by sqlc. DO NOT EDIT. +// versions: +// sqlc v1.18.0 + +package querytest + +import () + +type Myview struct { + int32 +} 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 new file mode 100644 index 0000000000..51db328f32 --- /dev/null +++ b/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v5/go/query.sql.go @@ -0,0 +1,19 @@ +// Code generated by sqlc. DO NOT EDIT. +// versions: +// sqlc v1.18.0 +// source: query.sql + +package querytest + +import ( + "context" +) + +const refresh = `-- name: Refresh :exec +REFRESH MATERIALIZED VIEW myview +` + +func (q *Queries) Refresh(ctx context.Context) error { + _, err := q.db.Exec(ctx, refresh) + return err +} diff --git a/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v5/query.sql b/internal/endtoend/testdata/refreshmatview/postgresql/pgx/v5/query.sql new file mode 100644 index 0000000000..db4a722fd4 --- /dev/null +++ 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 new file mode 100644 index 0000000000..f7344cf1ba --- /dev/null +++ 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", + } + ] +}