Making interface slightly more generic for create and update

This commit is contained in:
Matthew Dillon 2015-10-13 12:43:00 -07:00
parent b87077a1df
commit 299cd718b9

View file

@ -2,18 +2,19 @@ package models
import "github.com/thermokarst/bactdb/Godeps/_workspace/src/github.com/jmoiron/modl" import "github.com/thermokarst/bactdb/Godeps/_workspace/src/github.com/jmoiron/modl"
type updater interface { type base interface {
PreCreate(modl.SqlExecutor) error
PreUpdate(modl.SqlExecutor) error PreUpdate(modl.SqlExecutor) error
UpdateError() error UpdateError() error
} }
func Update(u updater) error { func Update(b base) error {
count, err := DBH.Update(u) count, err := DBH.Update(b)
if err != nil { if err != nil {
return err return err
} }
if count != 1 { if count != 1 {
return u.UpdateError() return b.UpdateError()
} }
return nil return nil
} }