refator func
Showing
1 changed file
with
4 additions
and
10 deletions
| ... | @@ -16,7 +16,7 @@ import ( | ... | @@ -16,7 +16,7 @@ import ( |
| 16 | 16 | ||
| 17 | // register models. | 17 | // register models. |
| 18 | // prefix means table name prefix. | 18 | // prefix means table name prefix. |
| 19 | func registerModel(model interface{}, prefix string) { | 19 | func registerModel(prefix string, model interface{}) { |
| 20 | val := reflect.ValueOf(model) | 20 | val := reflect.ValueOf(model) |
| 21 | ind := reflect.Indirect(val) | 21 | ind := reflect.Indirect(val) |
| 22 | typ := ind.Type() | 22 | typ := ind.Type() |
| ... | @@ -292,23 +292,17 @@ end: | ... | @@ -292,23 +292,17 @@ end: |
| 292 | 292 | ||
| 293 | // register models | 293 | // register models |
| 294 | func RegisterModel(models ...interface{}) { | 294 | func RegisterModel(models ...interface{}) { |
| 295 | if modelCache.done { | 295 | RegisterModelWithPrefix("", models) |
| 296 | panic(fmt.Errorf("RegisterModel must be run before BootStrap")) | ||
| 297 | } | ||
| 298 | |||
| 299 | for _, model := range models { | ||
| 300 | registerModel(model, "") | ||
| 301 | } | ||
| 302 | } | 296 | } |
| 303 | 297 | ||
| 304 | // register model with a prefix | 298 | // register models with a prefix |
| 305 | func RegisterModelWithPrefix(prefix string, models ...interface{}) { | 299 | func RegisterModelWithPrefix(prefix string, models ...interface{}) { |
| 306 | if modelCache.done { | 300 | if modelCache.done { |
| 307 | panic(fmt.Errorf("RegisterModel must be run before BootStrap")) | 301 | panic(fmt.Errorf("RegisterModel must be run before BootStrap")) |
| 308 | } | 302 | } |
| 309 | 303 | ||
| 310 | for _, model := range models { | 304 | for _, model := range models { |
| 311 | registerModel(model, prefix) | 305 | registerModel(prefix, model) |
| 312 | } | 306 | } |
| 313 | } | 307 | } |
| 314 | 308 | ... | ... |
-
Please register or sign in to post a comment