sqlboiler/templates/23_merge.tpl

101 lines
2.8 KiB
Smarty
Raw Normal View History

2017-05-10 21:29:02 +02:00
{{- $tableNamePlural := .Table.Name | plural | titleCase -}}
{{- $tableNameSingular := .Table.Name | singular | titleCase -}}
{{- if .Table.IsJoinTable -}}
{{- else -}}
{{- $dot := . }}
// Merge combines two {{$tableNamePlural}} into one. The primary record will be kept, and the secondary will be deleted.
func Merge{{$tableNamePlural}}(exec boil.Executor, primaryID uint64, secondaryID uint64) error {
txdb, ok := exec.(boil.Beginner)
if !ok {
return errors.New("database does not support transactions")
}
tx, txErr := txdb.Begin()
if txErr != nil {
return txErr
}
primary, err := Find{{$tableNameSingular}}(tx, primaryID)
if err != nil {
tx.Rollback()
return err
}
if primary == nil {
return errors.New("Primary {{$tableNameSingular}} not found")
}
secondary, err := Find{{$tableNameSingular}}(tx, secondaryID)
if err != nil {
tx.Rollback()
return err
}
if secondary == nil {
return errors.New("Secondary {{$tableNameSingular}} not found")
}
foreignKeys := []foreignKey{
2017-05-10 21:29:02 +02:00
{{- range .Tables -}}
{{- range .FKeys -}}
{{- if eq $dot.Table.Name .ForeignTable }}
{foreignTable: "{{.Table}}", foreignColumn: "{{.Column}}"},
2017-05-10 21:29:02 +02:00
{{- end -}}
{{- end -}}
{{- end }}
}
conflictingKeys := []conflictingUniqueKey{
{{- range .Tables -}}
{{- $table := . -}}
{{- range .FKeys -}}
{{- $fk := . -}}
{{- if eq $dot.Table.Name .ForeignTable -}}
{{- range $table.UKeys -}}
{{- if setInclude $fk.Column .Columns }}
{table: "{{$fk.Table}}", objectIdColumn: "{{$fk.Column}}", columns: []string{`{{ .Columns | join "`,`" }}`}},
{{- end -}}
{{- end -}}
{{- end -}}
{{- end -}}
{{- end }}
}
err = mergeModels(tx, primaryID, secondaryID, foreignKeys, conflictingKeys)
2017-05-10 21:29:02 +02:00
if err != nil {
tx.Rollback()
return err
}
pr := reflect.ValueOf(primary)
sr := reflect.ValueOf(secondary)
// for any column thats null on the primary and not null on the secondary, copy from secondary to primary
for i := 0; i < sr.Elem().NumField(); i++ {
pf := pr.Elem().Field(i)
sf := sr.Elem().Field(i)
if sf.IsValid() {
if nullable, ok := sf.Interface().(Nullable); ok && !nullable.IsZero() && pf.Interface().(Nullable).IsZero() {
pf.Set(sf)
}
}
}
err = primary.Update(tx)
if err != nil {
tx.Rollback()
return errors.WithStack(err)
}
err = secondary.Delete(tx)
if err != nil {
tx.Rollback()
return errors.WithStack(err)
}
tx.Commit()
return nil
}
// Merge combines two {{$tableNamePlural}} into one. The primary record will be kept, and the secondary will be deleted.
func Merge{{$tableNamePlural}}G(primaryID uint64, secondaryID uint64) error {
return Merge{{$tableNamePlural}}(boil.GetDB(), primaryID, secondaryID)
}
{{- end -}}{{/* join table */}}