自动建立表通过-doris

drivers/doris.go

@@ -0,0 +1,45 @@
+package drivers
+
+import (
+	"fmt"
+
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/jmoiron/sqlx"
+)
+
+type DorisDriver struct{}
+
+func (d *DorisDriver) Name() string {
+	return "doris"
+}
+
+func (d *DorisDriver) BuildDSN(config DBConfig) string {
+	return fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local",
+		config.Username,
+		config.Password,
+		config.Host,
+		config.Port,
+		config.Database)
+}
+
+func (d *DorisDriver) Open(config DBConfig) (*sqlx.DB, error) {
+	dsn := d.BuildDSN(config)
+	db, err := sqlx.Open("mysql", dsn)
+	if err != nil {
+		return nil, err
+	}
+
+	// 使用公共的连接池配置函数
+	configureConnectionPool(db, config)
+
+	// 测试连接
+	if err = db.Ping(); err != nil {
+		return nil, err
+	}
+
+	return db, nil
+}
+
+func init() {
+	Register(&DorisDriver{})
+}

factory/database/db_factory.go

@@ -45,12 +45,11 @@ func GetDBFactory() (*DBFactory, error) {
 		}
 
 		// 显示所支持的数据库驱动
-		driversStr := drivers.GetAllDrivers()
-		log.Printf("Available database drivers: %v\n", driversStr)
+		//driversStr := drivers.GetAllDrivers()
 
 		dbConfig := cfg.GetDatabase()
 		dbType := dbConfig.Type
