Golang动态SQL创建与导入实现指南

你是否知道任何用于排序数据库表中列（字段顺序）的库？

该方法输出的格式是数组，我们需要的是JSON格式。

在Golang中实现动态SQL创建和导入，可以使用以下方案：

package main

import (
    "database/sql"
    "fmt"
    "strings"
    "github.com/xuri/excelize/v2"
    _ "github.com/go-sql-driver/mysql"
)

type DynamicImporter struct {
    db *sql.DB
}

func (di *DynamicImporter) CreateTableFromExcel(filePath, tableName string) error {
    // 读取Excel文件
    f, err := excelize.OpenFile(filePath)
    if err != nil {
        return err
    }
    defer f.Close()

    // 获取第一个工作表
    sheetName := f.GetSheetName(0)
    rows, err := f.GetRows(sheetName)
    if err != nil {
        return err
    }

    if len(rows) == 0 {
        return fmt.Errorf("Excel文件为空")
    }

    // 第一行作为列名
    headers := rows[0]
    
    // 构建CREATE TABLE语句
    var columns []string
    for _, header := range headers {
        // 清理列名，移除特殊字符
        cleanCol := strings.ReplaceAll(header, " ", "_")
        cleanCol = strings.ReplaceAll(cleanCol, "-", "_")
        columns = append(columns, fmt.Sprintf("`%s` TEXT", cleanCol))
    }

    createSQL := fmt.Sprintf(
        "CREATE TABLE IF NOT EXISTS `%s` (%s)",
        tableName,
        strings.Join(columns, ", "),
    )

    // 执行创建表
    _, err = di.db.Exec(createSQL)
    return err
}

func (di *DynamicImporter) ImportExcelData(filePath, tableName string) error {
    f, err := excelize.OpenFile(filePath)
    if err != nil {
        return err
    }
    defer f.Close()

    sheetName := f.GetSheetName(0)
    rows, err := f.GetRows(sheetName)
    if err != nil {
        return err
    }

    if len(rows) < 2 {
        return fmt.Errorf("没有数据行")
    }

    headers := rows[0]
    
    // 准备INSERT语句
    placeholders := make([]string, len(headers))
    for i := range headers {
        placeholders[i] = "?"
    }

    insertSQL := fmt.Sprintf(
        "INSERT INTO `%s` (%s) VALUES (%s)",
        tableName,
        "`" + strings.Join(headers, "`,`") + "`",
        strings.Join(placeholders, ","),
    )

    // 准备语句
    stmt, err := di.db.Prepare(insertSQL)
    if err != nil {
        return err
    }
    defer stmt.Close()

    // 插入数据行
    for _, row := range rows[1:] {
        // 确保行数据长度与列数匹配
        values := make([]interface{}, len(headers))
        for i := 0; i < len(headers); i++ {
            if i < len(row) {
                values[i] = row[i]
            } else {
                values[i] = nil
            }
        }
        
        _, err := stmt.Exec(values...)
        if err != nil {
            return err
        }
    }

    return nil
}

// 动态SELECT示例
func (di *DynamicImporter) DynamicSelect(tableName string, columns []string, condition string, args ...interface{}) (*sql.Rows, error) {
    if len(columns) == 0 {
        columns = []string{"*"}
    }
    
    query := fmt.Sprintf(
        "SELECT %s FROM `%s`",
        strings.Join(columns, ", "),
        tableName,
    )
    
    if condition != "" {
        query += " WHERE " + condition
    }
    
    return di.db.Query(query, args...)
}

func main() {
    // 数据库连接
    db, err := sql.Open("mysql", "user:password@tcp(localhost:3306)/database")
    if err != nil {
        panic(err)
    }
    defer db.Close()

    importer := &DynamicImporter{db: db}
    
    // 创建表
    err = importer.CreateTableFromExcel("data.xlsx", "dynamic_table")
    if err != nil {
        panic(err)
    }

    // 导入数据
    err = importer.ImportExcelData("data.xlsx", "dynamic_table")
    if err != nil {
        panic(err)
    }

    // 动态查询示例
    rows, err := importer.DynamicSelect("dynamic_table", []string{"column1", "column2"}, "column3 = ?", "value")
    if err != nil {
        panic(err)
    }
    defer rows.Close()
    
    // 处理查询结果
    for rows.Next() {
        var col1, col2 string
        rows.Scan(&col1, &col2)
        fmt.Println(col1, col2)
    }
}

需要安装的依赖：

go get github.com/xuri/excelize/v2
go get github.com/go-sql-driver/mysql

这个实现包含以下关键功能：

1. 动态表创建：从Excel第一行读取列名，自动生成CREATE TABLE语句
2. 数据导入：使用预处理语句批量插入数据
3. 动态查询：支持动态指定查询列和条件
4. 列名清理：自动处理Excel列名中的特殊字符

注意：实际使用时需要根据数据类型调整字段类型定义，并添加适当的错误处理和事务支持。