sqlboiler/templates/14_upsert.tpl

86 lines
3.1 KiB
Smarty
Raw Normal View History

2016-08-04 05:11:45 +02:00
{{- $tableNameSingular := .Table.Name | singular | titleCase -}}
{{- $varNameSingular := .Table.Name | singular | camelCase -}}
// UpsertG attempts an insert, and does an update or ignore on conflict.
2016-09-01 03:12:46 +02:00
func (o *{{$tableNameSingular}}) UpsertG(updateOnConflict bool, conflictColumns []string, updateColumns []string, whitelist ...string) error {
return o.Upsert(boil.GetDB(), updateOnConflict, conflictColumns, updateColumns, whitelist...)
2016-08-04 05:11:45 +02:00
}
// UpsertGP attempts an insert, and does an update or ignore on conflict. Panics on error.
2016-09-01 03:12:46 +02:00
func (o *{{$tableNameSingular}}) UpsertGP(updateOnConflict bool, conflictColumns []string, updateColumns []string, whitelist ...string) {
if err := o.Upsert(boil.GetDB(), updateOnConflict, conflictColumns, updateColumns, whitelist...); err != nil {
2016-08-04 05:11:45 +02:00
panic(boil.WrapErr(err))
}
}
// UpsertP attempts an insert using an executor, and does an update or ignore on conflict.
// UpsertP panics on error.
2016-09-01 03:12:46 +02:00
func (o *{{$tableNameSingular}}) UpsertP(exec boil.Executor, updateOnConflict bool, conflictColumns []string, updateColumns []string, whitelist ...string) {
if err := o.Upsert(exec, updateOnConflict, conflictColumns, updateColumns, whitelist...); err != nil {
panic(boil.WrapErr(err))
}
}
2016-08-04 05:11:45 +02:00
// Upsert attempts an insert using an executor, and does an update or ignore on conflict.
func (o *{{$tableNameSingular}}) Upsert(exec boil.Executor, updateOnConflict bool, conflictColumns []string, updateColumns []string, whitelist ...string) error {
2016-08-04 05:11:45 +02:00
if o == nil {
return errors.New("{{.PkgName}}: no {{.Table.Name}} provided for upsert")
}
{{- template "timestamp_upsert_helper" . }}
2016-08-29 14:38:19 +02:00
{{if not .NoHooks -}}
if err := o.doBeforeUpsertHooks(exec); err != nil {
return err
}
{{- end}}
var err error
var ret []string
whitelist, ret = strmangle.InsertColumnSet(
{{$varNameSingular}}Columns,
{{$varNameSingular}}ColumnsWithDefault,
{{$varNameSingular}}ColumnsWithoutDefault,
2016-09-03 21:47:27 +02:00
boil.NonZeroDefaultSet({{$varNameSingular}}ColumnsWithDefault, o),
whitelist,
)
update := strmangle.UpdateColumnSet(
{{$varNameSingular}}Columns,
{{$varNameSingular}}PrimaryKeyColumns,
updateColumns,
)
conflict := conflictColumns
if len(conflict) == 0 {
conflict = make([]string, len({{$varNameSingular}}PrimaryKeyColumns))
copy(conflict, {{$varNameSingular}}PrimaryKeyColumns)
}
2016-09-01 03:20:16 +02:00
query := boil.BuildUpsertQuery("{{.Table.Name}}", updateOnConflict, ret, update, conflict, whitelist)
2016-08-04 05:11:45 +02:00
2016-08-13 18:58:18 +02:00
if boil.DebugMode {
fmt.Fprintln(boil.DebugWriter, query)
2016-09-02 09:09:42 +02:00
fmt.Fprintln(boil.DebugWriter, boil.GetStructValues(o, whitelist...))
2016-08-13 18:58:18 +02:00
}
{{- if .UseLastInsertID}}
return errors.New("don't know how to do this yet")
{{- else}}
if len(ret) != 0 {
2016-09-02 09:09:42 +02:00
err = exec.QueryRow(query, boil.GetStructValues(o, whitelist...)...).Scan(boil.GetStructPointers(o, ret...)...)
2016-08-04 05:11:45 +02:00
} else {
2016-09-02 09:09:42 +02:00
_, err = exec.Exec(query, boil.GetStructValues(o, whitelist...)...)
2016-08-04 05:11:45 +02:00
}
{{- end}}
2016-08-04 05:11:45 +02:00
if err != nil {
return errors.Wrap(err, "{{.PkgName}}: unable to upsert for {{.Table.Name}}")
2016-08-04 05:11:45 +02:00
}
2016-08-29 14:38:19 +02:00
{{if not .NoHooks -}}
if err := o.doAfterUpsertHooks(exec); err != nil {
2016-08-04 05:11:45 +02:00
return err
}
2016-08-28 12:48:50 +02:00
{{- end}}
2016-08-04 05:11:45 +02:00
return nil
}