mirror of
https://github.com/beego/bee.git
synced 2024-11-05 06:50:53 +00:00
89 lines
2.2 KiB
Go
89 lines
2.2 KiB
Go
package beegopro
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
|
|
"github.com/beego/bee/v2/internal/pkg/utils"
|
|
beeLogger "github.com/beego/bee/v2/logger"
|
|
)
|
|
|
|
type MysqlParser struct {
|
|
userOption UserOption
|
|
tmplOption TmplOption
|
|
}
|
|
|
|
func (m *MysqlParser) RegisterOption(userOption UserOption, tmplOption TmplOption) {
|
|
m.userOption = userOption
|
|
m.tmplOption = tmplOption
|
|
|
|
}
|
|
|
|
func (*MysqlParser) Parse(descriptor Descriptor) {
|
|
|
|
}
|
|
|
|
func (m *MysqlParser) GetRenderInfos(descriptor Descriptor) (output []RenderInfo) {
|
|
tableSchemas, err := m.getTableSchemas()
|
|
if err != nil {
|
|
beeLogger.Log.Fatalf("get table schemas err %s", err)
|
|
}
|
|
models := tableSchemas.ToTableMap()
|
|
|
|
output = make([]RenderInfo, 0)
|
|
// model table name, model table schema
|
|
for modelName, content := range models {
|
|
output = append(output, RenderInfo{
|
|
Module: descriptor.Module,
|
|
ModelName: modelName,
|
|
Content: content,
|
|
Option: m.userOption,
|
|
Descriptor: descriptor,
|
|
TmplPath: m.tmplOption.RenderPath,
|
|
})
|
|
}
|
|
return
|
|
}
|
|
|
|
func (t *MysqlParser) Unregister() {
|
|
|
|
}
|
|
|
|
func (m *MysqlParser) getTableSchemas() (resp TableSchemas, err error) {
|
|
dsn, err := utils.ParseDSN(m.userOption.Dsn)
|
|
if err != nil {
|
|
beeLogger.Log.Fatalf("parse dsn err %s", err)
|
|
return
|
|
}
|
|
|
|
conn, err := sql.Open("mysql", fmt.Sprintf("%s:%s@tcp(%s)/information_schema", dsn.User, dsn.Passwd, dsn.Addr))
|
|
if err != nil {
|
|
beeLogger.Log.Fatalf("Could not connect to mysql database using '%s': %s", m.userOption.Dsn, err)
|
|
return
|
|
}
|
|
defer conn.Close()
|
|
|
|
q := `SELECT TABLE_NAME, COLUMN_NAME, IS_NULLABLE, DATA_TYPE, CHARACTER_MAXIMUM_LENGTH,
|
|
NUMERIC_PRECISION, NUMERIC_SCALE,COLUMN_TYPE,COLUMN_KEY,COLUMN_COMMENT
|
|
FROM COLUMNS WHERE TABLE_SCHEMA = ? ORDER BY TABLE_NAME, ORDINAL_POSITION`
|
|
rows, err := conn.Query(q, dsn.DBName)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
columns := make(TableSchemas, 0)
|
|
for rows.Next() {
|
|
cs := TableSchema{}
|
|
err := rows.Scan(&cs.TableName, &cs.ColumnName, &cs.IsNullable, &cs.DataType,
|
|
&cs.CharacterMaximumLength, &cs.NumericPrecision, &cs.NumericScale,
|
|
&cs.ColumnType, &cs.ColumnKey, &cs.Comment)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
columns = append(columns, cs)
|
|
}
|
|
if err := rows.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
return columns, nil
|
|
}
|