mirror of
https://github.com/go-admin-team/go-admin.git
synced 2026-09-21 10:13:01 +00:00
zhangwenjian
95077b116b
The exclusion list was a subquery against `$GenConfig.DBName`.sys_tables, so it named the schema by hand. Generating from a schema that is not the one holding sys_tables made the whole query fail, and because the subquery read the table directly it also counted soft-deleted entries: deleting a generator entry never handed its table back. Read the registrations through the model on this connection instead. An empty list skips the clause - NOT IN (NULL) is unknown for every row, which would leave a fresh install with nothing to generate from.
69 lines
2.4 KiB
Go
69 lines
2.4 KiB
Go
package tools
|
||
|
||
import (
|
||
"errors"
|
||
"github.com/go-admin-team/go-admin-core/sdk/pkg"
|
||
|
||
"gorm.io/gorm"
|
||
|
||
config2 "github.com/go-admin-team/go-admin-core/sdk/config"
|
||
)
|
||
|
||
type DBTables struct {
|
||
TableName string `gorm:"column:TABLE_NAME" json:"tableName"`
|
||
Engine string `gorm:"column:ENGINE" json:"engine"`
|
||
TableRows string `gorm:"column:TABLE_ROWS" json:"tableRows"`
|
||
TableCollation string `gorm:"column:TABLE_COLLATION" json:"tableCollation"`
|
||
CreateTime string `gorm:"column:CREATE_TIME" json:"createTime"`
|
||
UpdateTime string `gorm:"column:UPDATE_TIME" json:"updateTime"`
|
||
TableComment string `gorm:"column:TABLE_COMMENT" json:"tableComment"`
|
||
}
|
||
|
||
func (e *DBTables) GetPage(tx *gorm.DB, pageSize int, pageIndex int) ([]DBTables, int, error) {
|
||
pkg.Assert(config2.DatabaseConfig.Driver == "mysql", "目前只支持mysql数据库", 500)
|
||
|
||
var doc []DBTables
|
||
var count int64
|
||
|
||
// Tables already registered with the generator are not candidates. Read them
|
||
// through the model on this connection: the subquery used to spell the
|
||
// schema out by hand, so it only resolved when sys_tables happened to live
|
||
// in the schema being generated from, and it counted soft-deleted rows.
|
||
var generated []string
|
||
if err := tx.Model(&SysTables{}).Pluck("table_name", &generated).Error; err != nil {
|
||
return nil, 0, err
|
||
}
|
||
|
||
table := tx.Table("information_schema.tables")
|
||
table = table.Where("table_schema= ? ", config2.GenConfig.DBName)
|
||
if len(generated) > 0 {
|
||
// NOT IN (NULL) is unknown for every row, so an empty list has to skip
|
||
// the clause instead of rendering it.
|
||
table = table.Where("TABLE_NAME not in (?)", generated)
|
||
}
|
||
|
||
if e.TableName != "" {
|
||
table = table.Where("TABLE_NAME = ?", e.TableName)
|
||
}
|
||
if err := table.Offset((pageIndex - 1) * pageSize).Limit(pageSize).Find(&doc).Offset(-1).Limit(-1).Count(&count).Error; err != nil {
|
||
return nil, 0, err
|
||
}
|
||
return doc, int(count), nil
|
||
}
|
||
|
||
func (e *DBTables) Get(tx *gorm.DB) (DBTables, error) {
|
||
pkg.Assert(config2.DatabaseConfig.Driver == "mysql", "目前只支持mysql数据库", 500)
|
||
|
||
var doc DBTables
|
||
if e.TableName == "" {
|
||
return doc, errors.New("table name cannot be empty！")
|
||
}
|
||
table := tx.Table("information_schema.tables")
|
||
table = table.Where("table_schema= ? ", config2.GenConfig.DBName)
|
||
table = table.Where("TABLE_NAME = ?", e.TableName)
|
||
if err := table.First(&doc).Error; err != nil {
|
||
return doc, err
|
||
}
|
||
return doc, nil
|
||
}
|