bee/internal/app/module/beegopro/parser_mysql.go

package beegopro

import (
	"database/sql"
	"fmt"

	"github.com/beego/bee/internal/pkg/utils"
	beeLogger "github.com/beego/bee/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
}