mirror of
https://github.com/ent/ent.git
synced 2026-05-24 09:31:56 +03:00
entc/gen: add context.Context to to query builder (#1195)
* Add ctx to dialectQuery * go generate
This commit is contained in:
@@ -69,7 +69,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
|
||||
}
|
||||
@@ -91,7 +91,7 @@ func (gq *GroupQuery) QueryAdmin() *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
|
||||
}
|
||||
@@ -339,7 +339,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
|
||||
}
|
||||
@@ -564,7 +564,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)
|
||||
@@ -859,7 +859,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)
|
||||
}
|
||||
|
||||
|
||||
@@ -69,7 +69,7 @@ func (pq *PetQuery) QueryFriends() *PetQuery {
|
||||
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
|
||||
}
|
||||
@@ -91,7 +91,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
|
||||
}
|
||||
@@ -339,7 +339,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
|
||||
}
|
||||
@@ -564,7 +564,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)
|
||||
@@ -859,7 +859,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)
|
||||
}
|
||||
|
||||
|
||||
@@ -71,7 +71,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
|
||||
}
|
||||
@@ -93,7 +93,7 @@ func (uq *UserQuery) QueryFriends() *UserQuery {
|
||||
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
|
||||
}
|
||||
@@ -115,7 +115,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
|
||||
}
|
||||
@@ -137,7 +137,7 @@ func (uq *UserQuery) QueryManage() *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
|
||||
}
|
||||
@@ -409,7 +409,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
|
||||
}
|
||||
@@ -726,7 +726,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)
|
||||
@@ -1021,7 +1021,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)
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user