sqlboiler/templates/relationship_to_one_setops.tpl

60 lines
2.1 KiB
Smarty
Raw Normal View History

2016-08-26 08:50:45 +02:00
{{- define "relationship_to_one_setops_helper" -}}
{{- $varNameSingular := .ForeignKey.ForeignTable | singular | camelCase}}
2016-08-26 08:50:45 +02:00
// Set{{.Function.Name}} of the {{.ForeignKey.Table | singular}} to the related item.
// Sets R.{{.Function.Name}} to related.
2016-08-27 07:04:49 +02:00
func ({{.Function.Receiver}} *{{.LocalTable.NameGo}}) Set{{.Function.Name}}(exec boil.Executor, insert bool, related *{{.ForeignTable.NameGo}}) error {
2016-08-27 07:43:50 +02:00
var err error
if insert {
if err = related.Insert(exec); err != nil {
2016-08-27 21:13:15 +02:00
return errors.Wrap(err, "failed to insert into foreign table")
2016-08-27 07:43:50 +02:00
}
}
oldVal := {{.Function.Receiver}}.{{.LocalTable.ColumnNameGo}}
{{.Function.Receiver}}.{{.Function.LocalAssignment}} = related.{{.Function.ForeignAssignment}}
if err = {{.Function.Receiver}}.Update(exec, "{{.ForeignKey.Column}}"); err != nil {
{{.Function.Receiver}}.{{.LocalTable.ColumnNameGo}} = oldVal
2016-08-27 21:13:15 +02:00
return errors.Wrap(err, "failed to update local table")
2016-08-27 07:43:50 +02:00
}
if {{.Function.Receiver}}.R == nil {
{{.Function.Receiver}}.R = &{{.LocalTable.NameGo}}R{
{{.Function.Name}}: related,
}
} else {
{{.Function.Receiver}}.R.{{.Function.Name}} = related
}
{{if .ForeignKey.Nullable}}
{{.Function.Receiver}}.{{.LocalTable.ColumnNameGo}}.Valid = true
{{end -}}
2016-08-27 07:43:50 +02:00
return nil
2016-08-26 08:50:45 +02:00
}
{{- if .ForeignKey.Nullable}}
2016-08-26 08:50:45 +02:00
// Remove{{.Function.Name}} relationship.
// Sets R.{{.Function.Name}} to nil.
2016-08-27 07:04:49 +02:00
func ({{.Function.Receiver}} *{{.LocalTable.NameGo}}) Remove{{.Function.Name}}(exec boil.Executor) error {
var err error
{{.Function.Receiver}}.{{.LocalTable.ColumnNameGo}}.Valid = false
if err = {{.Function.Receiver}}.Update(exec, "{{.ForeignKey.Column}}"); err != nil {
{{.Function.Receiver}}.{{.LocalTable.ColumnNameGo}}.Valid = true
2016-08-27 21:13:15 +02:00
return errors.Wrap(err, "failed to update local table")
}
{{.Function.Receiver}}.R.{{.Function.Name}} = nil
2016-08-26 08:50:45 +02:00
return nil
}
{{end -}}
{{- end -}}
2016-08-26 08:50:45 +02:00
{{- if .Table.IsJoinTable -}}
{{- else -}}
{{- $dot := . -}}
{{- range .Table.FKeys -}}
{{- $rel := textsFromForeignKey $dot.PkgName $dot.Tables $dot.Table . -}}
{{- template "relationship_to_one_setops_helper" $rel -}}
{{- end -}}
{{- end -}}