diff --git a/orm/models_boot.go b/orm/models_boot.go index 0b8db512..16423755 100644 --- a/orm/models_boot.go +++ b/orm/models_boot.go @@ -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) } }