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

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

@@ -4,6 +4,7 @@ import (
 	"database/sql"
 	"fmt"
 	"mayfly-go/internal/db/dbm/dbi"
+	"mayfly-go/pkg/utils/collx"
 	"strings"
 	"time"
 )
@@ -35,7 +36,7 @@ func (sd *SqliteDialect) BatchInsert(tx *sql.Tx, tableName string, columns []str
 		prefix = "insert or replace into"
 	}
 
-	sqlStr := fmt.Sprintf("%s %s (%s) values %s", prefix, sd.dc.GetMetaData().QuoteIdentifier(tableName), strings.Join(columns, ","), placeholder)
+	sqlStr := fmt.Sprintf("%s %s (%s) values %s", prefix, sd.QuoteIdentifier(tableName), strings.Join(columns, ","), placeholder)
 
 	// 把二维数组转为一维数组
 	var args []any
@@ -55,7 +56,7 @@ func (sd *SqliteDialect) CopyTable(copy *dbi.DbCopyTable) error {
 
 	// 生成新表名,为老表明+_copy_时间戳
 	newTableName := tableName + "_copy_" + time.Now().Format("20060102150405")
-	ddl, err := sd.dc.GetMetaData().GetTableDDL(tableName, false)
+	ddl, err := sd.dc.GetMetadata().GetTableDDL(tableName, false)
 	if err != nil {
 		return err
 	}
@@ -82,24 +83,101 @@ func (sd *SqliteDialect) CopyTable(copy *dbi.DbCopyTable) error {
 	return err
 }
 
-func (sd *SqliteDialect) CreateTable(columns []dbi.Column, tableInfo dbi.Table, dropOldTable bool) (int, error) {
-	sqlArr := sd.dc.GetMetaData().GenerateTableDDL(columns, tableInfo, dropOldTable)
-	for _, sqlStr := range sqlArr {
-		_, err := sd.dc.Exec(sqlStr)
-		if err != nil {
-			return 0, err
-		}
+// 获取建表ddl
+func (sd *SqliteDialect) GenerateTableDDL(columns []dbi.Column, tableInfo dbi.Table, dropBeforeCreate bool) []string {
+	sqlArr := make([]string, 0)
+	tbName := sd.QuoteIdentifier(tableInfo.TableName)
+	if dropBeforeCreate {
+		sqlArr = append(sqlArr, fmt.Sprintf("DROP TABLE IF EXISTS %s", tbName))
 	}
-	return len(sqlArr), nil
+	// 组装建表语句
+	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
 }
 
-func (sd *SqliteDialect) CreateIndex(tableInfo dbi.Table, indexs []dbi.Index) error {
-	sqlArr := sd.dc.GetMetaData().GenerateIndexDDL(indexs, tableInfo)
-	for _, sqlStr := range sqlArr {
-		_, err := sd.dc.Exec(sqlStr)
-		if err != nil {
-			return err
+func (sd *SqliteDialect) genColumnBasicSql(column dbi.Column) string {
+	incr := ""
+	if column.IsIdentity {
+		incr = " AUTOINCREMENT"
+	}
+
+	nullAble := ""
+	if !column.Nullable {
+		nullAble = " NOT NULL"
+	}
+
+	quoteColumnName := sd.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 nil
+
+	return fmt.Sprintf(" %s %s%s%s", quoteColumnName, column.GetColumnType(), nullAble, defVal)
+}
+
+// 获取建索引ddl
+func (sd *SqliteDialect) GenerateIndexDDL(indexs []dbi.Index, tableInfo dbi.Table) []string {
+	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] = sd.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, sd.QuoteIdentifier(index.IndexName), sd.QuoteIdentifier(tableInfo.TableName), strings.Join(colNames, ",")))
+	}
+	return sqls
+}
+
+func (sd *SqliteDialect) GetDataHelper() dbi.DataHelper {
+	return new(DataHelper)
+}
+
+func (sd *SqliteDialect) GetColumnHelper() dbi.ColumnHelper {
+	return new(ColumnHelper)
+}
+
+func (sd *SqliteDialect) GetDumpHelper() dbi.DumpHelper {
+	return new(DumpHelper)
 }