refactor: dbm包重构

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

@@ -4,167 +4,18 @@ import (
 	"database/sql"
 	"fmt"
 	"mayfly-go/internal/db/dbm/dbi"
-	"mayfly-go/pkg/errorx"
 	"mayfly-go/pkg/utils/anyx"
-	"mayfly-go/pkg/utils/collx"
 	"regexp"
 	"strings"
 	"time"
 )
 
-const (
-	MYSQL_META_FILE      = "metasql/mysql_meta.sql"
-	MYSQL_DBS            = "MYSQL_DBS"
-	MYSQL_TABLE_INFO_KEY = "MYSQL_TABLE_INFO"
-	MYSQL_INDEX_INFO_KEY = "MYSQL_INDEX_INFO"
-	MYSQL_COLUMN_MA_KEY  = "MYSQL_COLUMN_MA"
-)
-
 type MysqlDialect struct {
 	dc *dbi.DbConn
 }
 
-func (md *MysqlDialect) GetDbServer() (*dbi.DbServer, error) {
-	_, res, err := md.dc.Query("SELECT VERSION() version")
-	if err != nil {
-		return nil, err
-	}
-	ds := &dbi.DbServer{
-		Version: anyx.ConvString(res[0]["version"]),
-	}
-	return ds, nil
-}
-
-func (md *MysqlDialect) GetDbNames() ([]string, error) {
-	_, res, err := md.dc.Query(dbi.GetLocalSql(MYSQL_META_FILE, MYSQL_DBS))
-	if err != nil {
-		return nil, err
-	}
-
-	databases := make([]string, 0)
-	for _, re := range res {
-		databases = append(databases, anyx.ConvString(re["dbname"]))
-	}
-	return databases, nil
-}
-
-// 获取表基础元信息, 如表名等
-func (md *MysqlDialect) GetTables() ([]dbi.Table, error) {
-	_, res, err := md.dc.Query(dbi.GetLocalSql(MYSQL_META_FILE, MYSQL_TABLE_INFO_KEY))
-	if err != nil {
-		return nil, err
-	}
-
-	tables := make([]dbi.Table, 0)
-	for _, re := range res {
-		tables = append(tables, dbi.Table{
-			TableName:    anyx.ConvString(re["tableName"]),
-			TableComment: anyx.ConvString(re["tableComment"]),
-			CreateTime:   anyx.ConvString(re["createTime"]),
-			TableRows:    anyx.ConvInt(re["tableRows"]),
-			DataLength:   anyx.ConvInt64(re["dataLength"]),
-			IndexLength:  anyx.ConvInt64(re["indexLength"]),
-		})
-	}
-	return tables, nil
-}
-
-// 获取列元信息, 如列名等
-func (md *MysqlDialect) 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))
-	}), ",")
-
-	_, res, err := md.dc.Query(fmt.Sprintf(dbi.GetLocalSql(MYSQL_META_FILE, MYSQL_COLUMN_MA_KEY), tableName))
-	if err != nil {
-		return nil, err
-	}
-
-	columns := make([]dbi.Column, 0)
-	for _, re := range res {
-		columns = append(columns, dbi.Column{
-			TableName:     anyx.ConvString(re["tableName"]),
-			ColumnName:    anyx.ConvString(re["columnName"]),
-			ColumnType:    anyx.ConvString(re["columnType"]),
-			ColumnComment: anyx.ConvString(re["columnComment"]),
-			Nullable:      anyx.ConvString(re["nullable"]),
-			IsPrimaryKey:  anyx.ConvInt(re["isPrimaryKey"]) == 1,
-			IsIdentity:    anyx.ConvInt(re["isIdentity"]) == 1,
-			ColumnDefault: anyx.ConvString(re["columnDefault"]),
-			NumScale:      anyx.ConvString(re["numScale"]),
-		})
-	}
-	return columns, nil
-}
-
-// 获取表主键字段名，不存在主键标识则默认第一个字段
-func (md *MysqlDialect) GetPrimaryKey(tablename string) (string, error) {
-	columns, err := md.GetColumns(tablename)
-	if err != nil {
-		return "", err
-	}
-	if len(columns) == 0 {
-		return "", errorx.NewBiz("[%s] 表不存在", tablename)
-	}
-
-	for _, v := range columns {
-		if v.IsPrimaryKey {
-			return v.ColumnName, nil
-		}
-	}
-
-	return columns[0].ColumnName, nil
-}
-
-// 获取表索引信息
-func (md *MysqlDialect) GetTableIndex(tableName string) ([]dbi.Index, error) {
-	_, res, err := md.dc.Query(dbi.GetLocalSql(MYSQL_META_FILE, MYSQL_INDEX_INFO_KEY), tableName)
-	if err != nil {
-		return nil, err
-	}
-
-	indexs := make([]dbi.Index, 0)
-	for _, re := range res {
-		indexs = append(indexs, dbi.Index{
-			IndexName:    anyx.ConvString(re["indexName"]),
-			ColumnName:   anyx.ConvString(re["columnName"]),
-			IndexType:    anyx.ConvString(re["indexType"]),
-			IndexComment: anyx.ConvString(re["indexComment"]),
-			IsUnique:     anyx.ConvInt(re["isUnique"]) == 1,
-			SeqInIndex:   anyx.ConvInt(re["seqInIndex"]),
-		})
-	}
-	// 把查询结果以索引名分组，索引字段以逗号连接
-	result := make([]dbi.Index, 0)
-	key := ""
-	for _, v := range indexs {
-		// 当前的索引名
-		in := v.IndexName
-		if key == in {
-			// 索引字段已根据名称和顺序排序，故取最后一个即可
-			i := len(result) - 1
-			// 同索引字段以逗号连接
-			result[i].ColumnName = result[i].ColumnName + "," + v.ColumnName
-		} else {
-			key = in
-			result = append(result, v)
-		}
-	}
-	return result, nil
-}
-
-// 获取建表ddl
-func (md *MysqlDialect) GetTableDDL(tableName string) (string, error) {
-	_, res, err := md.dc.Query(fmt.Sprintf("show create table `%s` ", tableName))
-	if err != nil {
-		return "", err
-	}
-	return anyx.ConvString(res[0]["Create Table"]) + ";", nil
-}
-
-func (md *MysqlDialect) GetSchemas() ([]string, error) {
-	return nil, nil
+func (md *MysqlDialect) GetMetaData() dbi.MetaData {
+	return &MysqlMetaData{dc: md.dc}
 }
 
 // GetDbProgram 获取数据库程序模块，用于数据库备份与恢复