mirror of
https://github.com/go-admin-team/go-admin.git
synced 2026-09-21 18:20:50 +00:00
zhangwenjian
f201792d8f
pkg.Assert panics when its condition is false, so pkg.Assert(true, "目前只支持mysql数据库") is a no-op. On postgres or sqlserver the code generator did not report that it needs MySQL: DBTables returned an empty list with a nil error, and DBColumns ran its query on the zero-value *gorm.DB left over from the branch that never assigned, which is a nil dereference rather than a message. Assert the driver up front instead of asserting a constant in an else, which also removes the placeholder *gorm.DB the fall-through relied on. DBColumns.GetPage had no guard at all and gets the same one.
62 lines
2.3 KiB
Go
62 lines
2.3 KiB
Go
package tools
|
||
|
||
import (
|
||
"errors"
|
||
|
||
"github.com/go-admin-team/go-admin-core/sdk/config"
|
||
"github.com/go-admin-team/go-admin-core/sdk/pkg"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
type DBColumns struct {
|
||
TableSchema string `gorm:"column:TABLE_SCHEMA" json:"tableSchema"`
|
||
TableName string `gorm:"column:TABLE_NAME" json:"tableName"`
|
||
ColumnName string `gorm:"column:COLUMN_NAME" json:"columnName"`
|
||
ColumnDefault string `gorm:"column:COLUMN_DEFAULT" json:"columnDefault"`
|
||
IsNullable string `gorm:"column:IS_NULLABLE" json:"isNullable"`
|
||
DataType string `gorm:"column:DATA_TYPE" json:"dataType"`
|
||
CharacterMaximumLength string `gorm:"column:CHARACTER_MAXIMUM_LENGTH" json:"characterMaximumLength"`
|
||
CharacterSetName string `gorm:"column:CHARACTER_SET_NAME" json:"characterSetName"`
|
||
ColumnType string `gorm:"column:COLUMN_TYPE" json:"columnType"`
|
||
ColumnKey string `gorm:"column:COLUMN_KEY" json:"columnKey"`
|
||
Extra string `gorm:"column:EXTRA" json:"extra"`
|
||
ColumnComment string `gorm:"column:COLUMN_COMMENT" json:"columnComment"`
|
||
}
|
||
|
||
func (e *DBColumns) GetPage(tx *gorm.DB, pageSize int, pageIndex int) ([]DBColumns, int, error) {
|
||
pkg.Assert(config.DatabaseConfig.Driver == "mysql", "目前只支持mysql数据库", 500)
|
||
|
||
var doc []DBColumns
|
||
var count int64
|
||
|
||
if e.TableName == "" {
|
||
return nil, 0, errors.New("table name cannot be empty！")
|
||
}
|
||
|
||
table := tx.Table("information_schema.`COLUMNS`")
|
||
table = table.Where("table_schema= ? ", config.GenConfig.DBName)
|
||
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 *DBColumns) GetList(tx *gorm.DB) ([]DBColumns, error) {
|
||
pkg.Assert(config.DatabaseConfig.Driver == "mysql", "目前只支持mysql数据库", 500)
|
||
|
||
var doc []DBColumns
|
||
if e.TableName == "" {
|
||
return nil, errors.New("table name cannot be empty！")
|
||
}
|
||
|
||
table := tx.Table("information_schema.columns")
|
||
table = table.Where("table_schema= ? ", config.GenConfig.DBName)
|
||
table = table.Where("TABLE_NAME = ?", e.TableName).Order("ORDINAL_POSITION asc")
|
||
if err := table.Find(&doc).Error; err != nil {
|
||
return doc, err
|
||
}
|
||
return doc, nil
|
||
}
|