light-apiengine-feature/models/model.go

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{}) {
	beansLock.Lock()
	defer beansLock.Unlock()

	for _, model := range models {
		beans = append(beans, model)
	}
}

func AddTableName(tableName string) {
	beansLock.Lock()
    defer beansLock.Unlock()

    ModelNameList = append(ModelNameList, tableName)
}

func SyncDb(db *xorm.Engine) {
	if !config.AppConfig.SyncDb {
		return
	}
	
	for _, bean := range beans {
		if err := db.Sync2(bean); err != nil {
			fmt.Println(err)
		}
	}
}