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: use BuilderField for builder memebrs

Browse Source 
Summary: StructField should be used a member of the ent type (struct), and logic should be removed from templates.

Reviewed By: alexsn

Differential Revision: D17926151

fbshipit-source-id: f77c5ce8359816ec3a4ce1ab71dd98f205706deb
This commit is contained in:
Ariel Mashraki
2019-10-15 06:41:29 -07:00
committed by Facebook Github Bot
parent 2c8d7fcc8b
commit 64ed7580c0
10 changed files with 96 additions and 96 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
entc/gen/template/dialect/sql/create.tmpl
View File

@@ -21,7 +21,7 @@ func ({{ $receiver }} *{{ $builder }}) sqlSave(ctx context.Context) (*{{ $.Name
Insert({{ $.Package }}.Table).
Default()
{{- range $_, $f := $.Fields }}
if value := {{ $receiver }}.{{- $f.StructField }}; value != nil {
if value := {{ $receiver }}.{{- $f.BuilderField }}; value != nil {
{{- if $f.IsJSON }}
buf, err := json.Marshal(*value)
if err != nil {
@@ -44,9 +44,9 @@ func ({{ $receiver }} *{{ $builder }}) sqlSave(ctx context.Context) (*{{ $.Name
}
{{ $.Receiver }}.ID = {{ if $.ID.IsString }}strconv.FormatInt(id, 10){{ else }}{{ $.ID.Type }}(id){{ end }}
{{- range $_, $e := $.Edges }}
if len({{ $receiver }}.{{ $e.StructField }}) > 0 {
if len({{ $receiver }}.{{ $e.BuilderField }}) > 0 {
{{- if and $e.Unique $e.SelfRef }}{{/* O2O with self reference */}}
for eid := range {{ $receiver }}.{{ $e.StructField }} {
for eid := range {{ $receiver }}.{{ $e.BuilderField }} {
{{- template "dialect/sql/create/convertid" $e -}}
query, args := sql.Update({{ $.Package }}.{{ $e.TableConstant }}).
Set({{ $.Package }}.{{ $e.ColumnConstant }}, eid).
@@ -64,12 +64,12 @@ func ({{ $receiver }} *{{ $builder }}) sqlSave(ctx context.Context) (*{{ $.Name
if err != nil {
return nil, rollback(tx, err)
}
if int(affected) < len({{ $receiver }}.{{ $e.StructField }}) {
if int(affected) < len({{ $receiver }}.{{ $e.BuilderField }}) {
return nil, rollback(tx, &ErrConstraintFailed{msg: fmt.Sprintf("\"{{ $e.Name }}\" (%v) already connected to a different \"{{ $.Name }}\"", eid)})
}
}
{{- else if $e.M2M }}
for eid := range {{ $receiver }}.{{ $e.StructField }} {
for eid := range {{ $receiver }}.{{ $e.BuilderField }} {
{{- template "dialect/sql/create/convertid" $e -}}
{{ $a := 0 }}{{ $b := 1 }}{{- if $e.IsInverse }}{{ $a = 1 }}{{ $b = 0 }}{{ end }}
query, args := sql.Insert({{ $.Package }}.{{ $e.TableConstant }}).
@@ -84,7 +84,7 @@ func ({{ $receiver }} *{{ $builder }}) sqlSave(ctx context.Context) (*{{ $.Name
}
}
{{- else if $e.M2O }}
for eid := range {{ $receiver }}.{{ $e.StructField }} {
for eid := range {{ $receiver }}.{{ $e.BuilderField }} {
{{- template "dialect/sql/create/convertid" $e -}}
query, args := sql.Update({{ $.Package }}.{{ $e.TableConstant }}).
Set({{ $.Package }}.{{ $e.ColumnConstant }}, eid).
@@ -96,7 +96,7 @@ func ({{ $receiver }} *{{ $builder }}) sqlSave(ctx context.Context) (*{{ $.Name
}
{{- else if $e.O2M }}
p := sql.P()
for eid := range {{ $receiver }}.{{ $e.StructField }} {
for eid := range {{ $receiver }}.{{ $e.BuilderField }} {
{{- template "dialect/sql/create/convertid" $e -}}
p.Or().EQ({{ $e.Type.Package }}.{{ $e.Type.ID.Constant }}, eid)
}
@@ -111,17 +111,17 @@ func ({{ $receiver }} *{{ $builder }}) sqlSave(ctx context.Context) (*{{ $.Name
if err != nil {
return nil, rollback(tx, err)
}
if int(affected) < len({{ $receiver }}.{{ $e.StructField }}) {
return nil, rollback(tx, &ErrConstraintFailed{msg: fmt.Sprintf("one of \"{{ $e.Name }}\" %v already connected to a different \"{{ $.Name }}\"", keys({{ $receiver }}.{{ $e.StructField }}))})
if int(affected) < len({{ $receiver }}.{{ $e.BuilderField }}) {
return nil, rollback(tx, &ErrConstraintFailed{msg: fmt.Sprintf("one of \"{{ $e.Name }}\" %v already connected to a different \"{{ $.Name }}\"", keys({{ $receiver }}.{{ $e.BuilderField }}))})
}
{{- else }}{{/* O2O */}}
{{- if $.Type.ID.IsString }}
eid, err := strconv.Atoi(keys({{ $receiver }}.{{ $e.StructField }})[0])
eid, err := strconv.Atoi(keys({{ $receiver }}.{{ $e.BuilderField }})[0])
if err != nil {
return nil, err
}
{{- else }}
eid := keys({{ $receiver }}.{{ $e.StructField }})[0]
eid := keys({{ $receiver }}.{{ $e.BuilderField }})[0]
{{- end }}
{{- if $e.IsInverse }}
query, args := sql.Update({{ $.Package }}.{{ $e.TableConstant }}).
@@ -141,8 +141,8 @@ func ({{ $receiver }} *{{ $builder }}) sqlSave(ctx context.Context) (*{{ $.Name
if err != nil {
return nil, rollback(tx, err)
}
if int(affected) < len({{ $receiver }}.{{ $e.StructField }}) {
return nil, rollback(tx, &ErrConstraintFailed{msg: fmt.Sprintf("one of \"{{ $e.Name }}\" %v already connected to a different \"{{ $.Name }}\"", keys({{ $receiver }}.{{ $e.StructField }}))})
if int(affected) < len({{ $receiver }}.{{ $e.BuilderField }}) {
return nil, rollback(tx, &ErrConstraintFailed{msg: fmt.Sprintf("one of \"{{ $e.Name }}\" %v already connected to a different \"{{ $.Name }}\"", keys({{ $receiver }}.{{ $e.BuilderField }}))})
}
{{- end }}
}