mirror of
https://github.com/Syngnat/GoNavi.git
synced 2026-05-13 17:29:46 +08:00
* 🐛 fix(data-viewer): 修复ClickHouse尾部分页异常并增强DuckDB复杂类型兼容 - DataViewer 新增 ClickHouse 反向分页策略,修复最后页与倒数页查询失败 - DuckDB 查询失败时按列类型生成安全 SELECT,复杂类型转 VARCHAR 重试 - 分页状态统一使用 currentPage 回填,避免页码与总数推导不一致 - 增强查询异常日志与重试路径,降低大表场景卡顿与误报 * ✨ feat(frontend-driver): 驱动管理支持快速搜索并优化信息展示 - 新增搜索框,支持按 DuckDB/ClickHouse 等关键字快速定位驱动 - 显示“匹配 x / y”统计与无结果提示 - 优化头部区域排版,提升透明/暗色场景下的视觉对齐 * 🔧 fix(connection-modal): 修复多数据源URI导入解析并校正Oracle服务名校验 - 新增单主机URI解析映射,兼容 postgres/postgresql、sqlserver、redis、tdengine、dameng(dm)、kingbase、highgo、vastbase、clickhouse、oracle - 抽取 parseSingleHostUri 复用逻辑,统一 host/port/user/password/database 回填行为 - Oracle 连接新增服务名必填校验,移除“服务名为空回退用户名”的隐式逻辑 - 连接弹窗补充 Oracle 服务名输入项与 URI 示例 * 🐛 fix(query-export): 修复查询结果导出卡住并统一按数据源能力控制导出路径 - 查询结果页导出增加稳定兜底,异常时确保 loading 关闭避免持续转圈 - DataGrid 导出逻辑按数据源能力分流,优先走后端 ExportQuery 并保留结果集导出降级 - QueryEditor 传递结果导出 SQL,保证查询结果导出范围与当前结果一致 - 后端补充 ExportData/ExportQuery 关键日志,提升导出链路可观测性 * 🐛 fix(precision): 修复查询链路与分页统计的大整数精度丢失 - 代理响应数据解码改为 UseNumber,避免默认 float64 吞精度 - 统一归一化 json.Number 与超界整数,超出 JS 安全范围转字符串 - 修复 DataViewer 总数解析,超大值不再误转 Number 参与分页 - refs #142 * 🐛 fix(driver-manager): 修复驱动管理网络告警重复并强化代理引导 - 新增下载链路域名探测,区分“GitHub可达但驱动下载链路不可达” - 网络不可达场景仅保留红色强提醒,移除重复二级告警 - 强提醒增加“打开全局代理设置”入口,优先引导使用 GoNavi 全局代理 - 统一网络检测与目录说明提示图标尺寸,修复加载期视觉不一致 - refs #141 * ♻️ refactor(frontend-interaction): 统一标签拖拽与暗色主题交互实现 - 重构Tab拖拽排序实现,统一为可配置拖拽引擎 - 规范拖拽与点击事件边界,提升交互一致性 - 统一多组件暗色透明样式策略,减少硬编码色值 - 提升Redis/表格/连接面板在透明模式下的观感一致性 - refs #144 * ♻️ refactor(update-state): 重构在线更新状态流并按版本统一进度展示 - 重构更新检查与下载状态同步流程,减少前后端状态分叉 - 进度展示严格绑定 latestVersion,避免跨版本状态串用 - 优化 about 打开场景的静默检查状态回填逻辑 - 统一下载弹窗关闭/后台隐藏行为 - 保持现有安装流程并补齐目录打开能力 * 🎨 style(sidebar-log): 将SQL执行日志入口调整为悬浮胶囊样式 - 移除侧栏底部整条日志入口容器 - 新增悬浮按钮阴影/边框/透明背景并适配明暗主题 - 为树区域预留底部空间避免入口遮挡内容 * ✨ feat(redis-cluster): 支持集群模式逻辑多库隔离与 0-15 库切换 - 前端恢复 Redis 集群场景下 db0-db15 的数据库选择与展示 - 后端新增集群逻辑库命名空间前缀映射,统一 key/pattern 读写隔离 - 覆盖扫描、读取、写入、删除、重命名等核心操作的键映射规则 - 集群命令通道支持 SELECT 逻辑切库与 FLUSHDB 逻辑库清空 - refs #145 * ✨ feat(DataGrid): 大数据表虚拟滚动性能优化及UI一致性修复 - 启用动态虚拟滚动(数据量≥500行自动切换),解决万行数据表卡顿问题 - 虚拟模式下EditableCell改用div渲染,CSS选择器从元素级改为类级适配虚拟DOM - 修复虚拟模式双水平滚动条:样式化rc-virtual-list内置滚动条为胶囊外观,禁用自定义外部滚动条 - 为rc-virtual-list水平滚动条添加鼠标滚轮支持(MutationObserver + marginLeft驱动) - 修复白色主题透明模式下列名悬浮Tooltip对比度不足的问题 - 新增白色主题全局滚动条样式适配透明模式(App.css) - App.tsx主题token与组件样式优化 - refs #147 * 🔧 chore(app): 清理 App.tsx 类型告警并收敛前端壳层实现 - 清除未使用代码和冗余状态 - 替换弃用 API 以消除 IDE 提示 - 显式处理浮动 Promise 避免告警 - 保持现有更新检查和代理设置行为不变 * 🔧 fix(ci): 修复 Windows AMD64 下 DuckDB 驱动构建链路 - 将 DuckDB 工具链准备切换为优先使用 MSYS2 - 增加 gcc 和 g++ 存在性校验与版本验证 - 在 MSYS2 异常时回退 Chocolatey 安装 MinGW - 保持 Windows ARM64 跳过 DuckDB 构建与平台支持一致 * 🔧 fix(ci): 修复 Windows AMD64 下 DuckDB 驱动构建链路 - 将 DuckDB 工具链准备切换为优先使用 MSYS2 - 增加 gcc 和 g++ 存在性校验与版本验证 - 在 MSYS2 异常时回退 Chocolatey 安装 MinGW - 保持 Windows ARM64 跳过 DuckDB 构建与平台支持一致 * 🔧 fix(ci): 修复 Windows AMD64 下 DuckDB 驱动构建工具链 - 将 DuckDB 编译链从 MINGW64 切换为 MSYS2 UCRT64 - 修正 Windows AMD64 的 gcc 和 g++ 探测路径 - 增加 DuckDB 编译器版本校验步骤 * 📝 docs(contributing): 补充中英文贡献指南并统一 README 入口 - 新增英文版 CONTRIBUTING.md 作为正式贡献文档 - 新增中文版 CONTRIBUTING.zh-CN.md 作为中文贡献说明 - 调整 README 和 README.zh-CN 的贡献入口指向对应语言文档 * - feat(connection,metadata,kingbase): 增强多数据源连接能力并修复金仓/达梦/Oracle/ClickHouse兼容性问题 (#188) (#190) * feat(http-tunnel): 支持独立 HTTP 隧道连接并覆盖多数据源 refs #168 * fix(kingbase-data-grid): 修复金仓打开表卡顿并降低对象渲染开销 refs #178 * fix(kingbase-transaction): 修复金仓事务提交重复引号导致语法错误 refs #176 * fix(driver-agent): 修复老版本 Win10 升级后金仓驱动代理启动失败 refs #177 * chore(ci): 新增手动触发的 macOS 测试构建工作流 * chore(ci): 允许测试工作流在当前分支自动触发 * fix(query-editor): 修复 SQL 编辑中光标随机跳到末尾 refs #185 * feat(data-sync): 增加差异 SQL 预览能力便于审核 refs #174 * fix(clickhouse-connect): 自动识别并回退 HTTP/Native 协议连接 refs #181 * fix(oracle-metadata): 修复视图与函数加载按 schema 过滤异常 refs #155 * fix(dameng-databases): 修复显示全部库时数据库列表不完整 refs #154 * fix(connection,db-list): 统一处理空列表返回并修复达梦连接测试报错 refs #157 Co-authored-by: 辣条 <69459608+tianqijiuyun-latiao@users.noreply.github.com> * ✨ feat(release-notes): 支持自动生成 Release 更新说明并区分配置文件命名 * 🔁 chore(sync): 回灌 main 到 dev (#192) * - feat(connection,metadata,kingbase): 增强多数据源连接能力并修复金仓/达梦/Oracle/ClickHouse兼容性问题 (#188) * feat(http-tunnel): 支持独立 HTTP 隧道连接并覆盖多数据源 refs #168 * fix(kingbase-data-grid): 修复金仓打开表卡顿并降低对象渲染开销 refs #178 * fix(kingbase-transaction): 修复金仓事务提交重复引号导致语法错误 refs #176 * fix(driver-agent): 修复老版本 Win10 升级后金仓驱动代理启动失败 refs #177 * chore(ci): 新增手动触发的 macOS 测试构建工作流 * chore(ci): 允许测试工作流在当前分支自动触发 * fix(query-editor): 修复 SQL 编辑中光标随机跳到末尾 refs #185 * feat(data-sync): 增加差异 SQL 预览能力便于审核 refs #174 * fix(clickhouse-connect): 自动识别并回退 HTTP/Native 协议连接 refs #181 * fix(oracle-metadata): 修复视图与函数加载按 schema 过滤异常 refs #155 * fix(dameng-databases): 修复显示全部库时数据库列表不完整 refs #154 * fix(connection,db-list): 统一处理空列表返回并修复达梦连接测试报错 refs #157 * Release/0.5.3 (#191) --------- Co-authored-by: 辣条 <69459608+tianqijiuyun-latiao@users.noreply.github.com> Co-authored-by: Syngnat <92659908+Syngnat@users.noreply.github.com> * 🐛 fix(branch-sync): 修复 main 回灌 dev 时 mergeable 异步计算导致漏开自动合并 - 增加 mergeable 状态轮询,避免新建同步 PR 后立即返回 UNKNOWN - 在合并状态未稳定时输出中文告警与执行摘要 - 保持冲突分支、待计算分支与自动合并分支的处理路径清晰 * 🔁 chore(sync): 回灌 main 到 dev (#195) * - feat(connection,metadata,kingbase): 增强多数据源连接能力并修复金仓/达梦/Oracle/ClickHouse兼容性问题 (#188) * feat(http-tunnel): 支持独立 HTTP 隧道连接并覆盖多数据源 refs #168 * fix(kingbase-data-grid): 修复金仓打开表卡顿并降低对象渲染开销 refs #178 * fix(kingbase-transaction): 修复金仓事务提交重复引号导致语法错误 refs #176 * fix(driver-agent): 修复老版本 Win10 升级后金仓驱动代理启动失败 refs #177 * chore(ci): 新增手动触发的 macOS 测试构建工作流 * chore(ci): 允许测试工作流在当前分支自动触发 * fix(query-editor): 修复 SQL 编辑中光标随机跳到末尾 refs #185 * feat(data-sync): 增加差异 SQL 预览能力便于审核 refs #174 * fix(clickhouse-connect): 自动识别并回退 HTTP/Native 协议连接 refs #181 * fix(oracle-metadata): 修复视图与函数加载按 schema 过滤异常 refs #155 * fix(dameng-databases): 修复显示全部库时数据库列表不完整 refs #154 * fix(connection,db-list): 统一处理空列表返回并修复达梦连接测试报错 refs #157 * Release/0.5.3 (#191) * - chore(ci): 新增全平台测试包手动构建工作流 tianqijiuyun-latiao 今天 下午4:26 (#194) * feat(http-tunnel): 支持独立 HTTP 隧道连接并覆盖多数据源 refs #168 * fix(kingbase-data-grid): 修复金仓打开表卡顿并降低对象渲染开销 refs #178 * fix(kingbase-transaction): 修复金仓事务提交重复引号导致语法错误 refs #176 * fix(driver-agent): 修复老版本 Win10 升级后金仓驱动代理启动失败 refs #177 * chore(ci): 新增手动触发的 macOS 测试构建工作流 * chore(ci): 允许测试工作流在当前分支自动触发 * fix(query-editor): 修复 SQL 编辑中光标随机跳到末尾 refs #185 * feat(data-sync): 增加差异 SQL 预览能力便于审核 refs #174 * fix(clickhouse-connect): 自动识别并回退 HTTP/Native 协议连接 refs #181 * fix(oracle-metadata): 修复视图与函数加载按 schema 过滤异常 refs #155 * fix(dameng-databases): 修复显示全部库时数据库列表不完整 refs #154 * fix(connection,db-list): 统一处理空列表返回并修复达梦连接测试报错 refs #157 * fix(kingbase): 补齐主键识别并优化宽表卡顿 refs #176 refs #178 * fix(query-execution): 支持带前置注释的读查询结果识别 * chore(ci): 新增全平台测试包手动构建工作流 --------- Co-authored-by: 辣条 <69459608+tianqijiuyun-latiao@users.noreply.github.com> Co-authored-by: Syngnat <92659908+Syngnat@users.noreply.github.com> * ♻️ refactor(frontend-sync): 优化桌面交互细节并移除 main 回灌 dev 自动化 - 优化新建连接、主题设置、侧边栏工具区与 SQL 日志的界面表现 - 调整分页、筛选、透明模式与弹窗样式,统一整体交互层次 - 收口外观参数生效逻辑并补齐多组件适配 - 删除 sync-main-to-dev 工作流并同步维护者手动回灌说明 * feat: 统一筛选条件逻辑按钮宽度 (#201) * 🐛 fix(oracle-query): 修复 Oracle 表数据分页 SQL 兼容问题 refs #196 (#202) * ✨ feat(datasource): 支持 DuckDB Parquet 文件模式并优化弹窗打开链路 - 统一 DuckDB 文件库与 Parquet 文件接入能力 - 补充 URI、文件选择、只读挂载与连接缓存键处理 - 去掉数据源卡片点击前的同步驱动查询,修复打开卡顿 * ✨ feat(datasource): 支持 DuckDB Parquet 文件模式并优化弹窗打开链路 - 统一 DuckDB 文件库与 Parquet 文件接入能力 - 补充 URI、文件选择、只读挂载与连接缓存键处理 - 去掉数据源卡片点击前的同步驱动查询,修复打开卡顿 - refs #166 * 🐛 fix(dameng): 修复达梦连接成功后数据库列表为空问题 - 调整达梦数据库列表获取策略,优先回退查询当前 schema 与当前用户 - 保留可见用户与 owner 聚合逻辑,兼容低权限账号场景 - 补充前端空列表提示与后端单元测试,降低排查成本 - close #203 * ✨ feat(data-sync): 扩展跨库迁移链路并优化数据同步交互 - 统一同库同步与跨库迁移入口,补充模式区分与风险提示 - 扩展 ClickHouse 与 PG-like 双向迁移,并新增 PG-like、ClickHouse、TDengine 到 MongoDB 的迁移路由 - 完善 TDengine 目标端建表规划、回归测试与需求追踪文档 - refs #51 --------- Co-authored-by: Syngnat <yangguofeng919@gmail.com> Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com> Co-authored-by: 辣条 <69459608+tianqijiuyun-latiao@users.noreply.github.com> Co-authored-by: TSS <266256496+Zencok@users.noreply.github.com>
1015 lines
38 KiB
Go
1015 lines
38 KiB
Go
package sync
|
||
|
||
import (
|
||
"GoNavi-Wails/internal/connection"
|
||
"GoNavi-Wails/internal/db"
|
||
"fmt"
|
||
"regexp"
|
||
"sort"
|
||
"strconv"
|
||
"strings"
|
||
)
|
||
|
||
type SchemaMigrationPlan struct {
|
||
SourceSchema string
|
||
SourceTable string
|
||
SourceQueryTable string
|
||
TargetSchema string
|
||
TargetTable string
|
||
TargetQueryTable string
|
||
TargetTableExists bool
|
||
AutoCreate bool
|
||
PlannedAction string
|
||
Warnings []string
|
||
UnsupportedObjects []string
|
||
IndexesToCreate int
|
||
IndexesSkipped int
|
||
CreateTableSQL string
|
||
PreDataSQL []string
|
||
PostDataSQL []string
|
||
}
|
||
|
||
type groupedIndex struct {
|
||
Name string
|
||
Columns []string
|
||
Unique bool
|
||
IndexType string
|
||
SubPart int
|
||
}
|
||
|
||
func normalizeTargetTableStrategy(strategy string) string {
|
||
switch strings.ToLower(strings.TrimSpace(strategy)) {
|
||
case "smart":
|
||
return "smart"
|
||
case "auto_create_if_missing":
|
||
return "auto_create_if_missing"
|
||
case "existing_only", "":
|
||
return "existing_only"
|
||
default:
|
||
return "existing_only"
|
||
}
|
||
}
|
||
|
||
func supportsAutoCreateMigration(sourceType, targetType string) bool {
|
||
return normalizeMigrationDBType(sourceType) == "mysql" && normalizeMigrationDBType(targetType) == "kingbase"
|
||
}
|
||
|
||
func inspectTableColumns(database db.Database, schema, table string) ([]connection.ColumnDefinition, bool, error) {
|
||
cols, err := database.GetColumns(schema, table)
|
||
if err != nil {
|
||
if isLikelyTableNotFound(err) {
|
||
return nil, false, nil
|
||
}
|
||
return nil, false, err
|
||
}
|
||
if len(cols) == 0 {
|
||
return cols, false, nil
|
||
}
|
||
return cols, true, nil
|
||
}
|
||
|
||
func isLikelyTableNotFound(err error) bool {
|
||
if err == nil {
|
||
return false
|
||
}
|
||
text := strings.ToLower(strings.TrimSpace(err.Error()))
|
||
if text == "" {
|
||
return false
|
||
}
|
||
keywords := []string{
|
||
"doesn't exist",
|
||
"does not exist",
|
||
"not exist",
|
||
"unknown table",
|
||
"未找到表",
|
||
"不存在",
|
||
"invalid object",
|
||
"relation",
|
||
}
|
||
for _, keyword := range keywords {
|
||
if strings.Contains(text, keyword) {
|
||
return true
|
||
}
|
||
}
|
||
return false
|
||
}
|
||
|
||
func buildSchemaMigrationPlanLegacy(config SyncConfig, tableName string, sourceDB db.Database, targetDB db.Database) (SchemaMigrationPlan, []connection.ColumnDefinition, []connection.ColumnDefinition, error) {
|
||
plan := SchemaMigrationPlan{}
|
||
sourceType := resolveMigrationDBType(config.SourceConfig)
|
||
targetType := resolveMigrationDBType(config.TargetConfig)
|
||
plan.SourceSchema, plan.SourceTable = normalizeSchemaAndTable(sourceType, config.SourceConfig.Database, tableName)
|
||
plan.TargetSchema, plan.TargetTable = normalizeSchemaAndTable(targetType, config.TargetConfig.Database, tableName)
|
||
plan.SourceQueryTable = qualifiedNameForQuery(sourceType, plan.SourceSchema, plan.SourceTable, tableName)
|
||
plan.TargetQueryTable = qualifiedNameForQuery(targetType, plan.TargetSchema, plan.TargetTable, tableName)
|
||
plan.PlannedAction = "使用已有目标表导入"
|
||
if targetType == "tdengine" {
|
||
plan.Warnings = append(plan.Warnings, "TDengine 目标端当前仅支持 INSERT 写入;若存在差异更新/删除,执行期会被拒绝,请优先使用仅插入或全量覆盖模式")
|
||
}
|
||
|
||
sourceCols, sourceExists, err := inspectTableColumns(sourceDB, plan.SourceSchema, plan.SourceTable)
|
||
if err != nil {
|
||
return plan, nil, nil, fmt.Errorf("获取源表字段失败: %w", err)
|
||
}
|
||
if !sourceExists {
|
||
return plan, nil, nil, fmt.Errorf("源表不存在或无列定义: %s", tableName)
|
||
}
|
||
|
||
targetCols, targetExists, err := inspectTableColumns(targetDB, plan.TargetSchema, plan.TargetTable)
|
||
if err != nil {
|
||
return plan, sourceCols, nil, fmt.Errorf("获取目标表字段失败: %w", err)
|
||
}
|
||
plan.TargetTableExists = targetExists
|
||
|
||
strategy := normalizeTargetTableStrategy(config.TargetTableStrategy)
|
||
if targetExists {
|
||
missing := diffMissingColumnNames(sourceCols, targetCols)
|
||
if len(missing) > 0 {
|
||
plan.Warnings = append(plan.Warnings, fmt.Sprintf("目标表缺失字段 %d 个:%s", len(missing), strings.Join(missing, ", ")))
|
||
}
|
||
if config.AutoAddColumns && isMySQLLikeSourceType(sourceType) && normalizeMigrationDBType(targetType) == "kingbase" {
|
||
addSQL, addWarnings := buildMySQLToKingbaseAddColumnSQL(plan.TargetQueryTable, sourceCols, targetCols)
|
||
plan.PreDataSQL = append(plan.PreDataSQL, addSQL...)
|
||
plan.Warnings = append(plan.Warnings, addWarnings...)
|
||
if len(addSQL) > 0 {
|
||
plan.PlannedAction = fmt.Sprintf("补齐缺失字段(%d)后导入", len(addSQL))
|
||
}
|
||
}
|
||
if strategy != "existing_only" {
|
||
plan.Warnings = append(plan.Warnings, "目标表已存在,当前仅执行数据导入;不会自动重建已有索引/约束")
|
||
}
|
||
return dedupeSchemaMigrationPlan(plan), sourceCols, targetCols, nil
|
||
}
|
||
|
||
switch strategy {
|
||
case "existing_only":
|
||
plan.PlannedAction = "目标表不存在,需先手工创建"
|
||
plan.Warnings = append(plan.Warnings, "当前策略要求目标表已存在,执行时不会自动建表")
|
||
return dedupeSchemaMigrationPlan(plan), sourceCols, targetCols, nil
|
||
case "smart", "auto_create_if_missing":
|
||
if !supportsAutoCreateMigration(config.SourceConfig.Type, config.TargetConfig.Type) {
|
||
plan.PlannedAction = "当前库对暂不支持自动建表"
|
||
plan.Warnings = append(plan.Warnings, fmt.Sprintf("当前仅支持 MySQL -> Kingbase 自动建表,当前组合=%s -> %s", config.SourceConfig.Type, config.TargetConfig.Type))
|
||
return dedupeSchemaMigrationPlan(plan), sourceCols, targetCols, nil
|
||
}
|
||
plan.AutoCreate = true
|
||
plan.PlannedAction = "目标表不存在,将自动建表后导入"
|
||
createSQL, postSQL, warnings, unsupported, idxCreate, idxSkip, err := buildMySQLToKingbaseCreateTablePlan(config, plan.TargetQueryTable, sourceCols, sourceDB, plan.SourceSchema, plan.SourceTable)
|
||
if err != nil {
|
||
return plan, sourceCols, targetCols, err
|
||
}
|
||
plan.CreateTableSQL = createSQL
|
||
plan.PostDataSQL = append(plan.PostDataSQL, postSQL...)
|
||
plan.Warnings = append(plan.Warnings, warnings...)
|
||
plan.UnsupportedObjects = append(plan.UnsupportedObjects, unsupported...)
|
||
plan.IndexesToCreate = idxCreate
|
||
plan.IndexesSkipped = idxSkip
|
||
return dedupeSchemaMigrationPlan(plan), sourceCols, targetCols, nil
|
||
default:
|
||
return dedupeSchemaMigrationPlan(plan), sourceCols, targetCols, nil
|
||
}
|
||
}
|
||
|
||
func dedupeSchemaMigrationPlan(plan SchemaMigrationPlan) SchemaMigrationPlan {
|
||
plan.Warnings = dedupeStrings(plan.Warnings)
|
||
plan.UnsupportedObjects = dedupeStrings(plan.UnsupportedObjects)
|
||
return plan
|
||
}
|
||
|
||
func dedupeStrings(items []string) []string {
|
||
if len(items) == 0 {
|
||
return items
|
||
}
|
||
seen := make(map[string]struct{}, len(items))
|
||
out := make([]string, 0, len(items))
|
||
for _, item := range items {
|
||
text := strings.TrimSpace(item)
|
||
if text == "" {
|
||
continue
|
||
}
|
||
if _, ok := seen[text]; ok {
|
||
continue
|
||
}
|
||
seen[text] = struct{}{}
|
||
out = append(out, text)
|
||
}
|
||
return out
|
||
}
|
||
|
||
func diffMissingColumnNames(sourceCols, targetCols []connection.ColumnDefinition) []string {
|
||
if len(sourceCols) == 0 {
|
||
return nil
|
||
}
|
||
targetSet := make(map[string]struct{}, len(targetCols))
|
||
for _, col := range targetCols {
|
||
key := strings.ToLower(strings.TrimSpace(col.Name))
|
||
if key == "" {
|
||
continue
|
||
}
|
||
targetSet[key] = struct{}{}
|
||
}
|
||
missing := make([]string, 0)
|
||
for _, col := range sourceCols {
|
||
key := strings.ToLower(strings.TrimSpace(col.Name))
|
||
if key == "" {
|
||
continue
|
||
}
|
||
if _, ok := targetSet[key]; ok {
|
||
continue
|
||
}
|
||
missing = append(missing, col.Name)
|
||
}
|
||
sort.Strings(missing)
|
||
return missing
|
||
}
|
||
|
||
func buildMySQLToKingbaseAddColumnSQL(targetQueryTable string, sourceCols, targetCols []connection.ColumnDefinition) ([]string, []string) {
|
||
targetSet := make(map[string]struct{}, len(targetCols))
|
||
for _, col := range targetCols {
|
||
key := strings.ToLower(strings.TrimSpace(col.Name))
|
||
if key == "" {
|
||
continue
|
||
}
|
||
targetSet[key] = struct{}{}
|
||
}
|
||
|
||
var sqlList []string
|
||
var warnings []string
|
||
for _, col := range sourceCols {
|
||
key := strings.ToLower(strings.TrimSpace(col.Name))
|
||
if key == "" {
|
||
continue
|
||
}
|
||
if _, ok := targetSet[key]; ok {
|
||
continue
|
||
}
|
||
colType, _, mapWarnings := mapMySQLColumnToKingbase(col)
|
||
warnings = append(warnings, mapWarnings...)
|
||
if col.Extra != "" && strings.Contains(strings.ToLower(col.Extra), "auto_increment") {
|
||
warnings = append(warnings, fmt.Sprintf("字段 %s 为自增列,补齐到已有目标表时不会自动补建 identity/sequence", col.Name))
|
||
}
|
||
sqlList = append(sqlList, fmt.Sprintf("ALTER TABLE %s ADD COLUMN %s %s NULL",
|
||
quoteQualifiedIdentByType("kingbase", targetQueryTable),
|
||
quoteIdentByType("kingbase", col.Name),
|
||
colType,
|
||
))
|
||
}
|
||
return sqlList, dedupeStrings(warnings)
|
||
}
|
||
|
||
func buildMySQLToKingbaseCreateTablePlan(config SyncConfig, targetQueryTable string, sourceCols []connection.ColumnDefinition, sourceDB db.Database, sourceSchema, sourceTable string) (string, []string, []string, []string, int, int, error) {
|
||
columnDefs := make([]string, 0, len(sourceCols)+1)
|
||
warnings := make([]string, 0)
|
||
unsupported := make([]string, 0)
|
||
pkCols := make([]string, 0, 2)
|
||
|
||
for _, col := range sourceCols {
|
||
def, colWarnings := buildMySQLToKingbaseColumnDefinition(col)
|
||
warnings = append(warnings, colWarnings...)
|
||
columnDefs = append(columnDefs, fmt.Sprintf("%s %s", quoteIdentByType("kingbase", col.Name), def))
|
||
if col.Key == "PRI" || col.Key == "PK" {
|
||
pkCols = append(pkCols, quoteIdentByType("kingbase", col.Name))
|
||
}
|
||
}
|
||
if len(pkCols) > 0 {
|
||
columnDefs = append(columnDefs, fmt.Sprintf("PRIMARY KEY (%s)", strings.Join(pkCols, ", ")))
|
||
}
|
||
createSQL := fmt.Sprintf("CREATE TABLE %s (\n %s\n)", quoteQualifiedIdentByType("kingbase", targetQueryTable), strings.Join(columnDefs, ",\n "))
|
||
|
||
if !config.CreateIndexes {
|
||
return createSQL, nil, dedupeStrings(warnings), dedupeStrings(unsupported), 0, 0, nil
|
||
}
|
||
|
||
indexes, err := sourceDB.GetIndexes(sourceSchema, sourceTable)
|
||
if err != nil {
|
||
warnings = append(warnings, fmt.Sprintf("读取源表索引失败,已跳过索引迁移:%v", err))
|
||
return createSQL, nil, dedupeStrings(warnings), dedupeStrings(unsupported), 0, 0, nil
|
||
}
|
||
grouped := groupIndexDefinitions(indexes)
|
||
postSQL := make([]string, 0, len(grouped))
|
||
created := 0
|
||
skipped := 0
|
||
for _, idx := range grouped {
|
||
name := strings.TrimSpace(idx.Name)
|
||
if name == "" || strings.EqualFold(name, "primary") {
|
||
continue
|
||
}
|
||
if len(idx.Columns) == 0 {
|
||
skipped++
|
||
unsupported = append(unsupported, fmt.Sprintf("索引 %s 缺少列定义,已跳过", name))
|
||
continue
|
||
}
|
||
kind := strings.ToLower(strings.TrimSpace(idx.IndexType))
|
||
if idx.SubPart > 0 {
|
||
skipped++
|
||
unsupported = append(unsupported, fmt.Sprintf("索引 %s 使用前缀长度,当前暂不支持迁移", name))
|
||
continue
|
||
}
|
||
if kind != "" && kind != "btree" {
|
||
skipped++
|
||
unsupported = append(unsupported, fmt.Sprintf("索引 %s 类型=%s,当前暂不支持自动迁移", name, idx.IndexType))
|
||
continue
|
||
}
|
||
quotedCols := make([]string, 0, len(idx.Columns))
|
||
for _, col := range idx.Columns {
|
||
quotedCols = append(quotedCols, quoteIdentByType("kingbase", col))
|
||
}
|
||
prefix := "CREATE INDEX"
|
||
if idx.Unique {
|
||
prefix = "CREATE UNIQUE INDEX"
|
||
}
|
||
postSQL = append(postSQL, fmt.Sprintf("%s %s ON %s (%s)", prefix, quoteIdentByType("kingbase", name), quoteQualifiedIdentByType("kingbase", targetQueryTable), strings.Join(quotedCols, ", ")))
|
||
created++
|
||
}
|
||
return createSQL, postSQL, dedupeStrings(warnings), dedupeStrings(unsupported), created, skipped, nil
|
||
}
|
||
|
||
func buildMySQLToKingbaseColumnDefinition(col connection.ColumnDefinition) (string, []string) {
|
||
targetType, useIdentity, warnings := mapMySQLColumnToKingbase(col)
|
||
parts := []string{targetType}
|
||
if useIdentity {
|
||
parts = append(parts, "GENERATED BY DEFAULT AS IDENTITY")
|
||
}
|
||
if !useIdentity {
|
||
if defaultSQL, ok, warningText := mapMySQLDefaultToKingbase(col, targetType); warningText != "" {
|
||
warnings = append(warnings, warningText)
|
||
} else if ok {
|
||
parts = append(parts, "DEFAULT "+defaultSQL)
|
||
}
|
||
}
|
||
if strings.EqualFold(strings.TrimSpace(col.Nullable), "NO") {
|
||
parts = append(parts, "NOT NULL")
|
||
}
|
||
return strings.Join(parts, " "), dedupeStrings(warnings)
|
||
}
|
||
|
||
func mapMySQLColumnToKingbase(col connection.ColumnDefinition) (string, bool, []string) {
|
||
raw := strings.ToLower(strings.TrimSpace(col.Type))
|
||
warnings := make([]string, 0)
|
||
if raw == "" {
|
||
return "text", false, []string{fmt.Sprintf("字段 %s 类型为空,已降级为 text", col.Name)}
|
||
}
|
||
unsigned := strings.Contains(raw, "unsigned")
|
||
clean := strings.ReplaceAll(raw, " unsigned", "")
|
||
clean = strings.ReplaceAll(clean, " zerofill", "")
|
||
isAutoIncrement := strings.Contains(strings.ToLower(strings.TrimSpace(col.Extra)), "auto_increment")
|
||
|
||
switch {
|
||
case strings.HasPrefix(clean, "tinyint(1)") && !unsigned && !isAutoIncrement:
|
||
return "boolean", false, warnings
|
||
case strings.HasPrefix(clean, "tinyint"):
|
||
return ternaryString(unsigned, "smallint", "smallint"), false, warnings
|
||
case strings.HasPrefix(clean, "smallint"):
|
||
return ternaryString(unsigned, "integer", "smallint"), isAutoIncrement, warnings
|
||
case strings.HasPrefix(clean, "mediumint"):
|
||
return ternaryString(unsigned, "bigint", "integer"), isAutoIncrement, warnings
|
||
case strings.HasPrefix(clean, "int") || strings.HasPrefix(clean, "integer"):
|
||
return ternaryString(unsigned, "bigint", "integer"), isAutoIncrement, warnings
|
||
case strings.HasPrefix(clean, "bigint"):
|
||
if unsigned {
|
||
if isAutoIncrement {
|
||
warnings = append(warnings, fmt.Sprintf("字段 %s 为 unsigned bigint auto_increment,已降级为 numeric(20,0) 且不保留自增语义", col.Name))
|
||
}
|
||
return "numeric(20,0)", false, warnings
|
||
}
|
||
return "bigint", isAutoIncrement, warnings
|
||
case strings.HasPrefix(clean, "decimal"), strings.HasPrefix(clean, "numeric"):
|
||
return replaceTypeBase(clean, []string{"decimal", "numeric"}, "numeric"), false, warnings
|
||
case strings.HasPrefix(clean, "float"):
|
||
return "real", false, warnings
|
||
case strings.HasPrefix(clean, "double"):
|
||
return "double precision", false, warnings
|
||
case strings.HasPrefix(clean, "bit("):
|
||
if clean == "bit(1)" {
|
||
return "boolean", false, warnings
|
||
}
|
||
return clean, false, warnings
|
||
case strings.HasPrefix(clean, "bool"), strings.HasPrefix(clean, "boolean"):
|
||
return "boolean", false, warnings
|
||
case strings.HasPrefix(clean, "char("), strings.HasPrefix(clean, "varchar("):
|
||
return clean, false, warnings
|
||
case strings.HasPrefix(clean, "tinytext"), strings.HasPrefix(clean, "text"), strings.HasPrefix(clean, "mediumtext"), strings.HasPrefix(clean, "longtext"):
|
||
return "text", false, warnings
|
||
case strings.HasPrefix(clean, "json"):
|
||
return "jsonb", false, warnings
|
||
case strings.HasPrefix(clean, "date"):
|
||
return "date", false, warnings
|
||
case strings.HasPrefix(clean, "time"):
|
||
return "time", false, warnings
|
||
case strings.HasPrefix(clean, "datetime"), strings.HasPrefix(clean, "timestamp"):
|
||
return "timestamp", false, warnings
|
||
case strings.HasPrefix(clean, "year"):
|
||
warnings = append(warnings, fmt.Sprintf("字段 %s 类型 year 已映射为 integer", col.Name))
|
||
return "integer", false, warnings
|
||
case strings.HasPrefix(clean, "binary"), strings.HasPrefix(clean, "varbinary"), strings.HasPrefix(clean, "tinyblob"), strings.HasPrefix(clean, "blob"), strings.HasPrefix(clean, "mediumblob"), strings.HasPrefix(clean, "longblob"):
|
||
return "bytea", false, warnings
|
||
case strings.HasPrefix(clean, "enum"), strings.HasPrefix(clean, "set"):
|
||
warnings = append(warnings, fmt.Sprintf("字段 %s 类型 %s 已降级为 text", col.Name, col.Type))
|
||
return "text", false, warnings
|
||
default:
|
||
warnings = append(warnings, fmt.Sprintf("字段 %s 类型 %s 暂无专门映射,已降级为 text", col.Name, col.Type))
|
||
return "text", false, warnings
|
||
}
|
||
}
|
||
|
||
func replaceTypeBase(raw string, bases []string, target string) string {
|
||
for _, base := range bases {
|
||
if strings.HasPrefix(raw, base) {
|
||
return target + strings.TrimPrefix(raw, base)
|
||
}
|
||
}
|
||
return target
|
||
}
|
||
|
||
var numericPattern = regexp.MustCompile(`^[+-]?\d+(\.\d+)?$`)
|
||
|
||
func mapMySQLDefaultToKingbase(col connection.ColumnDefinition, targetType string) (string, bool, string) {
|
||
if col.Default == nil {
|
||
return "", false, ""
|
||
}
|
||
raw := strings.TrimSpace(*col.Default)
|
||
if raw == "" {
|
||
if isStringLikeTargetType(targetType) {
|
||
return "''", true, ""
|
||
}
|
||
return "", false, fmt.Sprintf("字段 %s 的空字符串默认值未保留", col.Name)
|
||
}
|
||
lower := strings.ToLower(raw)
|
||
if lower == "null" {
|
||
return "", false, ""
|
||
}
|
||
if strings.HasPrefix(lower, "current_timestamp") {
|
||
return "CURRENT_TIMESTAMP", true, ""
|
||
}
|
||
if targetType == "boolean" {
|
||
switch lower {
|
||
case "1", "true":
|
||
return "TRUE", true, ""
|
||
case "0", "false":
|
||
return "FALSE", true, ""
|
||
}
|
||
}
|
||
if numericPattern.MatchString(raw) && !isStringLikeTargetType(targetType) {
|
||
return raw, true, ""
|
||
}
|
||
if strings.ContainsAny(raw, "()") && !strings.HasPrefix(lower, "current_timestamp") {
|
||
return "", false, fmt.Sprintf("字段 %s 的默认值 %s 包含表达式,当前未自动迁移", col.Name, raw)
|
||
}
|
||
return "'" + strings.ReplaceAll(raw, "'", "''") + "'", true, ""
|
||
}
|
||
|
||
func isStringLikeTargetType(targetType string) bool {
|
||
text := strings.ToLower(strings.TrimSpace(targetType))
|
||
return strings.Contains(text, "char") || strings.Contains(text, "text") || strings.Contains(text, "json") || strings.Contains(text, "bytea")
|
||
}
|
||
|
||
func ternaryString(ok bool, a, b string) string {
|
||
if ok {
|
||
return a
|
||
}
|
||
return b
|
||
}
|
||
|
||
func groupIndexDefinitions(indexes []connection.IndexDefinition) []groupedIndex {
|
||
if len(indexes) == 0 {
|
||
return nil
|
||
}
|
||
groupMap := make(map[string][]connection.IndexDefinition)
|
||
order := make([]string, 0)
|
||
for _, idx := range indexes {
|
||
name := strings.TrimSpace(idx.Name)
|
||
if name == "" {
|
||
continue
|
||
}
|
||
if _, ok := groupMap[name]; !ok {
|
||
order = append(order, name)
|
||
}
|
||
groupMap[name] = append(groupMap[name], idx)
|
||
}
|
||
grouped := make([]groupedIndex, 0, len(groupMap))
|
||
for _, name := range order {
|
||
rows := groupMap[name]
|
||
sort.SliceStable(rows, func(i, j int) bool {
|
||
return rows[i].SeqInIndex < rows[j].SeqInIndex
|
||
})
|
||
gi := groupedIndex{Name: name, Unique: true, IndexType: "BTREE"}
|
||
for _, row := range rows {
|
||
if row.NonUnique != 0 {
|
||
gi.Unique = false
|
||
}
|
||
if strings.TrimSpace(row.IndexType) != "" {
|
||
gi.IndexType = row.IndexType
|
||
}
|
||
if row.SubPart > 0 && gi.SubPart == 0 {
|
||
gi.SubPart = row.SubPart
|
||
}
|
||
col := strings.TrimSpace(row.ColumnName)
|
||
if col != "" {
|
||
gi.Columns = append(gi.Columns, col)
|
||
}
|
||
}
|
||
grouped = append(grouped, gi)
|
||
}
|
||
return grouped
|
||
}
|
||
|
||
func intFromAny(v interface{}) int {
|
||
switch typed := v.(type) {
|
||
case int:
|
||
return typed
|
||
case int64:
|
||
return int(typed)
|
||
case float64:
|
||
return int(typed)
|
||
case string:
|
||
i, _ := strconv.Atoi(strings.TrimSpace(typed))
|
||
return i
|
||
default:
|
||
return 0
|
||
}
|
||
}
|
||
|
||
func isPGLikeSource(dbType string) bool {
|
||
switch normalizeMigrationDBType(dbType) {
|
||
case "postgres", "kingbase", "highgo", "vastbase", "duckdb":
|
||
return true
|
||
default:
|
||
return false
|
||
}
|
||
}
|
||
|
||
func buildPGLikeToMySQLPlan(config SyncConfig, tableName string, sourceDB db.Database, targetDB db.Database) (SchemaMigrationPlan, []connection.ColumnDefinition, []connection.ColumnDefinition, error) {
|
||
plan := SchemaMigrationPlan{}
|
||
sourceType := resolveMigrationDBType(config.SourceConfig)
|
||
targetType := resolveMigrationDBType(config.TargetConfig)
|
||
plan.SourceSchema, plan.SourceTable = normalizeSchemaAndTable(sourceType, config.SourceConfig.Database, tableName)
|
||
plan.TargetSchema, plan.TargetTable = normalizeSchemaAndTable(targetType, config.TargetConfig.Database, tableName)
|
||
plan.SourceQueryTable = qualifiedNameForQuery(sourceType, plan.SourceSchema, plan.SourceTable, tableName)
|
||
plan.TargetQueryTable = qualifiedNameForQuery(targetType, plan.TargetSchema, plan.TargetTable, tableName)
|
||
plan.PlannedAction = "使用已有目标表导入"
|
||
|
||
sourceCols, sourceExists, err := inspectTableColumns(sourceDB, plan.SourceSchema, plan.SourceTable)
|
||
if err != nil {
|
||
return plan, nil, nil, fmt.Errorf("获取源表字段失败: %w", err)
|
||
}
|
||
if !sourceExists {
|
||
return plan, nil, nil, fmt.Errorf("源表不存在或无列定义: %s", tableName)
|
||
}
|
||
|
||
targetCols, targetExists, err := inspectTableColumns(targetDB, plan.TargetSchema, plan.TargetTable)
|
||
if err != nil {
|
||
return plan, sourceCols, nil, fmt.Errorf("获取目标表字段失败: %w", err)
|
||
}
|
||
plan.TargetTableExists = targetExists
|
||
|
||
strategy := normalizeTargetTableStrategy(config.TargetTableStrategy)
|
||
if targetExists {
|
||
missing := diffMissingColumnNames(sourceCols, targetCols)
|
||
if len(missing) > 0 {
|
||
plan.Warnings = append(plan.Warnings, fmt.Sprintf("目标表缺失字段 %d 个:%s", len(missing), strings.Join(missing, ", ")))
|
||
}
|
||
if config.AutoAddColumns {
|
||
addSQL, addWarnings := buildPGLikeToMySQLAddColumnSQL(plan.TargetQueryTable, sourceCols, targetCols)
|
||
plan.PreDataSQL = append(plan.PreDataSQL, addSQL...)
|
||
plan.Warnings = append(plan.Warnings, addWarnings...)
|
||
if len(addSQL) > 0 {
|
||
plan.PlannedAction = fmt.Sprintf("补齐缺失字段(%d)后导入", len(addSQL))
|
||
}
|
||
}
|
||
if strategy != "existing_only" {
|
||
plan.Warnings = append(plan.Warnings, "目标表已存在,当前仅执行数据导入;不会自动重建已有索引/约束")
|
||
}
|
||
return dedupeSchemaMigrationPlan(plan), sourceCols, targetCols, nil
|
||
}
|
||
|
||
switch strategy {
|
||
case "existing_only":
|
||
plan.PlannedAction = "目标表不存在,需先手工创建"
|
||
plan.Warnings = append(plan.Warnings, "当前策略要求目标表已存在,执行时不会自动建表")
|
||
return dedupeSchemaMigrationPlan(plan), sourceCols, targetCols, nil
|
||
case "smart", "auto_create_if_missing":
|
||
plan.AutoCreate = true
|
||
plan.PlannedAction = "目标表不存在,将自动建表后导入"
|
||
createSQL, postSQL, warnings, unsupported, idxCreate, idxSkip, err := buildPGLikeToMySQLCreateTablePlan(config, plan.TargetQueryTable, sourceCols, sourceDB, plan.SourceSchema, plan.SourceTable)
|
||
if err != nil {
|
||
return plan, sourceCols, targetCols, err
|
||
}
|
||
plan.CreateTableSQL = createSQL
|
||
plan.PostDataSQL = append(plan.PostDataSQL, postSQL...)
|
||
plan.Warnings = append(plan.Warnings, warnings...)
|
||
plan.UnsupportedObjects = append(plan.UnsupportedObjects, unsupported...)
|
||
plan.IndexesToCreate = idxCreate
|
||
plan.IndexesSkipped = idxSkip
|
||
return dedupeSchemaMigrationPlan(plan), sourceCols, targetCols, nil
|
||
default:
|
||
return dedupeSchemaMigrationPlan(plan), sourceCols, targetCols, nil
|
||
}
|
||
}
|
||
|
||
func buildPGLikeToMySQLAddColumnSQL(targetQueryTable string, sourceCols, targetCols []connection.ColumnDefinition) ([]string, []string) {
|
||
targetSet := make(map[string]struct{}, len(targetCols))
|
||
for _, col := range targetCols {
|
||
key := strings.ToLower(strings.TrimSpace(col.Name))
|
||
if key == "" {
|
||
continue
|
||
}
|
||
targetSet[key] = struct{}{}
|
||
}
|
||
var sqlList []string
|
||
var warnings []string
|
||
for _, col := range sourceCols {
|
||
key := strings.ToLower(strings.TrimSpace(col.Name))
|
||
if key == "" {
|
||
continue
|
||
}
|
||
if _, ok := targetSet[key]; ok {
|
||
continue
|
||
}
|
||
colType, mapWarnings := mapPGLikeColumnToMySQL(col)
|
||
warnings = append(warnings, mapWarnings...)
|
||
if col.Extra != "" && strings.Contains(strings.ToLower(col.Extra), "auto_increment") {
|
||
warnings = append(warnings, fmt.Sprintf("字段 %s 为自增列,补齐到已有目标表时不会自动补建 AUTO_INCREMENT 属性", col.Name))
|
||
}
|
||
sqlList = append(sqlList, fmt.Sprintf("ALTER TABLE %s ADD COLUMN %s %s NULL",
|
||
quoteQualifiedIdentByType("mysql", targetQueryTable),
|
||
quoteIdentByType("mysql", col.Name),
|
||
colType,
|
||
))
|
||
}
|
||
return sqlList, dedupeStrings(warnings)
|
||
}
|
||
|
||
func buildPGLikeToMySQLCreateTablePlan(config SyncConfig, targetQueryTable string, sourceCols []connection.ColumnDefinition, sourceDB db.Database, sourceSchema, sourceTable string) (string, []string, []string, []string, int, int, error) {
|
||
columnDefs := make([]string, 0, len(sourceCols)+1)
|
||
warnings := make([]string, 0)
|
||
unsupported := make([]string, 0)
|
||
pkCols := make([]string, 0, 2)
|
||
for _, col := range sourceCols {
|
||
def, colWarnings := buildPGLikeToMySQLColumnDefinition(col)
|
||
warnings = append(warnings, colWarnings...)
|
||
columnDefs = append(columnDefs, fmt.Sprintf("%s %s", quoteIdentByType("mysql", col.Name), def))
|
||
if col.Key == "PRI" || col.Key == "PK" {
|
||
pkCols = append(pkCols, quoteIdentByType("mysql", col.Name))
|
||
}
|
||
}
|
||
if len(pkCols) > 0 {
|
||
columnDefs = append(columnDefs, fmt.Sprintf("PRIMARY KEY (%s)", strings.Join(pkCols, ", ")))
|
||
}
|
||
createSQL := fmt.Sprintf("CREATE TABLE %s (\n %s\n)", quoteQualifiedIdentByType("mysql", targetQueryTable), strings.Join(columnDefs, ",\n "))
|
||
if !config.CreateIndexes {
|
||
return createSQL, nil, dedupeStrings(warnings), dedupeStrings(unsupported), 0, 0, nil
|
||
}
|
||
indexes, err := sourceDB.GetIndexes(sourceSchema, sourceTable)
|
||
if err != nil {
|
||
warnings = append(warnings, fmt.Sprintf("读取源表索引失败,已跳过索引迁移:%v", err))
|
||
return createSQL, nil, dedupeStrings(warnings), dedupeStrings(unsupported), 0, 0, nil
|
||
}
|
||
grouped := groupIndexDefinitions(indexes)
|
||
postSQL := make([]string, 0, len(grouped))
|
||
created := 0
|
||
skipped := 0
|
||
for _, idx := range grouped {
|
||
name := strings.TrimSpace(idx.Name)
|
||
if name == "" || strings.EqualFold(name, "primary") {
|
||
continue
|
||
}
|
||
if len(idx.Columns) == 0 {
|
||
skipped++
|
||
unsupported = append(unsupported, fmt.Sprintf("索引 %s 缺少列定义,已跳过", name))
|
||
continue
|
||
}
|
||
kind := strings.ToLower(strings.TrimSpace(idx.IndexType))
|
||
if idx.SubPart > 0 {
|
||
skipped++
|
||
unsupported = append(unsupported, fmt.Sprintf("索引 %s 使用前缀长度,当前暂不支持迁移", name))
|
||
continue
|
||
}
|
||
if kind != "" && kind != "btree" {
|
||
skipped++
|
||
unsupported = append(unsupported, fmt.Sprintf("索引 %s 类型=%s,当前暂不支持自动迁移", name, idx.IndexType))
|
||
continue
|
||
}
|
||
quotedCols := make([]string, 0, len(idx.Columns))
|
||
for _, col := range idx.Columns {
|
||
quotedCols = append(quotedCols, quoteIdentByType("mysql", col))
|
||
}
|
||
prefix := "CREATE INDEX"
|
||
if idx.Unique {
|
||
prefix = "CREATE UNIQUE INDEX"
|
||
}
|
||
postSQL = append(postSQL, fmt.Sprintf("%s %s ON %s (%s)", prefix, quoteIdentByType("mysql", name), quoteQualifiedIdentByType("mysql", targetQueryTable), strings.Join(quotedCols, ", ")))
|
||
created++
|
||
}
|
||
return createSQL, postSQL, dedupeStrings(warnings), dedupeStrings(unsupported), created, skipped, nil
|
||
}
|
||
|
||
func buildPGLikeToMySQLColumnDefinition(col connection.ColumnDefinition) (string, []string) {
|
||
targetType, warnings := mapPGLikeColumnToMySQL(col)
|
||
parts := []string{targetType}
|
||
if strings.Contains(strings.ToLower(strings.TrimSpace(col.Extra)), "auto_increment") && canUseMySQLAutoIncrement(targetType) {
|
||
parts = append(parts, "AUTO_INCREMENT")
|
||
}
|
||
if defaultSQL, ok, warningText := mapPGLikeDefaultToMySQL(col, targetType); warningText != "" {
|
||
warnings = append(warnings, warningText)
|
||
} else if ok {
|
||
parts = append(parts, "DEFAULT "+defaultSQL)
|
||
}
|
||
if strings.EqualFold(strings.TrimSpace(col.Nullable), "NO") {
|
||
parts = append(parts, "NOT NULL")
|
||
}
|
||
return strings.Join(parts, " "), dedupeStrings(warnings)
|
||
}
|
||
|
||
func mapPGLikeColumnToMySQL(col connection.ColumnDefinition) (string, []string) {
|
||
raw := strings.ToLower(strings.TrimSpace(col.Type))
|
||
warnings := make([]string, 0)
|
||
if raw == "" {
|
||
return "text", []string{fmt.Sprintf("字段 %s 类型为空,已降级为 text", col.Name)}
|
||
}
|
||
switch {
|
||
case raw == "boolean" || strings.HasPrefix(raw, "bool"):
|
||
return "tinyint(1)", warnings
|
||
case raw == "smallint":
|
||
return "smallint", warnings
|
||
case raw == "integer" || raw == "int4":
|
||
return "int", warnings
|
||
case raw == "bigint" || raw == "int8":
|
||
return "bigint", warnings
|
||
case strings.HasPrefix(raw, "numeric") || strings.HasPrefix(raw, "decimal"):
|
||
return replaceTypeBase(raw, []string{"numeric", "decimal"}, "decimal"), warnings
|
||
case raw == "real" || raw == "float4":
|
||
return "float", warnings
|
||
case raw == "double precision" || raw == "float8":
|
||
return "double", warnings
|
||
case strings.HasPrefix(raw, "character varying"):
|
||
return strings.Replace(raw, "character varying", "varchar", 1), warnings
|
||
case strings.HasPrefix(raw, "character("):
|
||
return strings.Replace(raw, "character", "char", 1), warnings
|
||
case raw == "character":
|
||
return "char(1)", warnings
|
||
case raw == "text":
|
||
return "text", warnings
|
||
case raw == "json" || raw == "jsonb":
|
||
return "json", warnings
|
||
case raw == "bytea":
|
||
return "longblob", warnings
|
||
case raw == "date":
|
||
return "date", warnings
|
||
case strings.HasPrefix(raw, "time"):
|
||
return "time", warnings
|
||
case strings.HasPrefix(raw, "timestamp"):
|
||
return "datetime", warnings
|
||
case strings.HasPrefix(raw, "uuid"):
|
||
warnings = append(warnings, fmt.Sprintf("字段 %s 类型 uuid 已映射为 varchar(36)", col.Name))
|
||
return "varchar(36)", warnings
|
||
case strings.Contains(raw, "without time zone") || strings.Contains(raw, "with time zone"):
|
||
return "datetime", warnings
|
||
case strings.HasPrefix(raw, "json"):
|
||
return "json", warnings
|
||
case strings.HasSuffix(raw, "[]") || strings.HasPrefix(raw, "array"):
|
||
warnings = append(warnings, fmt.Sprintf("字段 %s 类型 %s 已降级为 json", col.Name, col.Type))
|
||
return "json", warnings
|
||
case raw == "user-defined":
|
||
warnings = append(warnings, fmt.Sprintf("字段 %s 为用户自定义类型,已降级为 text", col.Name))
|
||
return "text", warnings
|
||
default:
|
||
warnings = append(warnings, fmt.Sprintf("字段 %s 类型 %s 暂无专门映射,已降级为 text", col.Name, col.Type))
|
||
return "text", warnings
|
||
}
|
||
}
|
||
|
||
func canUseMySQLAutoIncrement(targetType string) bool {
|
||
text := strings.ToLower(strings.TrimSpace(targetType))
|
||
switch {
|
||
case strings.HasPrefix(text, "tinyint"), strings.HasPrefix(text, "smallint"), strings.HasPrefix(text, "mediumint"), strings.HasPrefix(text, "int"), strings.HasPrefix(text, "bigint"):
|
||
return true
|
||
default:
|
||
return false
|
||
}
|
||
}
|
||
|
||
func mapPGLikeDefaultToMySQL(col connection.ColumnDefinition, targetType string) (string, bool, string) {
|
||
if col.Default == nil {
|
||
return "", false, ""
|
||
}
|
||
raw := strings.TrimSpace(*col.Default)
|
||
if raw == "" || strings.EqualFold(raw, "null") {
|
||
return "", false, ""
|
||
}
|
||
lower := strings.ToLower(raw)
|
||
if strings.HasPrefix(lower, "nextval(") {
|
||
return "", false, ""
|
||
}
|
||
if strings.Contains(lower, "current_timestamp") || strings.Contains(lower, "now()") {
|
||
return "CURRENT_TIMESTAMP", true, ""
|
||
}
|
||
if targetType == "tinyint(1)" {
|
||
switch lower {
|
||
case "true", "1":
|
||
return "1", true, ""
|
||
case "false", "0":
|
||
return "0", true, ""
|
||
}
|
||
}
|
||
if numericPattern.MatchString(raw) && !isStringLikeTargetType(targetType) {
|
||
return raw, true, ""
|
||
}
|
||
if strings.ContainsAny(raw, "()") && !strings.Contains(lower, "current_timestamp") && !strings.Contains(lower, "now()") {
|
||
return "", false, fmt.Sprintf("字段 %s 的默认值 %s 包含表达式,当前未自动迁移", col.Name, raw)
|
||
}
|
||
return "'" + strings.ReplaceAll(raw, "'", "''") + "'", true, ""
|
||
}
|
||
|
||
func isPGLikeTarget(dbType string) bool {
|
||
switch normalizeMigrationDBType(dbType) {
|
||
case "postgres", "kingbase", "highgo", "vastbase", "duckdb":
|
||
return true
|
||
default:
|
||
return false
|
||
}
|
||
}
|
||
|
||
func buildMySQLToPGLikePlan(config SyncConfig, tableName string, sourceDB db.Database, targetDB db.Database) (SchemaMigrationPlan, []connection.ColumnDefinition, []connection.ColumnDefinition, error) {
|
||
plan := SchemaMigrationPlan{}
|
||
sourceType := resolveMigrationDBType(config.SourceConfig)
|
||
targetType := resolveMigrationDBType(config.TargetConfig)
|
||
plan.SourceSchema, plan.SourceTable = normalizeSchemaAndTable(sourceType, config.SourceConfig.Database, tableName)
|
||
plan.TargetSchema, plan.TargetTable = normalizeSchemaAndTable(targetType, config.TargetConfig.Database, tableName)
|
||
plan.SourceQueryTable = qualifiedNameForQuery(sourceType, plan.SourceSchema, plan.SourceTable, tableName)
|
||
plan.TargetQueryTable = qualifiedNameForQuery(targetType, plan.TargetSchema, plan.TargetTable, tableName)
|
||
plan.PlannedAction = "使用已有目标表导入"
|
||
|
||
sourceCols, sourceExists, err := inspectTableColumns(sourceDB, plan.SourceSchema, plan.SourceTable)
|
||
if err != nil {
|
||
return plan, nil, nil, fmt.Errorf("获取源表字段失败: %w", err)
|
||
}
|
||
if !sourceExists {
|
||
return plan, nil, nil, fmt.Errorf("源表不存在或无列定义: %s", tableName)
|
||
}
|
||
|
||
targetCols, targetExists, err := inspectTableColumns(targetDB, plan.TargetSchema, plan.TargetTable)
|
||
if err != nil {
|
||
return plan, sourceCols, nil, fmt.Errorf("获取目标表字段失败: %w", err)
|
||
}
|
||
plan.TargetTableExists = targetExists
|
||
|
||
strategy := normalizeTargetTableStrategy(config.TargetTableStrategy)
|
||
if targetExists {
|
||
missing := diffMissingColumnNames(sourceCols, targetCols)
|
||
if len(missing) > 0 {
|
||
plan.Warnings = append(plan.Warnings, fmt.Sprintf("目标表缺失字段 %d 个:%s", len(missing), strings.Join(missing, ", ")))
|
||
}
|
||
if config.AutoAddColumns {
|
||
addSQL, addWarnings := buildMySQLToPGLikeAddColumnSQL(targetType, plan.TargetQueryTable, sourceCols, targetCols)
|
||
plan.PreDataSQL = append(plan.PreDataSQL, addSQL...)
|
||
plan.Warnings = append(plan.Warnings, addWarnings...)
|
||
if len(addSQL) > 0 {
|
||
plan.PlannedAction = fmt.Sprintf("补齐缺失字段(%d)后导入", len(addSQL))
|
||
}
|
||
}
|
||
if strategy != "existing_only" {
|
||
plan.Warnings = append(plan.Warnings, "目标表已存在,当前仅执行数据导入;不会自动重建已有索引/约束")
|
||
}
|
||
return dedupeSchemaMigrationPlan(plan), sourceCols, targetCols, nil
|
||
}
|
||
|
||
switch strategy {
|
||
case "existing_only":
|
||
plan.PlannedAction = "目标表不存在,需先手工创建"
|
||
plan.Warnings = append(plan.Warnings, "当前策略要求目标表已存在,执行时不会自动建表")
|
||
return dedupeSchemaMigrationPlan(plan), sourceCols, targetCols, nil
|
||
case "smart", "auto_create_if_missing":
|
||
plan.AutoCreate = true
|
||
plan.PlannedAction = "目标表不存在,将自动建表后导入"
|
||
createSQL, postSQL, warnings, unsupported, idxCreate, idxSkip, err := buildMySQLToPGLikeCreateTablePlan(targetType, config, plan.TargetQueryTable, sourceCols, sourceDB, plan.SourceSchema, plan.SourceTable)
|
||
if err != nil {
|
||
return plan, sourceCols, targetCols, err
|
||
}
|
||
plan.CreateTableSQL = createSQL
|
||
plan.PostDataSQL = append(plan.PostDataSQL, postSQL...)
|
||
plan.Warnings = append(plan.Warnings, warnings...)
|
||
plan.UnsupportedObjects = append(plan.UnsupportedObjects, unsupported...)
|
||
plan.IndexesToCreate = idxCreate
|
||
plan.IndexesSkipped = idxSkip
|
||
return dedupeSchemaMigrationPlan(plan), sourceCols, targetCols, nil
|
||
default:
|
||
return dedupeSchemaMigrationPlan(plan), sourceCols, targetCols, nil
|
||
}
|
||
}
|
||
|
||
func buildMySQLToPGLikeAddColumnSQL(targetType string, targetQueryTable string, sourceCols, targetCols []connection.ColumnDefinition) ([]string, []string) {
|
||
targetSet := make(map[string]struct{}, len(targetCols))
|
||
for _, col := range targetCols {
|
||
key := strings.ToLower(strings.TrimSpace(col.Name))
|
||
if key == "" {
|
||
continue
|
||
}
|
||
targetSet[key] = struct{}{}
|
||
}
|
||
var sqlList []string
|
||
var warnings []string
|
||
for _, col := range sourceCols {
|
||
key := strings.ToLower(strings.TrimSpace(col.Name))
|
||
if key == "" {
|
||
continue
|
||
}
|
||
if _, ok := targetSet[key]; ok {
|
||
continue
|
||
}
|
||
colType, _, mapWarnings := mapMySQLColumnToKingbase(col)
|
||
warnings = append(warnings, mapWarnings...)
|
||
if col.Extra != "" && strings.Contains(strings.ToLower(col.Extra), "auto_increment") {
|
||
warnings = append(warnings, fmt.Sprintf("字段 %s 为自增列,补齐到已有目标表时不会自动补建 identity/sequence", col.Name))
|
||
}
|
||
sqlList = append(sqlList, fmt.Sprintf("ALTER TABLE %s ADD COLUMN %s %s NULL",
|
||
quoteQualifiedIdentByType(targetType, targetQueryTable),
|
||
quoteIdentByType(targetType, col.Name),
|
||
colType,
|
||
))
|
||
}
|
||
return sqlList, dedupeStrings(warnings)
|
||
}
|
||
|
||
func buildMySQLToPGLikeCreateTablePlan(targetType string, config SyncConfig, targetQueryTable string, sourceCols []connection.ColumnDefinition, sourceDB db.Database, sourceSchema, sourceTable string) (string, []string, []string, []string, int, int, error) {
|
||
columnDefs := make([]string, 0, len(sourceCols)+1)
|
||
warnings := make([]string, 0)
|
||
unsupported := make([]string, 0)
|
||
pkCols := make([]string, 0, 2)
|
||
for _, col := range sourceCols {
|
||
def, colWarnings := buildMySQLToPGLikeColumnDefinition(col)
|
||
warnings = append(warnings, colWarnings...)
|
||
columnDefs = append(columnDefs, fmt.Sprintf("%s %s", quoteIdentByType(targetType, col.Name), def))
|
||
if col.Key == "PRI" || col.Key == "PK" {
|
||
pkCols = append(pkCols, quoteIdentByType(targetType, col.Name))
|
||
}
|
||
}
|
||
if len(pkCols) > 0 {
|
||
columnDefs = append(columnDefs, fmt.Sprintf("PRIMARY KEY (%s)", strings.Join(pkCols, ", ")))
|
||
}
|
||
createSQL := fmt.Sprintf("CREATE TABLE %s (\n %s\n)", quoteQualifiedIdentByType(targetType, targetQueryTable), strings.Join(columnDefs, ",\n "))
|
||
if !config.CreateIndexes {
|
||
return createSQL, nil, dedupeStrings(warnings), dedupeStrings(unsupported), 0, 0, nil
|
||
}
|
||
indexes, err := sourceDB.GetIndexes(sourceSchema, sourceTable)
|
||
if err != nil {
|
||
warnings = append(warnings, fmt.Sprintf("读取源表索引失败,已跳过索引迁移:%v", err))
|
||
return createSQL, nil, dedupeStrings(warnings), dedupeStrings(unsupported), 0, 0, nil
|
||
}
|
||
grouped := groupIndexDefinitions(indexes)
|
||
postSQL := make([]string, 0, len(grouped))
|
||
created := 0
|
||
skipped := 0
|
||
for _, idx := range grouped {
|
||
name := strings.TrimSpace(idx.Name)
|
||
if name == "" || strings.EqualFold(name, "primary") {
|
||
continue
|
||
}
|
||
if len(idx.Columns) == 0 {
|
||
skipped++
|
||
unsupported = append(unsupported, fmt.Sprintf("索引 %s 缺少列定义,已跳过", name))
|
||
continue
|
||
}
|
||
kind := strings.ToLower(strings.TrimSpace(idx.IndexType))
|
||
if idx.SubPart > 0 {
|
||
skipped++
|
||
unsupported = append(unsupported, fmt.Sprintf("索引 %s 使用前缀长度,当前暂不支持迁移", name))
|
||
continue
|
||
}
|
||
if kind != "" && kind != "btree" {
|
||
skipped++
|
||
unsupported = append(unsupported, fmt.Sprintf("索引 %s 类型=%s,当前暂不支持自动迁移", name, idx.IndexType))
|
||
continue
|
||
}
|
||
quotedCols := make([]string, 0, len(idx.Columns))
|
||
for _, col := range idx.Columns {
|
||
quotedCols = append(quotedCols, quoteIdentByType(targetType, col))
|
||
}
|
||
prefix := "CREATE INDEX"
|
||
if idx.Unique {
|
||
prefix = "CREATE UNIQUE INDEX"
|
||
}
|
||
postSQL = append(postSQL, fmt.Sprintf("%s %s ON %s (%s)", prefix, quoteIdentByType(targetType, name), quoteQualifiedIdentByType(targetType, targetQueryTable), strings.Join(quotedCols, ", ")))
|
||
created++
|
||
}
|
||
return createSQL, postSQL, dedupeStrings(warnings), dedupeStrings(unsupported), created, skipped, nil
|
||
}
|
||
|
||
func buildMySQLToPGLikeColumnDefinition(col connection.ColumnDefinition) (string, []string) {
|
||
targetType, useIdentity, warnings := mapMySQLColumnToKingbase(col)
|
||
parts := []string{targetType}
|
||
if useIdentity {
|
||
parts = append(parts, "GENERATED BY DEFAULT AS IDENTITY")
|
||
}
|
||
if !useIdentity {
|
||
if defaultSQL, ok, warningText := mapMySQLDefaultToKingbase(col, targetType); warningText != "" {
|
||
warnings = append(warnings, warningText)
|
||
} else if ok {
|
||
parts = append(parts, "DEFAULT "+defaultSQL)
|
||
}
|
||
}
|
||
if strings.EqualFold(strings.TrimSpace(col.Nullable), "NO") {
|
||
parts = append(parts, "NOT NULL")
|
||
}
|
||
return strings.Join(parts, " "), dedupeStrings(warnings)
|
||
}
|