84 lines
2.2 KiB
Smarty
84 lines
2.2 KiB
Smarty
|
{{- $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")
|
||
|
|
}
|
||
|
|
|
||
|
|
relatedFields := map[string]string{
|
||
|
|
{{- range .Tables -}}
|
||
|
|
{{- range .FKeys -}}
|
||
|
|
{{- if eq $dot.Table.Name .ForeignTable }}
|
||
|
|
"{{.Table }}": "{{ .Column}}",
|
||
|
|
{{- end -}}
|
||
|
|
{{- end -}}
|
||
|
|
{{- end }}
|
||
|
|
}
|
||
|
|
err = mergeModels(tx, primaryID, secondaryID, relatedFields)
|
||
|
|
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 */}}
|