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

dialect/sql/sqlgraph: add update spec builder (#3301)

Browse Source
This commit is contained in:
Ariel Mashraki
2023-02-06 07:13:06 +02:00
committed by GitHub
parent ca7408e81c
commit 0e6ef32505
125 changed files with 268 additions and 2529 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
entc/gen/template/dialect/sql/update.tmpl
View File

@@ -37,27 +37,14 @@ func ({{ $receiver }} *{{ $builder }}) sqlSave(ctx context.Context) ({{ $ret }}
return {{ $ret }}, err
}
{{- end }}
_spec := &sqlgraph.UpdateSpec{
Node: &sqlgraph.NodeSpec{
Table: {{ $.Package }}.Table,
Columns: {{ $.Package }}.Columns,
{{- if $.HasOneFieldID }}
ID: &sqlgraph.FieldSpec{
Type: field.{{ $.ID.Type.ConstName }},
Column: {{ $.Package }}.{{ $.ID.Constant }},
},
{{- else }}
CompositeID: []*sqlgraph.FieldSpec{
{{- range $id := $.EdgeSchema.ID }}
{
Type: field.{{ $id.Type.ConstName }},
Column: {{ $.Package }}.{{ $id.Constant }},
},
{{- end }}
},
{{- end }}
},
}
_spec := sqlgraph.NewUpdateSpec({{ $.Package }}.Table, {{ $.Package }}.Columns,
{{- if $.HasOneFieldID -}}
sqlgraph.NewFieldSpec({{ $.Package }}.{{ $.ID.Constant }}, field.{{ $.ID.Type.ConstName }})
{{- else -}}
{{- range $id := $.EdgeSchema.ID -}}
sqlgraph.NewFieldSpec({{ $.Package }}.{{ $id.Constant }}, field.{{ $id.Type.ConstName }}),
{{- end -}}
{{- end }})
{{- if $one }}
{{- if $.HasOneFieldID }}
id, ok := {{ $mutation }}.{{ $.ID.MutationGet }}()