-
+		log.Printf("Available database drivers: %v\n", dbType)
 		// 获取对应的驱动
 		dbDriver, err := drivers.Get(dbType)
 		if err != nil {

sqldef/generators/doris.go

@@ -0,0 +1,188 @@
+package generators
+
+import (
+	"fmt"
+	"strings"
+)
+
+// DorisGenerator Doris SQL生成器
+type DorisGenerator struct{}
+
+// NewDorisGenerator 创建Doris生成器实例
+func NewDorisGenerator() *DorisGenerator {
+	return &DorisGenerator{}
+}
+
+func (dg *DorisGenerator) DBType() string {
+	return "doris"
+}
+
+func (dg *DorisGenerator) TableExistsSQL(tableName string) string {
+	return fmt.Sprintf(
+		"SELECT COUNT(*) FROM information_schema.tables WHERE table_schema = DATABASE() AND table_name = '%s'",
+		tableName,
+	)
+}
+
+func (dg *DorisGenerator) DropTableSQL(tableName string) string {
+	return fmt.Sprintf("DROP TABLE IF EXISTS %s", tableName)
+}
+
+func (dg *DorisGenerator) GenerateCreateTable(table TableDDL) string {
+	if table.Schema == nil {
+		return ""
+	}
+
+	var sql strings.Builder
+	var primaryKeyColumns []string
+
+	// 表头
+	sql.WriteString(fmt.Sprintf("CREATE TABLE IF NOT EXISTS %s (\n", table.Name))
+
+	// 列定义 - 第一遍收集主键列
+	for _, col := range table.Schema.Columns {
+		for _, opt := range col.Options {
+			if strings.ToUpper(opt) == "PRIMARY KEY" {
+				primaryKeyColumns = append(primaryKeyColumns, col.Name)
+				break
+			}
+		}
+	}
+
+	// 列定义 - 第二遍生成列定义
+	for i, col := range table.Schema.Columns {
+		colDef := fmt.Sprintf("  %s %s", col.Name, dg.getDorisType(col))
+
+		// 处理列选项，过滤掉 PRIMARY KEY（不在列级定义）
+		for _, opt := range col.Options {
+			upperOpt := strings.ToUpper(opt)
+			if upperOpt != "PRIMARY KEY" && upperOpt != "UNIQUE" {
+				colDef += " " + opt
+			}
+		}
+
+		// 如果是主键列，确保有 NOT NULL
+		isPrimaryKey := false
+		for _, pk := range primaryKeyColumns {
+			if pk == col.Name {
+				isPrimaryKey = true
+				break
+			}
+		}
+
+		if isPrimaryKey {
+			hasNotNull := false
+			for _, opt := range col.Options {
+				if strings.ToUpper(opt) == "NOT NULL" {
+					hasNotNull = true
+					break
+				}
+			}
+			if !hasNotNull {
+				colDef += " NOT NULL"
+			}
+		}
+
+		// 添加默认值
+		if col.Default != "" {
+			upperDefault := strings.ToUpper(col.Default)
+			if strings.Contains(upperDefault, "CURRENT_TIMESTAMP") ||
+				strings.Contains(upperDefault, "NOW()") ||
+				strings.Contains(upperDefault, "UUID()") ||
+				strings.Contains(upperDefault, "NULL") {
+				colDef += " DEFAULT " + col.Default
+			} else {
+				colDef += fmt.Sprintf(" DEFAULT '%s'", col.Default)
+			}
+		}
+
+		// 添加注释
+		if col.Comment != "" {
+			colDef += fmt.Sprintf(` COMMENT "%s"`, col.Comment)
+		}
+
+		if i < len(table.Schema.Columns)-1 {
+			sql.WriteString(colDef + ",\n")
+		} else {
+			sql.WriteString(colDef + "\n")
+		}
+	}
+
+	// 关闭列定义括号
+	sql.WriteString(")")
+
+	// 添加 UNIQUE KEY（如果有主键列）
+	if len(primaryKeyColumns) > 0 {
+		sql.WriteString(fmt.Sprintf("\nUNIQUE KEY(%s)", strings.Join(primaryKeyColumns, ", ")))
+	}
+
+	// 分布式配置 - 使用第一个主键列，如果没有主键则使用第一个列
+	distKey := ""
+	if len(primaryKeyColumns) > 0 {
+		distKey = primaryKeyColumns[0]
+	} else if len(table.Schema.Columns) > 0 {
+		distKey = table.Schema.Columns[0].Name
+	}
+
+	if distKey != "" {
+		sql.WriteString(fmt.Sprintf("\nDISTRIBUTED BY HASH(%s) BUCKETS 10", distKey))
+	}
+
+	// 副本数配置
+	sql.WriteString("\nPROPERTIES (\"replication_num\" = \"1\");")
+
+	return sql.String()
+}
+
+// getDorisType 获取Doris数据类型
+func (dg *DorisGenerator) getDorisType(col ColumnSchema) string {
+	switch col.Type {
+	case "DECIMAL":
+		if col.Precision > 0 && col.Scale > 0 {
+			return fmt.Sprintf("DECIMAL(%d,%d)", col.Precision, col.Scale)
+		}
+		return "DECIMAL"
+	case "VARCHAR":
+		if col.Length > 0 {
+			return fmt.Sprintf("VARCHAR(%d)", col.Length)
+		}
+		return "VARCHAR(255)"
+	case "CHAR":
+		if col.Length > 0 {
+			return fmt.Sprintf("CHAR(%d)", col.Length)
+		}
+		return "CHAR(1)"
+	case "INT":
+		return "INT"
+	case "BIGINT":
+		return "BIGINT"
+	case "TINYINT":
+		return "TINYINT"
+	case "BOOL":
+		return "BOOLEAN"
+	case "DATETIME":
+		return "DATETIME"
+	case "TIMESTAMP":
+		return "DATETIME" // Doris 中通常用 DATETIME
+	case "DATE":
+		return "DATE"
+	case "TIME":
+		return "TIME"
+	case "TEXT":
+		return "STRING"
+	case "JSON":
+		return "STRING"
+	case "BLOB":
+		return "STRING"
+	case "FLOAT":
+		return "FLOAT"
+	case "DOUBLE":
+		return "DOUBLE"
+	default:
+		return col.Type
+	}
+}
+
+func init() {
+	RegisterGenerator(NewDorisGenerator())
+}

sqldef/generators/mysql.go

@@ -1,4 +1,3 @@
-// mysql.go 根据//table_defintion.go 定义的表结构，编写mysql建立表和索引的代码。
 package generators
 
 import (
@@ -35,60 +34,108 @@ func (mg *MySQLGenerator) GenerateCreateTable(table TableDDL) string {
 	}
 
 	var sql strings.Builder
+	var pkColumns []string
 
 	// 表头
 	sql.WriteString(fmt.Sprintf("CREATE TABLE IF NOT EXISTS %s (\n", table.Name))
 
-	// 列定义
-	columns := table.Schema.Columns
-	for i, col := range columns {
-		sql.WriteString(fmt.Sprintf("  %s %s", col.Name, mg.getMySQLType(col)))
+	// 第一遍：收集主键列
+	for _, col := range table.Schema.Columns {
+		for _, opt := range col.Options {
+			if strings.ToUpper(opt) == "PRIMARY KEY" {
+				pkColumns = append(pkColumns, col.Name)
+				break
+			}
+		}
+	}
+
+	// 第二遍：构建列定义
+	columnDefs := make([]string, 0, len(table.Schema.Columns))
+	for _, col := range table.Schema.Columns {
+		colDef := fmt.Sprintf("  %s %s", col.Name, mg.getMySQLType(col))
 
-		// 添加选项
+		// 处理列选项
 		for _, opt := range col.Options {
-			sql.WriteString(" " + opt)
+			upperOpt := strings.ToUpper(opt)
+			// PRIMARY KEY 选项不在列级定义
+			if upperOpt != "PRIMARY KEY" {
+				colDef += " " + opt
+			}
+		}
+
+		// 如果是主键列，确保有 NOT NULL
+		isPrimaryKey := false
+		for _, pk := range pkColumns {
+			if pk == col.Name {
+				isPrimaryKey = true
+				break
+			}
+		}
+
+		if isPrimaryKey {
+			hasNotNull := false
+			for _, opt := range col.Options {
+				if strings.ToUpper(opt) == "NOT NULL" {
+					hasNotNull = true
+					break
+				}
+			}
+			if !hasNotNull {
+				colDef += " NOT NULL"
+			}
 		}
 
 		// 添加默认值
 		if col.Default != "" {
-			// 检查是否是函数调用（如CURRENT_TIMESTAMP）
-			if strings.Contains(strings.ToUpper(col.Default), "CURRENT_TIMESTAMP") ||
-				strings.Contains(strings.ToUpper(col.Default), "NOW()") {
-				sql.WriteString(" DEFAULT " + col.Default)
+			upperDefault := strings.ToUpper(col.Default)
+			if strings.Contains(upperDefault, "CURRENT_TIMESTAMP") ||
+				strings.Contains(upperDefault, "NOW()") ||
+				strings.Contains(upperDefault, "UUID()") ||
+				strings.Contains(upperDefault, "NULL") {
+				colDef += " DEFAULT " + col.Default
 			} else {
-				sql.WriteString(fmt.Sprintf(" DEFAULT '%s'", col.Default))
+				colDef += fmt.Sprintf(" DEFAULT '%s'", col.Default)
 			}
 		}
 
 		// 添加注释
 		if col.Comment != "" {
-			sql.WriteString(fmt.Sprintf(" COMMENT '%s'", col.Comment))
+			colDef += fmt.Sprintf(" COMMENT '%s'", escapeSingleQuote(col.Comment))
 		}
 
-		if i < len(columns)-1 {
-			sql.WriteString(",")
-		}
-		sql.WriteString("\n")
+		columnDefs = append(columnDefs, colDef)
 	}
 
-	// 添加索引（在MySQL中，索引可以在CREATE TABLE语句中定义）
+	// 将所有定义部分合并
+	allParts := make([]string, 0)
+	allParts = append(allParts, columnDefs...)
+
+	// 添加主键
+	if len(pkColumns) > 0 {
+		allParts = append(allParts, fmt.Sprintf("  PRIMARY KEY (%s)", strings.Join(pkColumns, ", ")))
+	}
+
+	// 添加索引
 	for _, idx := range table.Schema.Indexes {
 		if idx.Unique {
-			sql.WriteString(fmt.Sprintf("  ,UNIQUE KEY %s (%s)\n",
+			allParts = append(allParts, fmt.Sprintf("  UNIQUE KEY %s (%s)",
 				idx.Name, strings.Join(idx.Columns, ", ")))
 		} else {
-			sql.WriteString(fmt.Sprintf("  ,KEY %s (%s)\n",
+			allParts = append(allParts, fmt.Sprintf("  KEY %s (%s)",
 				idx.Name, strings.Join(idx.Columns, ", ")))
 		}
 	}
 
-	sql.WriteString(")")
+	// 用逗号连接所有部分
+	sql.WriteString(strings.Join(allParts, ",\n"))
+	sql.WriteString("\n)")
 
-	// 表选项
+	// 表注释
 	if table.Schema.Comment != "" {
-		sql.WriteString(fmt.Sprintf(" COMMENT='%s'", table.Schema.Comment))
+		sql.WriteString(fmt.Sprintf(" COMMENT='%s'", escapeSingleQuote(table.Schema.Comment)))
 	}
 
+	// 引擎和字符集
 	sql.WriteString(" ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci;")
 
 	return sql.String()
@@ -143,6 +190,11 @@ func (mg *MySQLGenerator) getMySQLType(col ColumnSchema) string {
 	}
 }
 
+// escapeSingleQuote 转义单引号
+func escapeSingleQuote(str string) string {
+	return strings.ReplaceAll(str, "'", "''")
+}
+
 func init() {
 	RegisterGenerator(NewMySQLGenerator())
 }

sqldef/generators/sql_generator.go

@@ -40,6 +40,7 @@ type TableDDL struct {
 	SQL     string
 	Comment string
 	Schema  *TableSchema // 新增Schema字段
+
 }
 
 // SQLGenerator SQL生成器接口

sqldef/table_create.go

@@ -18,6 +18,7 @@ type TableSyncer struct {
 
 // NewTableSyncer 创建表同步器
 func NewTableSyncer(db *sqlx.DB, dbType string) (*TableSyncer, error) {
+
 	// 获取SQL生成器
 	generator, err := generators.GetGenerator(dbType)
 	if err != nil {