be30fb79 by jessonchan Committed by astaxie

refator func

1 parent f4e7d63e
...@@ -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
......
Styling with Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!