sqlboiler/templates/09_relationship_to_many_setops.tpl

{{- if .Table.IsJoinTable -}}
{{- else -}}
  {{- $dot := . -}}
  {{- $table := .Table -}}
  {{- range .Table.ToManyRelationships -}}
    {{- $varNameSingular := .ForeignTable | singular | camelCase -}}
    {{- if (and .ForeignColumnUnique (not .ToJoinTable)) -}}
{{- template "relationship_to_one_setops_helper" (textsFromOneToOneRelationship $dot.PkgName $dot.Tables $table .) -}}
    {{- else -}}
    {{- $rel := textsFromRelationship $dot.Tables $table . -}}
    {{- $localNameSingular := .Table | singular | camelCase -}}
    {{- $foreignNameSingular := .ForeignTable | singular | camelCase}}

// Add{{$rel.Function.Name}} adds the given related objects to the existing relationships
// of the {{$table.Name | singular}}, optionally inserting them as new records.
// Appends related to {{$rel.Function.Receiver}}.R.{{$rel.Function.Name}}.
// Sets related.R.{{$rel.Function.ForeignName}} appropriately.
func ({{$rel.Function.Receiver}} *{{$rel.LocalTable.NameGo}}) Add{{$rel.Function.Name}}(exec boil.Executor, insert bool, related ...*{{$rel.ForeignTable.NameGo}}) error {
  var err error
  for _, rel := range related {
    {{if not .ToJoinTable -}}
    rel.{{$rel.Function.ForeignAssignment}} = {{$rel.Function.Receiver}}.{{$rel.Function.LocalAssignment}}
      {{if .ForeignColumnNullable -}}
    rel.{{$rel.ForeignTable.ColumnNameGo}}.Valid = true
      {{end -}}
    {{end -}}
    if insert {
      if err = rel.Insert(exec); err != nil {
        return errors.Wrap(err, "failed to insert into foreign table")
      }
    }{{if not .ToJoinTable}} else {
      if err = rel.Update(exec, "{{.ForeignColumn}}"); err != nil {
        return errors.Wrap(err, "failed to update foreign table")
      }
    }{{end -}}
  }

  {{if .ToJoinTable -}}
  for _, rel := range related {
    query := `insert into "{{.JoinTable}}" ({{.JoinLocalColumn}}, {{.JoinForeignColumn}}) values ($1, $2)`
    values := []interface{}{{"{"}}{{$rel.Function.Receiver}}.{{$rel.LocalTable.ColumnNameGo}}, rel.{{$rel.ForeignTable.ColumnNameGo}}}

    if boil.DebugMode {
      fmt.Fprintln(boil.DebugWriter, query)
      fmt.Fprintln(boil.DebugWriter, values)
    }

    _, err = exec.Exec(query, values...)
    if err != nil {
      return errors.Wrap(err, "failed to insert into join table")
    }
  }
  {{end -}}

  if {{$rel.Function.Receiver}}.R == nil {
    {{$rel.Function.Receiver}}.R = &{{$localNameSingular}}R{
      {{$rel.Function.Name}}: related,
    }
  } else {
    {{$rel.Function.Receiver}}.R.{{$rel.Function.Name}} = append({{$rel.Function.Receiver}}.R.{{$rel.Function.Name}}, related...)
  }

  {{if .ToJoinTable -}}
  for _, rel := range related {
    if rel.R == nil {
      rel.R = &{{$foreignNameSingular}}R{
        {{$rel.Function.ForeignName}}: {{$rel.LocalTable.NameGo}}Slice{{"{"}}{{$rel.Function.Receiver}}{{"}"}},
      }
    } else {
      rel.R.{{$rel.Function.ForeignName}} = append(rel.R.{{$rel.Function.ForeignName}}, {{$rel.Function.Receiver}})
    }
  }
  {{else -}}
  for _, rel := range related {
    if rel.R == nil {
      rel.R = &{{$foreignNameSingular}}R{
        {{$rel.Function.ForeignName}}: {{$rel.Function.Receiver}},
      }
    } else {
      rel.R.{{$rel.Function.ForeignName}} = {{$rel.Function.Receiver}}
    }
  }
  {{end -}}

  return nil
}
{{- if (or .ForeignColumnNullable .ToJoinTable)}}

