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

optimize Parse

Browse Source
This commit is contained in:
daheige 2021-06-14 22:26:23 +08:00
parent b79bf9f463
commit aa2271c65e
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
schema/schema.go
View File

@ -71,6 +71,11 @@ type Tabler interface {
TableName() string TableName() string
} }
var callbackActions = []string{
"BeforeCreate", "AfterCreate", "BeforeUpdate", "AfterUpdate",
"BeforeSave", "AfterSave", "BeforeDelete", "AfterDelete", "AfterFind",
}
// Parse get data type from dialector // Parse get data type from dialector
func Parse(dest interface{}, cacheStore *sync.Map, namer Namer) (*Schema, error) { func Parse(dest interface{}, cacheStore *sync.Map, namer Namer) (*Schema, error) {
if dest == nil { if dest == nil {
@ -221,14 +226,13 @@ func Parse(dest interface{}, cacheStore *sync.Map, namer Namer) (*Schema, error)
} }
} }
callbacks := []string{"BeforeCreate", "AfterCreate", "BeforeUpdate", "AfterUpdate", "BeforeSave", "AfterSave", "BeforeDelete", "AfterDelete", "AfterFind"} for _, name := range callbackActions {
for _, name := range callbacks {
if methodValue := modelValue.MethodByName(name); methodValue.IsValid() { if methodValue := modelValue.MethodByName(name); methodValue.IsValid() {
switch methodValue.Type().String() { switch methodValue.Type().String() {
case "func(*gorm.DB) error": // TODO hack case "func(*gorm.DB) error": // TODO hack
reflect.Indirect(reflect.ValueOf(schema)).FieldByName(name).SetBool(true) reflect.Indirect(reflect.ValueOf(schema)).FieldByName(name).SetBool(true)
default: default:
logger.Default.Warn(context.Background(), "Model %v don't match %vInterface, should be %v(*gorm.DB)", schema, name, name) logger.Default.Warn(context.Background(), "Model %v don't match %sInterface, should be %s(*gorm.DB)", schema, name, name)
} }
} }
} }