diff --git a/example/gen_admin.go b/example/gen_admin.go index 2a436d7..69256c5 100644 --- a/example/gen_admin.go +++ b/example/gen_admin.go @@ -40,7 +40,7 @@ func (t *Admin) Delete(db *gorm.DB) (err error) { // Updates update record func (t *Admin) Updates(db *gorm.DB, m map[string]interface{}) (err error) { - if err = db.Model(&Admin{}).Where("id = ?", t.ID).Updates(m).Error; err != nil { + if err = db.Model(t).Updates(m).Error; err != nil { err = ErrUpdateAdmin return diff --git a/example/gen_user.go b/example/gen_user.go index ca04379..5119627 100644 --- a/example/gen_user.go +++ b/example/gen_user.go @@ -40,7 +40,7 @@ func (t *User) Delete(db *gorm.DB) (err error) { // Updates update record func (t *User) Updates(db *gorm.DB, m map[string]interface{}) (err error) { - if err = db.Model(&User{}).Where("id = ?", t.ID).Updates(m).Error; err != nil { + if err = db.Model(t).Updates(m).Error; err != nil { err = ErrUpdateUser return diff --git a/template.go b/template.go index bf19fed..affdb8e 100644 --- a/template.go +++ b/template.go @@ -69,7 +69,7 @@ import ( // Updates update record func (t *{{.StructName}}) Updates(db *gorm.DB, m map[string]interface{})(err error) { - if err = db.Model(&{{.StructName}}{}).Where("id = ?",t.ID).Updates(m).Error;err!=nil{ + if err = db.Model(t).Updates(m).Error;err!=nil{ {{if $LogName}} {{ $LogName}}.Errorln(err) {{end}} {{if $TransformErr}} err = ErrUpdate{{.StructName}} {{end}} return