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/gen: add context.Context to to query builder (#1195)

Browse Source 
* Add ctx to dialectQuery

* go generate
This commit is contained in:
Marwan Sulaiman
2021-01-20 04:46:08 -05:00
committed by GitHub
parent 59ffee504d
commit 8063d4ac93
84 changed files with 421 additions and 421 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
examples/m2m2types/ent/group_query.go
View File

@@ -67,7 +67,7 @@ func (gq *GroupQuery) QueryUsers() *UserQuery {
if err := gq.prepareQuery(ctx); err != nil {
return nil, err
}
selector := gq.sqlQuery()
selector := gq.sqlQuery(ctx)
if err := selector.Err(); err != nil {
return nil, err
}
@@ -303,7 +303,7 @@ func (gq *GroupQuery) GroupBy(field string, fields ...string) *GroupGroupBy {
if err := gq.prepareQuery(ctx); err != nil {
return nil, err
}
return gq.sqlQuery(), nil
return gq.sqlQuery(ctx), nil
}
return group
}
@@ -495,7 +495,7 @@ func (gq *GroupQuery) querySpec() *sqlgraph.QuerySpec {
return _spec
}
func (gq *GroupQuery) sqlQuery() *sql.Selector {
func (gq *GroupQuery) sqlQuery(ctx context.Context) *sql.Selector {
builder := sql.Dialect(gq.driver.Dialect())
t1 := builder.Table(group.Table)
selector := builder.Select(t1.Columns(group.Columns...)...).From(t1)
@@ -790,7 +790,7 @@ func (gs *GroupSelect) Scan(ctx context.Context, v interface{}) error {
if err := gs.prepareQuery(ctx); err != nil {
return err
}
gs.sql = gs.GroupQuery.sqlQuery()
gs.sql = gs.GroupQuery.sqlQuery(ctx)
return gs.sqlScan(ctx, v)
}

8
examples/m2m2types/ent/user_query.go
View File

@@ -67,7 +67,7 @@ func (uq *UserQuery) QueryGroups() *GroupQuery {
if err := uq.prepareQuery(ctx); err != nil {
return nil, err
}
selector := uq.sqlQuery()
selector := uq.sqlQuery(ctx)
if err := selector.Err(); err != nil {
return nil, err
}
@@ -303,7 +303,7 @@ func (uq *UserQuery) GroupBy(field string, fields ...string) *UserGroupBy {
if err := uq.prepareQuery(ctx); err != nil {
return nil, err
}
return uq.sqlQuery(), nil
return uq.sqlQuery(ctx), nil
}
return group
}
@@ -495,7 +495,7 @@ func (uq *UserQuery) querySpec() *sqlgraph.QuerySpec {
return _spec
}
func (uq *UserQuery) sqlQuery() *sql.Selector {
func (uq *UserQuery) sqlQuery(ctx context.Context) *sql.Selector {
builder := sql.Dialect(uq.driver.Dialect())
t1 := builder.Table(user.Table)
selector := builder.Select(t1.Columns(user.Columns...)...).From(t1)
@@ -790,7 +790,7 @@ func (us *UserSelect) Scan(ctx context.Context, v interface{}) error {
if err := us.prepareQuery(ctx); err != nil {
return err
}
us.sql = us.UserQuery.sqlQuery()
us.sql = us.UserQuery.sqlQuery(ctx)
return us.sqlScan(ctx, v)
}