golang/ent
1
0
Fork 0
mirror of https://github.com/ent/ent.git synced 2026-05-05 09:00:57 +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

4
entc/gen/template/dialect/sql/query.tmpl
View File

@@ -134,7 +134,7 @@ func ({{ $receiver }} *{{ $builder }}) querySpec() *sqlgraph.QuerySpec {
return _spec
}
func ({{ $receiver }} *{{ $builder }}) sqlQuery() *sql.Selector {
func ({{ $receiver }} *{{ $builder }}) sqlQuery(ctx context.Context) *sql.Selector {
builder := sql.Dialect({{ $receiver }}.driver.Dialect())
t1 := builder.Table({{ $.Package }}.Table)
selector := builder.Select(t1.Columns({{ $.Package }}.Columns...)...).From(t1)
@@ -166,7 +166,7 @@ func ({{ $receiver }} *{{ $builder }}) sqlQuery() *sql.Selector {
{{- $e := $.Scope.Edge }} {{/* the edge we need to genegrate the path to. */}}
{{- $ident := $.Scope.Ident -}}
{{- $receiver := $.Scope.Receiver }}
selector := {{ $receiver }}.sqlQuery()
selector := {{ $receiver }}.sqlQuery(ctx)
if err := selector.Err(); err != nil {
return nil, err
}