sqlboiler/templates/relationship_to_one_setops.tpl

{{- define "relationship_to_one_setops_helper" -}}
{{- $varNameSingular := .ForeignKey.ForeignTable | singular | camelCase}}

// Set{{.Function.Name}} of the {{.ForeignKey.Table | singular}} to the related item.
// Sets {{.Function.Receiver}}.R.{{.Function.Name}} to related.
// Adds {{.Function.Receiver}} to related.R.{{.Function.ForeignName}}.
func ({{.Function.Receiver}} *{{.LocalTable.NameGo}}) Set{{.Function.Name}}(exec boil.Executor, insert bool, related *{{.ForeignTable.NameGo}}) error {
  var err error
  if insert {
    if err = related.Insert(exec); err != nil {
      return errors.Wrap(err, "failed to insert into foreign table")
    }
  }

  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
    return errors.Wrap(err, "failed to update local table")
  }

  if {{.Function.Receiver}}.R == nil {
    {{.Function.Receiver}}.R = &{{.LocalTable.NameGo}}R{
      {{.Function.Name}}: related,
    }
  } else {
    {{.Function.Receiver}}.R.{{.Function.Name}} = related
  }

  {{if (or .ForeignKey.Unique .Function.OneToOne) -}}
  if related.R == nil {
    related.R = &{{.ForeignTable.NameGo}}R{
      {{.Function.ForeignName}}: {{.Function.Receiver}},
    }
  } else {
    related.R.{{.Function.ForeignName}} = {{.Function.Receiver}}
  }
  {{else -}}
  if related.R == nil {
    related.R = &{{.ForeignTable.NameGo}}R{
      {{.Function.ForeignName}}: {{.LocalTable.NameGo}}Slice{{"{"}}{{.Function.Receiver}}{{"}"}},
    }
  } else {
    related.R.{{.Function.ForeignName}} = append(related.R.{{.Function.ForeignName}}, {{.Function.Receiver}})
  }
  {{end -}}

  {{if .ForeignKey.Nullable}}
  {{.Function.Receiver}}.{{.LocalTable.ColumnNameGo}}.Valid = true
  {{end -}}
  return nil
}
{{- if .ForeignKey.Nullable}}

// Remove{{.Function.Name}} relationship.
// Sets {{.Function.Receiver}}.R.{{.Function.Name}} to nil.
// Removes {{.Function.Receiver}} from all passed in related items' relationships struct (Optional).
func ({{.Function.Receiver}} *{{.LocalTable.NameGo}}) Remove{{.Function.Name}}(exec boil.Executor, related ...*{{.ForeignTable.NameGo}}) 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
    return errors.Wrap(err, "failed to update local table")
  }

  for _, rel := range related {
    if rel.R == nil {
      continue
    }

    {{if .ForeignKey.Unique -}}
    rel.R.{{.Function.ForeignName}} = nil
    {{else -}}
    for i, ri := range rel.R.{{.Function.ForeignName}} {
      if {{.Function.Receiver}}.{{.Function.LocalAssignment}} != ri.{{.Function.LocalAssignment}} {
        continue
      }

      ln := len(rel.R.{{.Function.ForeignName}})
      if ln > 1 && i < ln-1 {
        rel.R.{{.Function.ForeignName}}[i], rel.R.{{.Function.ForeignName}}[ln-1] =
          rel.R.{{.Function.ForeignName}}[ln-1], rel.R.{{.Function.ForeignName}}[i]
      }
      rel.R.{{.Function.ForeignName}} = rel.R.{{.Function.ForeignName}}[:ln-1]
      break
    }
    {{end -}}
  }

  {{.Function.Receiver}}.R.{{.Function.Name}} = nil
  return nil
}
{{end -}}
{{- end -}}
{{- 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 -}}