// Set{{$rel.Function.Name}} removes all previously related items of the
// {{$table.Name | singular}} replacing them completely with the passed
// in related items, optionally inserting them as new records.
// Sets {{$rel.Function.Receiver}}.R.{{$rel.Function.ForeignName}}'s {{$rel.Function.Name}} accordingly.
// Replaces {{$rel.Function.Receiver}}.R.{{$rel.Function.Name}} with related.
// Sets related.R.{{$rel.Function.ForeignName}}'s {{$rel.Function.Name}} accordingly.
func ({{$rel.Function.Receiver}} *{{$rel.LocalTable.NameGo}}) Set{{$rel.Function.Name}}(exec boil.Executor, insert bool, related ...*{{$rel.ForeignTable.NameGo}}) error {
  {{if .ToJoinTable -}}
  query := `delete from "{{.JoinTable}}" where "{{.JoinLocalColumn}}" = $1`
  values := []interface{}{{"{"}}{{$rel.Function.Receiver}}.{{$rel.LocalTable.ColumnNameGo}}}
  {{else -}}
  query := `update "{{.ForeignTable}}" set "{{.ForeignColumn}}" = null where "{{.ForeignColumn}}" = $1`
  values := []interface{}{{"{"}}{{$rel.Function.Receiver}}.{{$rel.LocalTable.ColumnNameGo}}}
  {{end -}}
  if boil.DebugMode {
    fmt.Fprintln(boil.DebugWriter, query)
    fmt.Fprintln(boil.DebugWriter, values)
  }

  _, err := exec.Exec(query, values...)
  if err != nil {
    return errors.Wrap(err, "failed to remove relationships before set")
  }

  {{if .ToJoinTable -}}
  remove{{$rel.LocalTable.NameGo}}From{{$rel.ForeignTable.NameGo}}Slice({{$rel.Function.Receiver}}, related)
  {{$rel.Function.Receiver}}.R.{{$rel.Function.Name}} = nil
  {{else -}}
  if {{$rel.Function.Receiver}}.R != nil {
    for _, rel := range {{$rel.Function.Receiver}}.R.{{$rel.Function.Name}} {
      rel.{{$rel.ForeignTable.ColumnNameGo}}.Valid = false
      if rel.R == nil {
        continue
      }

      rel.R.{{$rel.Function.ForeignName}} = nil
    }

    {{$rel.Function.Receiver}}.R.{{$rel.Function.Name}} = nil
  }
  {{end -}}

  return {{$rel.Function.Receiver}}.Add{{$rel.Function.Name}}(exec, insert, related...)
}

// Remove{{$rel.Function.Name}} relationships from objects passed in.
// Removes related items from R.{{$rel.Function.Name}} (uses pointer comparison, removal does not keep order)
// Sets related.R.{{$rel.Function.ForeignName}}.
func ({{$rel.Function.Receiver}} *{{$rel.LocalTable.NameGo}}) Remove{{$rel.Function.Name}}(exec boil.Executor, related ...*{{$rel.ForeignTable.NameGo}}) error {
  var err error
  {{if .ToJoinTable -}}
  query := fmt.Sprintf(
    `delete from "{{.JoinTable}}" where "{{.JoinLocalColumn}}" = $1 and "{{.JoinForeignColumn}}" in (%s)`,
    strmangle.Placeholders(len(related), 1, 1),
  )
  values := []interface{}{{"{"}}{{$rel.Function.Receiver}}.{{$rel.LocalTable.ColumnNameGo}}}

  if boil.DebugMode {
    fmt.Fprintln(boil.DebugWriter, query)
    fmt.Fprintln(boil.DebugWriter, values)
  }

  _, err = exec.Exec(query, values...)
  if err != nil {
    return errors.Wrap(err, "failed to remove relationships before set")
  }
  {{else -}}
  for _, rel := range related {
    rel.{{$rel.ForeignTable.ColumnNameGo}}.Valid = false
    {{if not .ToJoinTable -}}
    if rel.R != nil {
      rel.R.{{$rel.Function.ForeignName}} = nil
    }
    {{end -}}
    if err = rel.Update(exec, "{{.ForeignColumn}}"); err != nil {
      return err
    }
  }
  {{end -}}

  {{if .ToJoinTable -}}
  remove{{$rel.LocalTable.NameGo}}From{{$rel.ForeignTable.NameGo}}Slice({{$rel.Function.Receiver}}, related)
  {{end -}}
  if {{$rel.Function.Receiver}}.R == nil {
    return nil
  }

  for _, rel := range related {
    for i, ri := range {{$rel.Function.Receiver}}.R.{{$rel.Function.Name}} {
      if rel != ri {
        continue
      }

      ln := len({{$rel.Function.Receiver}}.R.{{$rel.Function.Name}})
      if ln > 1 && i < ln-1 {
        {{$rel.Function.Receiver}}.R.{{$rel.Function.Name}}[i] = {{$rel.Function.Receiver}}.R.{{$rel.Function.Name}}[ln-1]
      }
      {{$rel.Function.Receiver}}.R.{{$rel.Function.Name}} = {{$rel.Function.Receiver}}.R.{{$rel.Function.Name}}[:ln-1]
      break
    }
  }

  return nil
}

{{if .ToJoinTable -}}
func remove{{$rel.LocalTable.NameGo}}From{{$rel.ForeignTable.NameGo}}Slice({{$rel.Function.Receiver}} *{{$rel.LocalTable.NameGo}}, related []*{{$rel.ForeignTable.NameGo}}) {
  for _, rel := range related {
    if rel.R == nil {
      continue
    }
    for i, ri := range rel.R.{{$rel.Function.ForeignName}} {
      if {{$rel.Function.Receiver}}.{{$rel.Function.LocalAssignment}} != ri.{{$rel.Function.LocalAssignment}} {
        continue
      }

      ln := len(rel.R.{{$rel.Function.ForeignName}})
      if ln > 1 && i < ln-1 {
        rel.R.{{$rel.Function.ForeignName}}[i] = rel.R.{{$rel.Function.ForeignName}}[ln-1]
      }
      rel.R.{{$rel.Function.ForeignName}} = rel.R.{{$rel.Function.ForeignName}}[:ln-1]
      break
    }
  }
}
{{end -}}{{- /* if join table */ -}}
{{- end -}}{{- /* if nullable foreign key */ -}}
{{- end -}}{{- /* if unique foreign key */ -}}
{{- end -}}{{- /* range relationships */ -}}
{{- end -}}{{- /* outer if join table */ -}}