diff --git a/internal/dinosql/parser.go b/internal/dinosql/parser.go index 9b8886a36f..83fed3a888 100644 --- a/internal/dinosql/parser.go +++ b/internal/dinosql/parser.go @@ -439,6 +439,8 @@ func parseQuery(c core.Catalog, stmt nodes.Node, source string, rewriteParameter return nil, errUnsupportedStatementType } + // spew.Dump(stmt) + rawSQL, err := pluckQuery(source, raw) if err != nil { return nil, err @@ -942,6 +944,18 @@ func outputColumns(qc *QueryCatalog, node nodes.Node) ([]core.Column, error) { cols = append(cols, core.Column{Name: name, DataType: "any"}) } + case nodes.SubLink: + name := "exists" + if res.Name != nil { + name = *res.Name + } + switch n.SubLinkType { + case nodes.EXISTS_SUBLINK: + cols = append(cols, core.Column{Name: name, DataType: "bool", NotNull: true}) + default: + cols = append(cols, core.Column{Name: name, DataType: "any", NotNull: false}) + } + case nodes.TypeCast: if n.TypeName == nil { return nil, errors.New("no type name type cast") diff --git a/internal/endtoend/testdata/select_exists/go/db.go b/internal/endtoend/testdata/select_exists/go/db.go new file mode 100644 index 0000000000..6a99519302 --- /dev/null +++ b/internal/endtoend/testdata/select_exists/go/db.go @@ -0,0 +1,29 @@ +// Code generated by sqlc. DO NOT EDIT. + +package querytest + +import ( + "context" + "database/sql" +) + +type DBTX interface { + ExecContext(context.Context, string, ...interface{}) (sql.Result, error) + PrepareContext(context.Context, string) (*sql.Stmt, error) + QueryContext(context.Context, string, ...interface{}) (*sql.Rows, error) + QueryRowContext(context.Context, string, ...interface{}) *sql.Row +} + +func New(db DBTX) *Queries { + return &Queries{db: db} +} + +type Queries struct { + db DBTX +} + +func (q *Queries) WithTx(tx *sql.Tx) *Queries { + return &Queries{ + db: tx, + } +} diff --git a/internal/endtoend/testdata/select_exists/go/models.go b/internal/endtoend/testdata/select_exists/go/models.go new file mode 100644 index 0000000000..7c3b98644d --- /dev/null +++ b/internal/endtoend/testdata/select_exists/go/models.go @@ -0,0 +1,9 @@ +// Code generated by sqlc. DO NOT EDIT. + +package querytest + +import () + +type Bar struct { + ID int32 +} diff --git a/internal/endtoend/testdata/select_exists/go/query.sql.go b/internal/endtoend/testdata/select_exists/go/query.sql.go new file mode 100644 index 0000000000..f1e203d85c --- /dev/null +++ b/internal/endtoend/testdata/select_exists/go/query.sql.go @@ -0,0 +1,27 @@ +// Code generated by sqlc. DO NOT EDIT. +// source: query.sql + +package querytest + +import ( + "context" +) + +const barExists = `-- name: BarExists :one +SELECT + EXISTS ( + SELECT + 1 + FROM + bar + where + id = $1 + ) +` + +func (q *Queries) BarExists(ctx context.Context, id int32) (bool, error) { + row := q.db.QueryRowContext(ctx, barExists, id) + var exists bool + err := row.Scan(&exists) + return exists, err +} diff --git a/internal/endtoend/testdata/select_exists/query.sql b/internal/endtoend/testdata/select_exists/query.sql new file mode 100644 index 0000000000..aae94835e1 --- /dev/null +++ b/internal/endtoend/testdata/select_exists/query.sql @@ -0,0 +1,12 @@ +CREATE TABLE bar (id serial not null); + +-- name: BarExists :one +SELECT + EXISTS ( + SELECT + 1 + FROM + bar + where + id = $1 + ); diff --git a/internal/endtoend/testdata/select_exists/sqlc.json b/internal/endtoend/testdata/select_exists/sqlc.json new file mode 100644 index 0000000000..1161aac713 --- /dev/null +++ b/internal/endtoend/testdata/select_exists/sqlc.json @@ -0,0 +1,9 @@ +{ + "version": "1", + "packages": [{ + "path": "go", + "name": "querytest", + "schema": "query.sql", + "queries": "query.sql" + }] +}