mirror of
https://github.com/ent/ent.git
synced 2026-05-24 09:31:56 +03:00
2d962d5cba
Reviewed By: idoshveki Differential Revision: D16599899 fbshipit-source-id: bcab65e09645409f8a822e2eebda1aac0f804c7c
70 lines
2.9 KiB
Cheetah
70 lines
2.9 KiB
Cheetah
{{ define "setter" }}
|
|
{{ $builder := pascal $.Scope.Builder }}
|
|
{{ $receiver := receiver $builder }}
|
|
|
|
{{ range $_, $f := $.Fields }}
|
|
{{ $p := lower (printf "%.1s" $f.Type) }}
|
|
{{ $func := print "Set" (pascal $f.Name) }}
|
|
// {{ $func }} sets the {{ $f.Name }} field.
|
|
func ({{ $receiver }} *{{ $builder }}) {{ $func }}({{ $p }} {{ $f.Type }}) *{{ $builder }} {
|
|
{{ $receiver }}.{{ $f.StructField }} = &{{ $p }}
|
|
return {{ $receiver }}
|
|
}
|
|
{{ if or $f.Optional $f.Nullable $f.HasDefault }}
|
|
{{ $nillableFunc := print "SetNillable" (pascal $f.Name) }}
|
|
// {{ $nillableFunc }} sets the {{ $f.Name }} field if the given value is not nil.
|
|
func ({{ $receiver }} *{{ $builder }}) {{ $nillableFunc }}({{ $p }} *{{ $f.Type }}) *{{ $builder }} {
|
|
if {{ $p }} != nil {
|
|
{{ $receiver }}.{{ $func }}(*{{ $p }})
|
|
}
|
|
return {{ $receiver }}
|
|
}
|
|
{{ end }}
|
|
{{ end }}
|
|
|
|
{{ range $_, $e := $.Edges }}
|
|
{{ $op := "add" }}{{ if $e.Unique }}{{ $op = "set" }}{{ end }}
|
|
{{ $idsFunc := print (pascal $op) (singular $e.Name | pascal) "IDs" }}{{ if $e.Unique }}{{ $idsFunc = print (pascal $op) (pascal $e.Name) "ID" }}{{ end }}
|
|
// {{ $idsFunc }} {{ $op }}s the {{ $e.Name }} edge to {{ $e.Type.Name }} by id{{ if not $e.Unique }}s{{ end }}.
|
|
func ({{ $receiver }} *{{ $builder }}) {{ $idsFunc }}({{ if $e.Unique }}id{{ else }}ids ...{{ end }} {{ $.ID.Type }}) *{{ $builder }} {
|
|
if {{ $receiver }}.{{ $e.StructField }} == nil {
|
|
{{ $receiver }}.{{ $e.StructField }} = make(map[{{ $.ID.Type }}]struct{})
|
|
}
|
|
{{ if $e.Unique -}}
|
|
{{ $receiver }}.{{ $e.StructField }}[id] = struct{}{}
|
|
{{- else -}}
|
|
for i := range ids {
|
|
{{ $receiver }}.{{ $e.StructField }}[ids[i]] = struct{}{}
|
|
}
|
|
{{- end }}
|
|
return {{ $receiver }}
|
|
}
|
|
{{ if and $e.Unique $e.Optional }}
|
|
{{ $nillableIDsFunc := print "SetNillable" (pascal $e.Name) "ID" }}
|
|
// {{ $nillableIDsFunc }} sets the {{ $e.Name }} edge to {{ $e.Type.Name }} by id if the given value is not nil.
|
|
func ({{ $receiver }} *{{ $builder }}) {{ $nillableIDsFunc }}(id *{{ $.ID.Type }}) *{{ $builder }} {
|
|
if id != nil {
|
|
{{ $receiver}} = {{ $receiver }}.{{ $idsFunc }}(*id)
|
|
}
|
|
return {{ $receiver }}
|
|
}
|
|
{{ end }}
|
|
{{ $p := lower (printf "%.1s" $e.Type.Name) }}
|
|
{{ if eq $p $receiver }} {{ $p = "v" }} {{ end }}
|
|
{{ $func := print (pascal $op) (pascal $e.Name) }}
|
|
// {{ $func }} {{ $op }}s the {{ $e.Name }} edge{{if not $e.Unique}}s{{ end }} to {{ $e.Type.Name }}.
|
|
func ({{ $receiver }} *{{ $builder }}) {{ $func }}({{ $p }} {{ if not $e.Unique }}...{{ end }}*{{ $e.Type.Name}}) *{{ $builder }} {
|
|
{{ if $e.Unique -}}
|
|
return {{ $receiver }}.{{ $idsFunc }}({{ $p }}.ID)
|
|
{{- else -}}
|
|
ids := make([]{{ $.ID.Type }}, len({{ $p }}))
|
|
{{ $i := "i" }}{{ if eq $i $p }}{{ $i = "j" }}{{ end -}}
|
|
for {{ $i }} := range {{ $p }} {
|
|
ids[{{ $i }}] = {{ $p }}[{{ $i }}].ID
|
|
}
|
|
return {{ $receiver }}.{{ $idsFunc }}(ids...)
|
|
{{- end }}
|
|
}
|
|
{{ end }}
|
|
{{ end }}
|