{{ define "setter" }} {{ $builder := pascal $.Scope.Builder }} {{ $receiver := receiver $builder }} {{ range $_, $f := $.Fields }} {{ $p := receiver $f.Type.String }} {{ $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 }} } {{/* avoid generting nillable setters for slices because the nil value for slice is valid */}} {{ if and (not $f.Type.Slice) (or $f.Optional $f.Nillable $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 }}