mirror of
https://gitee.com/mirrors/AllinSSL.git
synced 2026-03-13 18:10:53 +08:00
cf 默认不跳过预检查
This commit is contained in:
@@ -4,32 +4,32 @@ import (
|
||||
"database/sql"
|
||||
"fmt"
|
||||
"os"
|
||||
|
||||
|
||||
_ "modernc.org/sqlite" // 使用 pure Go 实现的 SQLite 驱动
|
||||
)
|
||||
|
||||
func EnsureDatabaseWithTables(targetDBPath string, baseDBPath string, tables []string) error {
|
||||
// 1. 检查数据库是否存在
|
||||
if _, err := os.Stat(targetDBPath); err == nil {
|
||||
fmt.Printf("数据库 %s 已存在,跳过迁移。\n", targetDBPath)
|
||||
// fmt.Printf("数据库 %s 已存在,跳过迁移。\n", targetDBPath)
|
||||
return nil
|
||||
}
|
||||
|
||||
fmt.Printf("数据库 %s 不存在,开始从基础数据库迁移表...\n", targetDBPath)
|
||||
|
||||
|
||||
// fmt.Printf("数据库 %s 不存在,开始从基础数据库迁移表...\n", targetDBPath)
|
||||
|
||||
// 2. 打开源数据库(只读)和目标数据库(新建)
|
||||
baseDB, err := sql.Open("sqlite", baseDBPath)
|
||||
if err != nil {
|
||||
return fmt.Errorf("打开基础数据库失败: %v", err)
|
||||
}
|
||||
defer baseDB.Close()
|
||||
|
||||
|
||||
targetDB, err := sql.Open("sqlite", targetDBPath)
|
||||
if err != nil {
|
||||
return fmt.Errorf("创建目标数据库失败: %v", err)
|
||||
}
|
||||
defer targetDB.Close()
|
||||
|
||||
|
||||
for _, table := range tables {
|
||||
// 2.1 获取建表语句
|
||||
var createSQL string
|
||||
@@ -38,26 +38,26 @@ func EnsureDatabaseWithTables(targetDBPath string, baseDBPath string, tables []s
|
||||
if err != nil {
|
||||
return fmt.Errorf("获取表 %s 的结构失败: %v", table, err)
|
||||
}
|
||||
|
||||
|
||||
// 2.2 在目标库中创建表
|
||||
_, err = targetDB.Exec(createSQL)
|
||||
if err != nil {
|
||||
return fmt.Errorf("创建表 %s 失败: %v", table, err)
|
||||
}
|
||||
|
||||
|
||||
// 2.3 从基础库读取数据并插入目标库
|
||||
rows, err := baseDB.Query(fmt.Sprintf("SELECT * FROM %s", table))
|
||||
if err != nil {
|
||||
return fmt.Errorf("读取表 %s 数据失败: %v", table, err)
|
||||
}
|
||||
|
||||
|
||||
cols, _ := rows.Columns()
|
||||
values := make([]interface{}, len(cols))
|
||||
valuePtrs := make([]interface{}, len(cols))
|
||||
|
||||
|
||||
tx, _ := targetDB.Begin()
|
||||
stmt, _ := tx.Prepare(buildInsertSQL(table, len(cols)))
|
||||
|
||||
|
||||
for rows.Next() {
|
||||
for i := range values {
|
||||
valuePtrs[i] = &values[i]
|
||||
@@ -65,13 +65,13 @@ func EnsureDatabaseWithTables(targetDBPath string, baseDBPath string, tables []s
|
||||
rows.Scan(valuePtrs...)
|
||||
stmt.Exec(values...)
|
||||
}
|
||||
|
||||
|
||||
stmt.Close()
|
||||
tx.Commit()
|
||||
rows.Close()
|
||||
}
|
||||
|
||||
fmt.Println("迁移完成。")
|
||||
|
||||
// fmt.Println("迁移完成。")
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user