re
/
gorm
forked from mirror/gorm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add SetTableNameHandler

This commit is contained in:
Jinzhu 2015-04-08 14:00:29 +08:00
parent b9a39be9c5
commit e2eef50fb4
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
main.go
View File

@ -487,3 +487,7 @@ func (s *DB) SetJoinTableHandler(source interface{}, column string, handler Join
}
}
}
func (s *DB) SetTableNameHandler(source interface{}, handler func(*DB) string) {
s.NewScope(source).GetModelStruct().TableName = handler
}

3
scope.go
View File

@ -247,8 +247,7 @@ func (scope *Scope) TableName() string {
}
if scope.GetModelStruct().TableName != nil {
scope.Search.tableName = scope.GetModelStruct().TableName(scope.db)
return scope.Search.tableName
return scope.GetModelStruct().TableName(scope.db)
}
scope.Err(errors.New("wrong table name"))