소스 검색

modify model

zhangjq 6 년 전
부모
커밋
4b168b4e2e

+ 2 - 1
models/SysAttachment_gen.go

@@ -1,7 +1,7 @@
 
 package models
 import (
-"time"
+	"time"
 )
 
 type SysAttachment struct {
@@ -31,5 +31,6 @@ func (t *SysAttachment) TableName() string {
 }
 
 func init() {
+    AddTableName("sys_attachment")
 	RegisterModel(new(SysAttachment))
 }

+ 2 - 1
models/SysDomain_gen.go

@@ -1,7 +1,7 @@
 
 package models
 import (
-"time"
+	"time"
 )
 
 type SysDomain struct {
@@ -45,5 +45,6 @@ func (t *SysDomain) TableName() string {
 }
 
 func init() {
+    AddTableName("sys_domain")
 	RegisterModel(new(SysDomain))
 }

+ 2 - 1
models/SysMenu_gen.go

@@ -1,7 +1,7 @@
 
 package models
 import (
-"time"
+	"time"
 )
 
 type SysMenu struct {
@@ -47,5 +47,6 @@ func (t *SysMenu) TableName() string {
 }
 
 func init() {
+    AddTableName("sys_menu")
 	RegisterModel(new(SysMenu))
 }

+ 2 - 1
models/SysOrg_gen.go

@@ -1,7 +1,7 @@
 
 package models
 import (
-"time"
+	"time"
 )
 
 type SysOrg struct {
@@ -43,5 +43,6 @@ func (t *SysOrg) TableName() string {
 }
 
 func init() {
+    AddTableName("sys_org")
 	RegisterModel(new(SysOrg))
 }

+ 2 - 1
models/SysPermission_gen.go

@@ -1,7 +1,7 @@
 
 package models
 import (
-"time"
+	"time"
 )
 
 type SysPermission struct {
@@ -29,5 +29,6 @@ func (t *SysPermission) TableName() string {
 }
 
 func init() {
+    AddTableName("sys_permission")
 	RegisterModel(new(SysPermission))
 }

+ 2 - 1
models/SysRoleMenu_gen.go

@@ -1,7 +1,7 @@
 
 package models
 import (
-"time"
+	"time"
 )
 
 type SysRoleMenu struct {
@@ -31,5 +31,6 @@ func (t *SysRoleMenu) TableName() string {
 }
 
 func init() {
+    AddTableName("sys_role_menu")
 	RegisterModel(new(SysRoleMenu))
 }

+ 2 - 1
models/SysRolePermission_gen.go

@@ -1,7 +1,7 @@
 
 package models
 import (
-"time"
+	"time"
 )
 
 type SysRolePermission struct {
@@ -31,5 +31,6 @@ func (t *SysRolePermission) TableName() string {
 }
 
 func init() {
+    AddTableName("sys_role_permission")
 	RegisterModel(new(SysRolePermission))
 }

+ 2 - 1
models/SysRole_gen.go

@@ -1,7 +1,7 @@
 
 package models
 import (
-"time"
+	"time"
 )
 
 type SysRole struct {
@@ -35,5 +35,6 @@ func (t *SysRole) TableName() string {
 }
 
 func init() {
+    AddTableName("sys_role")
 	RegisterModel(new(SysRole))
 }

+ 2 - 1
models/SysUserRole_gen.go

@@ -1,7 +1,7 @@
 
 package models
 import (
-"time"
+	"time"
 )
 
 type SysUserRole struct {
@@ -31,5 +31,6 @@ func (t *SysUserRole) TableName() string {
 }
 
 func init() {
+    AddTableName("sys_user_role")
 	RegisterModel(new(SysUserRole))
 }

+ 2 - 1
models/SysUser_gen.go

@@ -1,7 +1,7 @@
 
 package models
 import (
-"time"
+	"time"
 )
 
 type SysUser struct {
@@ -49,5 +49,6 @@ func (t *SysUser) TableName() string {
 }
 
 func init() {
+    AddTableName("sys_user")
 	RegisterModel(new(SysUser))
 }

+ 14 - 1
models/model.go

@@ -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)
+		}
+	}
 }