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

Save cached model structs globally

Browse Source
This commit is contained in:
Jinzhu 2015-03-11 18:33:50 +08:00
parent b0cf9a9e4b
commit df33db6ff9
2 changed files with 12 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
main.go
View File

@ -23,7 +23,6 @@ type DB struct {
Value interface{}
Error error
RowsAffected int64
ModelStructs map[reflect.Type]*ModelStruct
callback *callback
db sqlCommon
parent *DB
@ -69,7 +68,6 @@ func Open(dialect string, args ...interface{}) (DB, error) {
source: source,
values: map[string]interface{}{},
db: dbSql,
ModelStructs: map[reflect.Type]*ModelStruct{},
}
db.parent = &db
}
@ -125,7 +123,7 @@ func (s *DB) LogMode(enable bool) *DB {
}
func (s *DB) SingularTable(enable bool) {
s.parent.ModelStructs = map[reflect.Type]*ModelStruct{}
modelStructs = map[reflect.Type]*ModelStruct{}
s.parent.singularTable = enable
}

10
model_struct.go
View File

@ -11,6 +11,8 @@ import (
"time"
)
var modelStructs = map[reflect.Type]*ModelStruct{}
type ModelStruct struct {
PrimaryFields []*StructField
StructFields []*StructField
@ -82,11 +84,9 @@ func (scope *Scope) GetModelStruct() *ModelStruct {
scopeType = scopeType.Elem()
}
if scope.db != nil {
if value, ok := scope.db.parent.ModelStructs[scopeType]; ok {
if value, ok := modelStructs[scopeType]; ok {
return value
}
}
modelStruct.ModelType = scopeType
if scopeType.Kind() != reflect.Struct {
@ -284,9 +284,7 @@ func (scope *Scope) GetModelStruct() *ModelStruct {
modelStruct.StructFields = append(modelStruct.StructFields, field)
}
if scope.db != nil {
scope.db.parent.ModelStructs[scopeType] = &modelStruct
}
modelStructs[scopeType] = &modelStruct
return &modelStruct
}