feat(compiler): Support DO statements

internal/compiler/output_columns.go

@@ -115,6 +115,8 @@ func (c *Compiler) outputColumns(qc *QueryCatalog, node ast.Node) ([]*Column, er
  if isUnion{
  return c.outputColumns(qc, n.Larg)
  }
+ case *ast.DoStmt:
+ targets = &ast.List{}
  case *ast.CallStmt:
  targets = &ast.List{}
  case *ast.TruncateStmt, *ast.RefreshMatViewStmt, *ast.NotifyStmt, *ast.ListenStmt:
@@ -509,6 +511,8 @@ func (c *Compiler) sourceTables(qc *QueryCatalog, node ast.Node) ([]*Table, erro
  list = &ast.List{
  Items: append(n.FromClause.Items, n.Relations.Items...),
  }
+ case *ast.DoStmt:
+ list = &ast.List{}
  case *ast.CallStmt:
  list = &ast.List{}
  case *ast.NotifyStmt, *ast.ListenStmt:

internal/compiler/parse.go

@@ -39,6 +39,7 @@ func (c *Compiler) parseQuery(stmt ast.Node, src string, o opts.Parser) (*Query,
  case *ast.CallStmt:
  case *ast.SelectStmt:
  case *ast.DeleteStmt:
+ case *ast.DoStmt:
  case *ast.InsertStmt:
  if err := validate.InsertStmt(n); err != nil{
  return nil, err

internal/endtoend/testdata/do/postgresql/pgx/query.sql

@@ -0,0 +1,7 @@
+-- name: DoStuff :exec
+DO $$
+ BEGIN
+ ALTER TABLE authors
+ ADD COLUMN marked_for_processing bool;
+ END
+$$;

internal/endtoend/testdata/do/postgresql/pgx/schema.sql

@@ -0,0 +1,5 @@
+CREATE TABLE authors (
+ id BIGSERIAL PRIMARY KEY,
+ name text NOT NULL,
+ bio text
+);

internal/endtoend/testdata/do/postgresql/pgx/sqlc.json

@@ -0,0 +1,12 @@
+{
+ "version": "1",
+ "packages": [
+{
+ "path": "db",
+ "engine": "postgresql",
+ "schema": "schema.sql",
+ "queries": "query.sql",
+ "sql_package": "pgx/v5"
+ }
+ ]
+}

internal/endtoend/testdata/do/postgresql/pq/query.sql

@@ -0,0 +1,7 @@
+-- name: DoStuff :exec
+DO $$
+ BEGIN
+ ALTER TABLE authors
+ ADD COLUMN marked_for_processing bool;
+ END
+$$;

internal/endtoend/testdata/do/postgresql/pq/schema.sql

@@ -0,0 +1,5 @@
+CREATE TABLE authors (
+ id BIGSERIAL PRIMARY KEY,
+ name text NOT NULL,
+ bio text
+);

internal/endtoend/testdata/do/postgresql/pq/sqlc.json

@@ -0,0 +1,11 @@
+{
+ "version": "1",
+ "packages": [
+{
+ "path": "db",
+ "engine": "postgresql",
+ "schema": "schema.sql",
+ "queries": "query.sql"
+ }
+ ]
+}