entc/gen: support field selections with eager-loading (#1100)

This commit is contained in:
Ariel Mashraki
2020-12-29 09:19:53 +02:00
committed by GitHub
parent 75d93200e9
commit a761561c9c
69 changed files with 482 additions and 70 deletions

View File

@@ -97,7 +97,13 @@ func ({{ $receiver }} *{{ $builder }}) querySpec() *sqlgraph.QuerySpec {
Unique: true,
}
if fields := {{ $receiver }}.fields; len(fields) > 0 {
_spec.Node.Columns = fields
_spec.Node.Columns = make([]string, 0, len(fields))
_spec.Node.Columns = append(_spec.Node.Columns, {{ $.Package }}.{{ $.ID.Constant }})
for i := range fields {
if fields[i] != {{ $.Package }}.{{ $.ID.Constant }} {
_spec.Node.Columns = append(_spec.Node.Columns, fields[i])
}
}
}
if ps := {{ $receiver }}.predicates; len(ps) > 0 {
_spec.Predicate = func(selector *sql.Selector) {