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 eager-loading support (#263)

Browse Source 
* entc/gen: add OwnFK indicator for type edges

* entc/gen: add Edges field for generated types

* entc/gen: add With<T> method to query-builder template

* entc/gen: scan and assign foreign-keys on eager-loading

* entc/gen: load fk-relations (wip)

* entc/integration: add o2m/m2o tests for eager-loading

* entc/gen: add m2m support for eager-loading

* entc/gen: add integration tests for m2m and subgraphs

* entc/gen/integration: add tests for o2o eager-loading

* all: generate all assets
This commit is contained in:
Ariel Mashraki
2020-01-13 17:21:26 +02:00
committed by GitHub
parent cd366c07e2
commit caf721df47
171 changed files with 6400 additions and 398 deletions
Download Patch File Download Diff File Expand all files Collapse all files

57
entc/gen/template/dialect/sql/decode.tmpl
View File

@@ -10,17 +10,29 @@ in the LICENSE file in the root directory of this source tree.
// scanValues returns the types for scanning values from sql.Rows.
func (*{{ $.Name }}) scanValues() []interface{} {
return []interface{} {
&{{ if not $.ID.UserDefined }}sql.NullInt64{{ else }}{{ $.ID.NullType }}{{ end }}{},
&{{ if not $.ID.UserDefined }}sql.NullInt64{{ else }}{{ $.ID.NullType }}{{ end }}{}, // {{ $.ID.Name }}
{{- range $_, $f := $.Fields }}
&{{ $f.NullType }}{},
&{{ $f.NullType }}{}, // {{ $f.Name }}
{{- end }}
}
}
{{- with $.ForeignKeys }}
// fkValues returns the types for scanning foreign-keys values from sql.Rows.
func (*{{ $.Name }}) fkValues() []interface{} {
return []interface{} {
{{- range $fk := . }}
{{- $f := $fk.Field }}
&{{ if not $f.UserDefined }}sql.NullInt64{{ else }}{{ $f.NullType }}{{ end }}{}, // {{ $f.Name }}
{{- end }}
}
}
{{- end }}
// assignValues assigns the values that were returned from sql.Rows (after scanning)
// to the {{ $.Name }} fields.
func ({{ $receiver }} *{{ $.Name }}) assignValues(values ...interface{}) error {
if m, n := len(values), len({{ $.Package }}.Columns); m != n {
if m, n := len(values), len({{ $.Package }}.Columns); m < n {
return fmt.Errorf("mismatch number of scan values: %d != %d", m, n)
}
{{- if and $.ID.UserDefined (or $.ID.IsString $.ID.IsUUID) }}
@@ -40,6 +52,26 @@ func ({{ $receiver }} *{{ $.Name }}) assignValues(values ...interface{}) error {
{{ template "dialect/sql/decode/field" . }}
{{- end }}
{{- end }}
{{- with $.ForeignKeys }}
values = values[{{ len $.Fields }}:]
if len(values) == len({{ $.Package }}.ForeignKeys) {
{{- range $i, $fk := . }}
{{- $f := $fk.Field }}
{{- if and $f.UserDefined (or $f.IsString $f.IsUUID) }}
{{- with extend $ "Idx" 0 "Field" $f "Rec" $receiver "StructField" $f.Name }}
{{ template "dialect/sql/decode/field" . }}
{{- end }}
{{- else }}
if value, ok := values[{{ $i }}].(*sql.NullInt64); !ok {
return fmt.Errorf("unexpected type %T for edge-field {{ $f.Name}}", value)
} else if value.Valid {
{{ $receiver }}.{{ $f.Name }} = new({{ $f.Type }})
*{{ $receiver }}.{{ $f.Name }} = {{ if $f.IsString }}strconv.FormatInt(value.Int64, 10){{ else }}{{ $f.Type }}(value.Int64){{ end }}
}
{{- end }}
{{- end }}
}
{{- end }}
return nil
}
{{ end }}
@@ -48,11 +80,12 @@ func ({{ $receiver }} *{{ $.Name }}) assignValues(values ...interface{}) error {
{{- $i := $.Scope.Idx -}}
{{- $f := $.Scope.Field -}}
{{- $ret := $.Scope.Rec -}}
{{- $field := $f.StructField }}{{ with $.Scope.StructField }}{{ $field = . }}{{ end }}
{{- if $f.IsJSON }}
if value, ok := values[{{ $i }}].(*{{ $f.NullType }}); !ok {
return fmt.Errorf("unexpected type %T for field {{ $f.Name }}", values[{{ $i }}])
} else if value != nil && len(*value) > 0 {
if err := json.Unmarshal(*value, &{{ $ret }}.{{ $f.StructField }}); err != nil {
if err := json.Unmarshal(*value, &{{ $ret }}.{{ $field }}); err != nil {
return fmt.Errorf("unmarshal field {{ $f.Name }}: %v", err)
}
}
@@ -63,14 +96,14 @@ func ({{ $receiver }} *{{ $.Name }}) assignValues(values ...interface{}) error {
{{- if hasPrefix $nulltype "sql" }}
} else if value.Valid {
{{- if $f.Nillable }}
{{ $ret }}.{{ $f.StructField }} = new({{ $f.Type }})
*{{ $ret }}.{{ $f.StructField }} = {{ $f.NullTypeField "value" }}
{{ $ret }}.{{ $field }} = new({{ $f.Type }})
*{{ $ret }}.{{ $field }} = {{ $f.NullTypeField "value" }}
{{- else }}
{{ $ret }}.{{ $f.StructField }} = {{ $f.NullTypeField "value" }}
{{ $ret }}.{{ $field }} = {{ $f.NullTypeField "value" }}
{{- end }}
{{- else }}
} else if value != nil {
{{ $ret }}.{{ $f.StructField }} = *value
{{ $ret }}.{{ $field }} = *value
{{- end }}
}
{{- end }}
@@ -78,3 +111,11 @@ func ({{ $receiver }} *{{ $.Name }}) assignValues(values ...interface{}) error {
{{ define "dialect/sql/decode/many" }}
{{ end }}
{{/* Additional fields for the generated model for holding the foreign-keys */}}
{{ define "dialect/sql/model/fields" }}
{{- range $fk := $.ForeignKeys }}
{{- $f := $fk.Field }}
{{ $f.Name }} {{ if $f.Nillable }}*{{ end }}{{ $f.Type }}
{{- end }}
{{ end }}