Commit 9f95fd33 authored by astaxie's avatar astaxie

Merge pull request #608 from JessonChan/develop

refactor func
parents 29e113a4 74c309ce
......@@ -16,7 +16,7 @@ import (
// register models.
// prefix means table name prefix.
func registerModel(model interface{}, prefix string) {
func registerModel(prefix string, model interface{}) {
val := reflect.ValueOf(model)
ind := reflect.Indirect(val)
typ := ind.Type()
......@@ -292,23 +292,17 @@ end:
// register models
func RegisterModel(models ...interface{}) {
if modelCache.done {
panic(fmt.Errorf("RegisterModel must be run before BootStrap"))
}
for _, model := range models {
registerModel(model, "")
}
RegisterModelWithPrefix("", models)
}
// register model with a prefix
// register models with a prefix
func RegisterModelWithPrefix(prefix string, models ...interface{}) {
if modelCache.done {
panic(fmt.Errorf("RegisterModel must be run before BootStrap"))
}
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