refactor: dbm重构、调整metadata与dialect接口

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

@@ -19,13 +19,13 @@ const (
 	SQLITE_INDEX_INFO_KEY = "SQLITE_INDEX_INFO"
 )
 
-type SqliteMetaData struct {
-	dbi.DefaultMetaData
+type SqliteMetadata struct {
+	dbi.DefaultMetadata
 
 	dc *dbi.DbConn
 }
 
-func (sd *SqliteMetaData) GetDbServer() (*dbi.DbServer, error) {
+func (sd *SqliteMetadata) GetDbServer() (*dbi.DbServer, error) {
 	_, res, err := sd.dc.Query("SELECT SQLITE_VERSION() as version")
 	if err != nil {
 		return nil, err
@@ -36,7 +36,7 @@ func (sd *SqliteMetaData) GetDbServer() (*dbi.DbServer, error) {
 	return ds, nil
 }
 
-func (sd *SqliteMetaData) GetDbNames() ([]string, error) {
+func (sd *SqliteMetadata) GetDbNames() ([]string, error) {
 	databases := make([]string, 0)
 	_, res, err := sd.dc.Query("PRAGMA database_list")
 	if err != nil {
@@ -50,9 +50,10 @@ func (sd *SqliteMetaData) GetDbNames() ([]string, error) {
 }
 
 // 获取表基础元信息, 如表名等
-func (sd *SqliteMetaData) GetTables(tableNames ...string) ([]dbi.Table, error) {
+func (sd *SqliteMetadata) GetTables(tableNames ...string) ([]dbi.Table, error) {
+	dialect := sd.dc.GetDialect()
 	names := strings.Join(collx.ArrayMap[string, string](tableNames, func(val string) string {
-		return fmt.Sprintf("'%s'", dbi.RemoveQuote(sd, val))
+		return fmt.Sprintf("'%s'", dialect.RemoveQuote(val))
 	}), ",")
 
 	var res []map[string]any
@@ -86,7 +87,7 @@ func (sd *SqliteMetaData) GetTables(tableNames ...string) ([]dbi.Table, error) {
 // 如 decimal(10,2)  提取decimal, 10 ,2
 // 如:text 提取text,null,null
 // 如:varchar(100)  提取varchar, 100
-func (sd *SqliteMetaData) getDataTypes(dataType string) (string, string, string) {
+func (sd *SqliteMetadata) getDataTypes(dataType string) (string, string, string) {
 	matches := dataTypeRegexp.FindStringSubmatch(dataType)
 	if len(matches) == 0 {
 		return dataType, "", ""
@@ -95,9 +96,9 @@ func (sd *SqliteMetaData) getDataTypes(dataType string) (string, string, string)
 }
 
 // 获取列元信息, 如列名等
-func (sd *SqliteMetaData) GetColumns(tableNames ...string) ([]dbi.Column, error) {
+func (sd *SqliteMetadata) GetColumns(tableNames ...string) ([]dbi.Column, error) {
 	columns := make([]dbi.Column, 0)
-	columnHelper := sd.dc.GetMetaData().GetColumnHelper()
+	columnHelper := sd.dc.GetDialect().GetColumnHelper()
 	for i := 0; i < len(tableNames); i++ {
 		tableName := tableNames[i]
 		_, res, err := sd.dc.Query(fmt.Sprintf("PRAGMA table_info(%s)", tableName))
@@ -142,7 +143,7 @@ func (sd *SqliteMetaData) GetColumns(tableNames ...string) ([]dbi.Column, error)
 	return columns, nil
 }
 
-func (sd *SqliteMetaData) GetPrimaryKey(tableName string) (string, error) {
+func (sd *SqliteMetadata) GetPrimaryKey(tableName string) (string, error) {
 	_, res, err := sd.dc.Query(fmt.Sprintf("PRAGMA table_info(%s)", tableName))
 	if err != nil {
 		return "", err
@@ -173,7 +174,7 @@ func extractIndexFields(indexSQL string) string {
 }
 
 // 获取表索引信息
-func (sd *SqliteMetaData) GetTableIndex(tableName string) ([]dbi.Index, error) {
+func (sd *SqliteMetadata) GetTableIndex(tableName string) ([]dbi.Index, error) {
 	_, res, err := sd.dc.Query(fmt.Sprintf(dbi.GetLocalSql(SQLITE_META_FILE, SQLITE_INDEX_INFO_KEY), tableName))
 	if err != nil {
 		return nil, err
@@ -198,96 +199,8 @@ func (sd *SqliteMetaData) GetTableIndex(tableName string) ([]dbi.Index, error) {
 	return indexs, nil
 }
 
-// 获取建索引ddl
-func (sd *SqliteMetaData) GenerateIndexDDL(indexs []dbi.Index, tableInfo dbi.Table) []string {
-	meta := sd.dc.GetMetaData()
-	sqls := make([]string, 0)
-	for _, index := range indexs {
-		unique := ""
-		if index.IsUnique {
-			unique = "unique"
-		}
-		// 取出列名，添加引号
-		cols := strings.Split(index.ColumnName, ",")
-		colNames := make([]string, len(cols))
-		for i, name := range cols {
-			colNames[i] = meta.QuoteIdentifier(name)
-		}
-		// 创建前尝试删除
-		sqls = append(sqls, fmt.Sprintf("DROP INDEX IF EXISTS \"%s\"", index.IndexName))
-
-		sqlTmp := "CREATE %s INDEX %s ON %s (%s) "
-		sqls = append(sqls, fmt.Sprintf(sqlTmp, unique, meta.QuoteIdentifier(index.IndexName), meta.QuoteIdentifier(tableInfo.TableName), strings.Join(colNames, ",")))
-	}
-	return sqls
-}
-
-func (sd *SqliteMetaData) genColumnBasicSql(column dbi.Column) string {
-	incr := ""
-	if column.IsIdentity {
-		incr = " AUTOINCREMENT"
-	}
-
-	nullAble := ""
-	if !column.Nullable {
-		nullAble = " NOT NULL"
-	}
-
-	quoteColumnName := sd.dc.GetMetaData().QuoteIdentifier(column.ColumnName)
-
-	// 如果是主键，则直接返回，不判断默认值
-	if column.IsPrimaryKey {
-		return fmt.Sprintf(" %s integer PRIMARY KEY%s%s", quoteColumnName, incr, nullAble)
-	}
-
-	defVal := "" // 默认值需要判断引号，如函数是不需要引号的 // 为了防止跨源函数不支持 当默认值是函数时，不需要设置默认值
-	if column.ColumnDefault != "" && !strings.Contains(column.ColumnDefault, "(") {
-		// 哪些字段类型默认值需要加引号
-		mark := false
-		if collx.ArrayAnyMatches([]string{"char", "text", "date", "time", "lob"}, strings.ToLower(string(column.DataType))) {
-			// 当数据类型是日期时间，默认值是日期时间函数时，默认值不需要引号
-			if collx.ArrayAnyMatches([]string{"date", "time"}, strings.ToLower(string(column.DataType))) &&
-				collx.ArrayAnyMatches([]string{"DATE", "TIME"}, strings.ToUpper(column.ColumnDefault)) {
-				mark = false
-			} else {
-				mark = true
-			}
-		}
-		if mark {
-			defVal = fmt.Sprintf(" DEFAULT '%s'", column.ColumnDefault)
-		} else {
-			defVal = fmt.Sprintf(" DEFAULT %s", column.ColumnDefault)
-		}
-	}
-
-	return fmt.Sprintf(" %s %s%s%s", quoteColumnName, column.GetColumnType(), nullAble, defVal)
-}
-
 // 获取建表ddl
-func (sd *SqliteMetaData) GenerateTableDDL(columns []dbi.Column, tableInfo dbi.Table, dropBeforeCreate bool) []string {
-	sqlArr := make([]string, 0)
-	tbName := sd.dc.GetMetaData().QuoteIdentifier(tableInfo.TableName)
-	if dropBeforeCreate {
-		sqlArr = append(sqlArr, fmt.Sprintf("DROP TABLE IF EXISTS %s", tbName))
-	}
-	// 组装建表语句
-	createSql := fmt.Sprintf("CREATE TABLE %s (\n", tbName)
-	fields := make([]string, 0)
-
-	// 把通用类型转换为达梦类型
-	for _, column := range columns {
-		fields = append(fields, sd.genColumnBasicSql(column))
-	}
-	createSql += strings.Join(fields, ",\n")
-	createSql += "\n)"
-
-	sqlArr = append(sqlArr, createSql)
-
-	return sqlArr
-}
-
-// 获取建表ddl
-func (sd *SqliteMetaData) GetTableDDL(tableName string, dropBeforeCreate bool) (string, error) {
+func (sd *SqliteMetadata) GetTableDDL(tableName string, dropBeforeCreate bool) (string, error) {
 	var builder strings.Builder
 
 	if dropBeforeCreate {
@@ -306,18 +219,6 @@ func (sd *SqliteMetaData) GetTableDDL(tableName string, dropBeforeCreate bool) (
 	return builder.String(), nil
 }
 
-func (sd *SqliteMetaData) GetSchemas() ([]string, error) {
+func (sd *SqliteMetadata) GetSchemas() ([]string, error) {
 	return nil, nil
 }
-
-func (sd *SqliteMetaData) GetDataHelper() dbi.DataHelper {
-	return new(DataHelper)
-}
-
-func (sd *SqliteMetaData) GetColumnHelper() dbi.ColumnHelper {
-	return new(ColumnHelper)
-}
-
-func (sd *SqliteMetaData) GetDumpHelper() dbi.DumpHelper {
-	return new(DumpHelper)
-}