refactor: dbm包重构

server/internal/db/dbm/mysql/dialect.go

@@ -4,8 +4,6 @@ import (
 	"database/sql"
 	"fmt"
 	"mayfly-go/internal/db/dbm/dbi"
-	"mayfly-go/pkg/utils/anyx"
-	"regexp"
 	"strings"
 	"time"
 )
@@ -14,10 +12,6 @@ type MysqlDialect struct {
 	dc *dbi.DbConn
 }
 
-func (md *MysqlDialect) GetMetaData() dbi.MetaData {
-	return &MysqlMetaData{dc: md.dc}
-}
-
 // GetDbProgram 获取数据库程序模块，用于数据库备份与恢复
 func (md *MysqlDialect) GetDbProgram() (dbi.DbProgram, error) {
 	return NewDbProgramMysql(md.dc), nil
@@ -45,7 +39,7 @@ func (md *MysqlDialect) BatchInsert(tx *sql.Tx, tableName string, columns []stri
 		prefix = "replace into"
 	}
 
-	sqlStr := fmt.Sprintf("%s %s (%s) values %s", prefix, md.dc.Info.Type.QuoteIdentifier(tableName), strings.Join(columns, ","), placeholder)
+	sqlStr := fmt.Sprintf("%s %s (%s) values %s", prefix, md.dc.GetMetaData().QuoteIdentifier(tableName), strings.Join(columns, ","), placeholder)
 	// 执行批量insert sql
 	// 把二维数组转为一维数组
 	var args []any
@@ -55,69 +49,6 @@ func (md *MysqlDialect) BatchInsert(tx *sql.Tx, tableName string, columns []stri
 	return md.dc.TxExec(tx, sqlStr, args...)
 }
 
-func (md *MysqlDialect) GetDataConverter() dbi.DataConverter {
-	return converter
-}
-
-var (
-	// 数字类型
-	numberRegexp = regexp.MustCompile(`(?i)int|double|float|number|decimal|byte|bit`)
-	// 日期时间类型
-	datetimeRegexp = regexp.MustCompile(`(?i)datetime|timestamp`)
-	// 日期类型
-	dateRegexp = regexp.MustCompile(`(?i)date`)
-	// 时间类型
-	timeRegexp = regexp.MustCompile(`(?i)time`)
-
-	converter = new(DataConverter)
-)
-
-type DataConverter struct {
-}
-
-func (dc *DataConverter) GetDataType(dbColumnType string) dbi.DataType {
-	if numberRegexp.MatchString(dbColumnType) {
-		return dbi.DataTypeNumber
-	}
-	// 日期时间类型
-	if datetimeRegexp.MatchString(dbColumnType) {
-		return dbi.DataTypeDateTime
-	}
-	// 日期类型
-	if dateRegexp.MatchString(dbColumnType) {
-		return dbi.DataTypeDate
-	}
-	// 时间类型
-	if timeRegexp.MatchString(dbColumnType) {
-		return dbi.DataTypeTime
-	}
-	return dbi.DataTypeString
-}
-
-func (dc *DataConverter) FormatData(dbColumnValue any, dataType dbi.DataType) string {
-	return anyx.ToString(dbColumnValue)
-}
-
-func (dc *DataConverter) ParseData(dbColumnValue any, dataType dbi.DataType) any {
-	// 如果dataType是datetime而dbColumnValue是string类型，则需要转换为time.Time类型
-	_, ok := dbColumnValue.(string)
-	if ok {
-		if dataType == dbi.DataTypeDateTime {
-			res, _ := time.Parse(time.DateTime, anyx.ToString(dbColumnValue))
-			return res
-		}
-		if dataType == dbi.DataTypeDate {
-			res, _ := time.Parse(time.DateOnly, anyx.ToString(dbColumnValue))
-			return res
-		}
-		if dataType == dbi.DataTypeTime {
-			res, _ := time.Parse(time.TimeOnly, anyx.ToString(dbColumnValue))
-			return res
-		}
-	}
-	return dbColumnValue
-}
-
 func (md *MysqlDialect) CopyTable(copy *dbi.DbCopyTable) error {
 
 	tableName := copy.TableName

server/internal/db/dbm/mysql/meta.go

@@ -40,5 +40,9 @@ func (md *MysqlMeta) GetSqlDb(d *dbi.DbInfo) (*sql.DB, error) {
 }
 
 func (md *MysqlMeta) GetDialect(conn *dbi.DbConn) dbi.Dialect {
-	return &MysqlDialect{conn}
+	return &MysqlDialect{dc: conn}
+}
+
+func (md *MysqlMeta) GetMetaData(conn *dbi.DbConn) *dbi.MetaDataX {
+	return dbi.NewMetaDataX(&MysqlMetaData{dc: conn})
 }

server/internal/db/dbm/mysql/metadata.go

@@ -7,7 +7,11 @@ import (
 	"mayfly-go/pkg/errorx"
 	"mayfly-go/pkg/utils/anyx"
 	"mayfly-go/pkg/utils/collx"
+	"regexp"
 	"strings"
+	"time"
+
+	"github.com/kanzihuang/vitess/go/vt/sqlparser"
 )
 
 const (
@@ -19,6 +23,8 @@ const (
 )
 
 type MysqlMetaData struct {
+	dbi.DefaultMetaData
+
 	dc *dbi.DbConn
 }
 
@@ -69,9 +75,8 @@ func (md *MysqlMetaData) GetTables() ([]dbi.Table, error) {
 
 // 获取列元信息, 如列名等
 func (md *MysqlMetaData) GetColumns(tableNames ...string) ([]dbi.Column, error) {
-	dbType := md.dc.Info.Type
 	tableName := strings.Join(collx.ArrayMap[string, string](tableNames, func(val string) string {
-		return fmt.Sprintf("'%s'", dbType.RemoveQuote(val))
+		return fmt.Sprintf("'%s'", dbi.RemoveQuote(md, val))
 	}), ",")
 
 	_, res, err := md.dc.Query(fmt.Sprintf(dbi.GetLocalSql(MYSQL_META_FILE, MYSQL_COLUMN_MA_KEY), tableName))
@@ -164,3 +169,80 @@ func (md *MysqlMetaData) GetTableDDL(tableName string) (string, error) {
 func (md *MysqlMetaData) GetSchemas() ([]string, error) {
 	return nil, errors.New("不支持schema")
 }
+
+func (md *MysqlMetaData) GetIdentifierQuoteString() string {
+	return "`"
+}
+
+func (md *MysqlMetaData) QuoteLiteral(literal string) string {
+	literal = strings.ReplaceAll(literal, `\`, `\\`)
+	literal = strings.ReplaceAll(literal, `'`, `''`)
+	return "'" + literal + "'"
+}
+
+func (md *MysqlMetaData) SqlParserDialect() sqlparser.Dialect {
+	return sqlparser.MysqlDialect{}
+}
+
+func (md *MysqlMetaData) GetDataConverter() dbi.DataConverter {
+	return converter
+}
+
+var (
+	// 数字类型
+	numberRegexp = regexp.MustCompile(`(?i)int|double|float|number|decimal|byte|bit`)
+	// 日期时间类型
+	datetimeRegexp = regexp.MustCompile(`(?i)datetime|timestamp`)
+	// 日期类型
+	dateRegexp = regexp.MustCompile(`(?i)date`)
+	// 时间类型
+	timeRegexp = regexp.MustCompile(`(?i)time`)
+
+	converter = new(DataConverter)
+)
+
+type DataConverter struct {
+}
+
+func (dc *DataConverter) GetDataType(dbColumnType string) dbi.DataType {
+	if numberRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeNumber
+	}
+	// 日期时间类型
+	if datetimeRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeDateTime
+	}
+	// 日期类型
+	if dateRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeDate
+	}
+	// 时间类型
+	if timeRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeTime
+	}
+	return dbi.DataTypeString
+}
+
+func (dc *DataConverter) FormatData(dbColumnValue any, dataType dbi.DataType) string {
+	return anyx.ToString(dbColumnValue)
+}
+
+func (dc *DataConverter) ParseData(dbColumnValue any, dataType dbi.DataType) any {
+	// 如果dataType是datetime而dbColumnValue是string类型，则需要转换为time.Time类型
+	_, ok := dbColumnValue.(string)
+	if ok {
+		if dataType == dbi.DataTypeDateTime {
+			res, _ := time.Parse(time.DateTime, anyx.ToString(dbColumnValue))
+			return res
+		}
+		if dataType == dbi.DataTypeDate {
+			res, _ := time.Parse(time.DateOnly, anyx.ToString(dbColumnValue))
+			return res
+		}
+		if dataType == dbi.DataTypeTime {
+			res, _ := time.Parse(time.TimeOnly, anyx.ToString(dbColumnValue))
+			return res
+		}
+	}
+	return dbColumnValue
+}