golang/ent
1
0
Fork 0
mirror of https://github.com/ent/ent.git synced 2026-05-24 09:31:56 +03:00
Code Issues Packages Projects Releases Wiki Activity

entc/integration: add examples for sub-queries + minor fixed for selector builder (#2509)

Browse Source 
* dialect/sql: move FROM to inside switch

* dialect/sql: adds simple example of subquery usage

* dialect/sql: adds a more complex example with join

* Apply suggestions from code review

Co-authored-by: Ariel Mashraki <7413593+a8m@users.noreply.github.com>

* dialect/sql: fix missing type and change modify to where

Co-authored-by: Ariel Mashraki <7413593+a8m@users.noreply.github.com>
This commit is contained in:
Pedro Henrique
2022-05-05 16:46:13 -03:00
committed by GitHub
parent 0c75ffc007
commit 4d1c0af20d
3 changed files with 36 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
dialect/sql/builder.go
View File

@@ -2657,12 +2657,13 @@ func (s *Selector) Query() (string, []interface{}) {
} else {
b.WriteString("*")
}
b.WriteString(" FROM ")
switch t := s.from.(type) {
case *SelectTable:
b.WriteString(" FROM ")
t.SetDialect(s.dialect)
b.WriteString(t.ref())
case *Selector:
b.WriteString(" FROM ")
t.SetDialect(s.dialect)
b.Nested(func(b *Builder) {
b.Join(t)
@@ -2670,6 +2671,7 @@ func (s *Selector) Query() (string, []interface{}) {
b.WriteString(" AS ")
b.Ident(t.as)
case *WithBuilder:
b.WriteString(" FROM ")
t.SetDialect(s.dialect)
b.Ident(t.Name())
}

8
dialect/sql/builder_test.go
View File

@@ -1288,6 +1288,14 @@ func TestBuilder(t *testing.T) {
wantQuery: "WITH `groups` AS (SELECT * FROM `groups` WHERE `name` = ?) SELECT `age` FROM `groups`",
wantArgs: []interface{}{"bar"},
},
{
input: SelectExpr(Raw("1")),
wantQuery: "SELECT 1",
},
{
input: Select("*").From(SelectExpr(Raw("1")).As("s")),
wantQuery: "SELECT * FROM (SELECT 1) AS `s`",
},
{
input: func() Querier {
builder := Dialect(dialect.Postgres)