entc/gen: add context.Context to to query builder (#1195)

* 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

View File

@@ -67,7 +67,7 @@ func (pq *PetQuery) QueryOwner() *UserQuery {
if err := pq.prepareQuery(ctx); err != nil {
return nil, err
}
selector := pq.sqlQuery()
selector := pq.sqlQuery(ctx)
if err := selector.Err(); err != nil {
return nil, err
}
@@ -303,7 +303,7 @@ func (pq *PetQuery) GroupBy(field string, fields ...string) *PetGroupBy {
if err := pq.prepareQuery(ctx); err != nil {
return nil, err
}
return pq.sqlQuery(), nil
return pq.sqlQuery(ctx), nil
}
return group
}
@@ -463,7 +463,7 @@ func (pq *PetQuery) querySpec() *sqlgraph.QuerySpec {
return _spec
}
func (pq *PetQuery) sqlQuery() *sql.Selector {
func (pq *PetQuery) sqlQuery(ctx context.Context) *sql.Selector {
builder := sql.Dialect(pq.driver.Dialect())
t1 := builder.Table(pet.Table)
selector := builder.Select(t1.Columns(pet.Columns...)...).From(t1)
@@ -758,7 +758,7 @@ func (ps *PetSelect) Scan(ctx context.Context, v interface{}) error {
if err := ps.prepareQuery(ctx); err != nil {
return err
}
ps.sql = ps.PetQuery.sqlQuery()
ps.sql = ps.PetQuery.sqlQuery(ctx)
return ps.sqlScan(ctx, v)
}

View File

@@ -67,7 +67,7 @@ func (uq *UserQuery) QueryPets() *PetQuery {
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
}
@@ -460,7 +460,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)
@@ -755,7 +755,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)
}