tablet/gorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update model_struct.go

Browse Source 
处理在同时定义gorm.DefaultTableNameHandler和Model.TableName时,db.Find()与db.Where()函数获取表名结果不一致的问题
This commit is contained in:
hot1232 2018-11-20 18:05:26 +08:00 committed by GitHub
parent 472c70caa4
commit 859ebcfaa6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
model_struct.go
View File

@ -13,9 +13,7 @@ import (
) )
// DefaultTableNameHandler default table name handler // DefaultTableNameHandler default table name handler
var DefaultTableNameHandler = func(db *DB, defaultTableName string) string { var DefaultTableNameHandler = func(db *DB, defaultTableName string) string
return defaultTableName
}
var modelStructsMap sync.Map var modelStructsMap sync.Map
@ -46,7 +44,9 @@ func (s *ModelStruct) TableName(db *DB) string {
s.defaultTableName = tableName s.defaultTableName = tableName
} }
} }
if nil == DefaultTableNameHandler {
return s.defaultTableName
}
return DefaultTableNameHandler(db, s.defaultTableName) return DefaultTableNameHandler(db, s.defaultTableName)
} }