From be30fb7937cdcbf2b9796f2e4e39c128ade0dacf Mon Sep 17 00:00:00 2001 From: jessonchan Date: Wed, 14 May 2014 20:08:51 +0800 Subject: [PATCH] refator func --- orm/models_boot.go | 14 ++++---------- 1 file changed, 4 insertions(+), 10 deletions(-) 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) } }