|
|
@@ -2,12 +2,14 @@
|
|
|
package models
|
|
|
|
|
|
import (
|
|
|
+ "fmt"
|
|
|
"git.qianqiusoft.com/qianqiusoft/light-apiengine/config"
|
|
|
"github.com/xormplus/xorm"
|
|
|
"sync"
|
|
|
)
|
|
|
|
|
|
var beans []interface{}
|
|
|
+var ModelNameList []string
|
|
|
var beansLock sync.Mutex
|
|
|
|
|
|
func RegisterModel(models ...interface{}) {
|
|
|
@@ -19,10 +21,21 @@ func RegisterModel(models ...interface{}) {
|
|
|
}
|
|
|
}
|
|
|
|
|
|
+func AddTableName(tableName string) {
|
|
|
+ beansLock.Lock()
|
|
|
+ defer beansLock.Unlock()
|
|
|
+
|
|
|
+ ModelNameList = append(ModelNameList, tableName)
|
|
|
+}
|
|
|
+
|
|
|
func SyncDb(db *xorm.Engine) {
|
|
|
if !config.AppConfig.SyncDb {
|
|
|
return
|
|
|
}
|
|
|
|
|
|
- db.Sync2(beans...)
|
|
|
+ for _, bean := range beans {
|
|
|
+ if err := db.Sync2(bean); err != nil {
|
|
|
+ fmt.Println(err)
|
|
|
+ }
|
|
|
+ }
|
|
|
}
|