Commit be30fb79 authored by jessonchan's avatar jessonchan Committed by astaxie

refator func

parent f4e7d63e
...@@ -16,7 +16,7 @@ import ( ...@@ -16,7 +16,7 @@ import (
// register models. // register models.
// prefix means table name prefix. // prefix means table name prefix.
func registerModel(model interface{}, prefix string) { func registerModel(prefix string, model interface{}) {
val := reflect.ValueOf(model) val := reflect.ValueOf(model)
ind := reflect.Indirect(val) ind := reflect.Indirect(val)
typ := ind.Type() typ := ind.Type()
...@@ -292,23 +292,17 @@ end: ...@@ -292,23 +292,17 @@ end:
// register models // register models
func RegisterModel(models ...interface{}) { func RegisterModel(models ...interface{}) {
if modelCache.done { RegisterModelWithPrefix("", models)
panic(fmt.Errorf("RegisterModel must be run before BootStrap"))
}
for _, model := range models {
registerModel(model, "")
}
} }
// register model with a prefix // register models with a prefix
func RegisterModelWithPrefix(prefix string, models ...interface{}) { func RegisterModelWithPrefix(prefix string, models ...interface{}) {
if modelCache.done { if modelCache.done {
panic(fmt.Errorf("RegisterModel must be run before BootStrap")) panic(fmt.Errorf("RegisterModel must be run before BootStrap"))
} }
for _, model := range models { for _, model := range models {
registerModel(model, prefix) registerModel(prefix, model)
} }
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment