Merge pull request #1 from zsinx/master

change updates function of template
xjellyx · Aug 17, 2021 · d25f264 · d25f264
2 parents 0a46555 + 6f9c01a
commit d25f264
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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