LBRYCommunity/sqlboiler
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity

Add delete and insert hooks to templates

Browse source
This commit is contained in:
Patrick O'brien 2016-08-28 22:16:08 +10:00
parent d9ec68b76a
commit 677d45cef4
3 changed files with 46 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
templates/10_insert.tpl
View file

@ -31,6 +31,13 @@ func (o *{{$tableNameSingular}}) Insert(exec boil.Executor, whitelist ... string
return errors.New("{{.PkgName}}: no {{.Table.Name}} provided for insertion")
}
var err error
{{if eq .NoHooks false -}}
if err := o.doBeforeInsertHooks(); err != nil {
return err
}
{{- end}}
wl, returnColumns := strmangle.InsertColumnSet(
{{$varNameSingular}}Columns,
{{$varNameSingular}}ColumnsWithDefault,
@ -39,13 +46,6 @@ func (o *{{$tableNameSingular}}) Insert(exec boil.Executor, whitelist ... string
whitelist,
)
var err error
{{if eq .NoHooks false -}}
if err := o.doBeforeInsertHooks(); err != nil {
return err
}
{{- end}}
ins := fmt.Sprintf(`INSERT INTO {{.Table.Name}} ("%s") VALUES (%s)`, strings.Join(wl, `","`), strmangle.Placeholders(len(wl), 1, 1))
{{if .UseLastInsertID}}

14
templates/12_upsert.tpl
View file

@ -26,6 +26,13 @@ func (o *{{$tableNameSingular}}) Upsert(exec boil.Executor, updateOnConflict boo
return errors.New("{{.PkgName}}: no {{.Table.Name}} provided for upsert")
}
{{if eq .NoHooks false -}}
if err := o.doBeforeUpsertHooks(); err != nil {
return err
}
{{- end}}
var err error
var ret []string
whitelist, ret = strmangle.InsertColumnSet(
{{$varNameSingular}}Columns,
@ -47,13 +54,6 @@ func (o *{{$tableNameSingular}}) Upsert(exec boil.Executor, updateOnConflict boo
query := generateUpsertQuery("{{.Table.Name}}", updateOnConflict, ret, update, conflict, whitelist)
var err error
{{if eq .NoHooks false -}}
if err := o.doBeforeUpsertHooks(); err != nil {
return err
}
{{- end}}
if boil.DebugMode {
fmt.Fprintln(boil.DebugWriter, query)
fmt.Fprintln(boil.DebugWriter, boil.GetStructValues(o, {{$varNameSingular}}TitleCases, whitelist...))

32
templates/13_delete.tpl
View file

@ -35,6 +35,12 @@ func (o *{{$tableNameSingular}}) Delete(exec boil.Executor) error {
return errors.New("{{.PkgName}}: no {{$tableNameSingular}} provided for delete")
}
{{if eq .NoHooks false -}}
if err := o.doBeforeDeleteHooks(); err != nil {
return err
}
{{- end}}
args := o.inPrimaryKeyArgs()
sql := `DELETE FROM {{.Table.Name}} WHERE {{whereClause 1 .Table.PKey.Columns}}`
@ -49,6 +55,12 @@ func (o *{{$tableNameSingular}}) Delete(exec boil.Executor) error {
return errors.Wrap(err, "{{.PkgName}}: unable to delete from {{.Table.Name}}")
}
{{if eq .NoHooks false -}}
if err := o.doAfterDeleteHooks(); err != nil {
return err
}
{{- end}}
return nil
}
@ -107,6 +119,16 @@ func (o {{$tableNameSingular}}Slice) DeleteAll(exec boil.Executor) error {
return nil
}
{{if eq .NoHooks false -}}
if len({{$varNameSingular}}BeforeDeleteHooks) != 0 {
for _, obj := range o {
if err := obj.doBeforeDeleteHooks(); err != nil {
return err
}
}
}
{{- end}}
args := o.inPrimaryKeyArgs()
sql := fmt.Sprintf(
@ -125,5 +147,15 @@ func (o {{$tableNameSingular}}Slice) DeleteAll(exec boil.Executor) error {
return errors.Wrap(err, "{{.PkgName}}: unable to delete all from {{$varNameSingular}} slice")
}
{{if eq .NoHooks false -}}
if len({{$varNameSingular}}AfterDeleteHooks) != 0 {
for _, obj := range o {
if err := obj.doAfterDeleteHooks(); err != nil {
return err
}
}
}
{{- end}}
return nil
}