diff --git a/db-history.go b/db-history.go new file mode 100644 index 0000000..076df75 --- /dev/null +++ b/db-history.go @@ -0,0 +1,111 @@ +package main + +// ============================================================ +// db-history.go - Schema 版本声明与字段映射 +// 这是整个迁移引擎的“数据源”,记录每个版本的完整 Schema 定义。 +// 迁移工具通过对比当前 Schema 与目标 Schema 的差异来决定迁移路径。 +// ============================================================ + +// SchemaVersionDef 记录一个 Schema 版本的完整字段定义 +type SchemaVersionDef struct { + Version string // 如 "v1", "v2", "v3" + TableName string // 表名,如 "proxies" + Columns map[string]ColumnDef // 字段名 → 字段定义 +} + +// ColumnDef 描述一个字段的结构 +type ColumnDef struct { + Type string // 如 "INTEGER", "TEXT", "BOOLEAN" + NotNull bool // 是否 NOT NULL + Default string // 默认值表达式(如 "0"、"'CHANGE_ME'") + Primary bool // 是否主键 +} + +// schemaHistory 存储所有已知的 Schema 版本(从旧到新排列) +// 每个版本记录的是“完整的表结构”,而不是增量变更。 +// 新增版本时,在这里追加一条记录即可。 +var schemaHistory = []SchemaVersionDef{ + // v1:初始版本(frpc-console 1.0) + { + Version: "v1", + TableName: "proxies", + Columns: map[string]ColumnDef{ + "id": {Type: "INTEGER", Primary: true}, + "name": {Type: "TEXT", NotNull: true}, + "type": {Type: "TEXT", NotNull: true, Default: "'tcp'"}, + "local_ip": {Type: "TEXT", NotNull: true}, + "local_port": {Type: "INTEGER", NotNull: true}, + "remote_port": {Type: "INTEGER", NotNull: true}, + "enabled": {Type: "INTEGER", NotNull: true, Default: "1"}, + "created_at": {Type: "DATETIME", Default: "CURRENT_TIMESTAMP"}, + "updated_at": {Type: "DATETIME", Default: "CURRENT_TIMESTAMP"}, + }, + }, + // v2:当前版本(frpc-console 2.0 LTS) + // 注意:wire_protocol_v2 是全局配置(global_config),不在 proxies 表中 + { + Version: "v2", + TableName: "proxies", + Columns: map[string]ColumnDef{ + "id": {Type: "INTEGER", Primary: true}, + "name": {Type: "TEXT", NotNull: true}, + "type": {Type: "TEXT", NotNull: true, Default: "'tcp'"}, + "local_ip": {Type: "TEXT", NotNull: true}, + "local_port": {Type: "INTEGER", NotNull: true}, + "remote_port": {Type: "INTEGER", NotNull: true}, + "enabled": {Type: "INTEGER", NotNull: true, Default: "1"}, + "created_at": {Type: "DATETIME", Default: "CURRENT_TIMESTAMP"}, + "updated_at": {Type: "DATETIME", Default: "CURRENT_TIMESTAMP"}, + }, + }, + // v3:未来版本(规划中) + // 示例:新增隧道分组、流量统计等字段 + // { + // Version: "v3", + // TableName: "proxies", + // Columns: map[string]ColumnDef{ + // // ... 完整字段定义 + // }, + // }, +} + +// getSchemaDef 根据版本号获取 Schema 定义 +func getSchemaDef(version string) *SchemaVersionDef { + for _, def := range schemaHistory { + if def.Version == version { + return &def + } + } + return nil +} + +// getLatestSchemaDef 获取最新的 Schema 版本定义 +func getLatestSchemaDef() *SchemaVersionDef { + if len(schemaHistory) == 0 { + return nil + } + return &schemaHistory[len(schemaHistory)-1] +} + +// schemaVersionsEqual 判断两个 Schema 版本是否完全相同 +func schemaVersionsEqual(v1, v2 *SchemaVersionDef) bool { + if v1 == nil || v2 == nil { + return false + } + if v1.TableName != v2.TableName { + return false + } + if len(v1.Columns) != len(v2.Columns) { + return false + } + for name, col1 := range v1.Columns { + col2, ok := v2.Columns[name] + if !ok { + return false + } + if col1.Type != col2.Type || col1.NotNull != col2.NotNull || col1.Default != col2.Default { + return false + } + } + return true +} diff --git a/db.go b/db.go index 9e0f969..f14a0d0 100644 --- a/db.go +++ b/db.go @@ -8,6 +8,7 @@ import ( "io" "log" "os" + "sort" "strings" "time" @@ -21,7 +22,7 @@ var DB *sql.DB // ============================================================ const ( - SchemaVersion = "2.0.0" // 当前数据库 Schema 版本,与项目版本同步 + SchemaVersion = "v2" // 当前数据库 Schema 版本(与 schemaHistory 中的版本对应) ) // ============================================================ @@ -41,10 +42,10 @@ type GlobalConfig struct { HeartbeatInterval int `json:"heartbeatInterval"` HeartbeatTimeout int `json:"heartbeatTimeout"` PoolCount int `json:"poolCount"` - WireProtocolV2 bool `json:"wireProtocolV2"` // v2.0 正式启用 + WireProtocolV2 bool `json:"wireProtocolV2"` // v2 协议全局开关 } -// Proxy 隧道表 +// Proxy 隧道表(不含 wire_protocol_v2) type Proxy struct { ID int `json:"id"` Name string `json:"name"` @@ -74,22 +75,19 @@ func InitDB() error { return err } - // ---- 创建所有表 ---- if err := createTables(); err != nil { return err } - // ---- 执行版本迁移 ---- if err := runMigrations(); err != nil { return err } - // ---- 确保 JWT 密钥存在 ---- if err := ensureJwtSecret(); err != nil { return err } - log.Println("✅ 数据库初始化完成 (Schema v" + SchemaVersion + ")") + log.Println("✅ 数据库初始化完成 (Schema " + SchemaVersion + ")") return nil } @@ -152,7 +150,7 @@ func createTables() error { return err } - // 应用配置表(存储 JWT 密钥、Schema 版本等) + // 应用配置表 _, err = DB.Exec(` CREATE TABLE IF NOT EXISTS app_config ( key TEXT PRIMARY KEY, @@ -164,7 +162,7 @@ func createTables() error { return err } - // 初始化默认配置(仅当表为空时) + // 初始化默认配置 var count int DB.QueryRow("SELECT COUNT(*) FROM global_config").Scan(&count) if count == 0 { @@ -188,25 +186,21 @@ func createTables() error { // 迁移引擎 // ============================================================ -// getSchemaVersion 读取当前数据库的 Schema 版本 -func getSchemaVersion() string { +// getCurrentSchemaVersion 读取当前数据库的 Schema 版本 +func getCurrentSchemaVersion() string { var version string err := DB.QueryRow("SELECT value FROM app_config WHERE key = 'schema_version'").Scan(&version) if err != nil { if err == sql.ErrNoRows { - // 没有版本记录 → 首次启动或 v1.x 升级 - // 检查是否已有数据(通过 users 表判断) var count int DB.QueryRow("SELECT COUNT(*) FROM users").Scan(&count) if count > 0 { - // 有用户数据 → 这是 v1.x 升级,标记为 1.5 - return "1.5.0" + return "v1" } - // 全新安装 → 直接标记为当前版本 return SchemaVersion } log.Printf("⚠️ 读取 Schema 版本失败: %v", err) - return "1.5.0" // 保守降级 + return "v1" } return version } @@ -224,11 +218,11 @@ func setSchemaVersion(version string) error { func backupDatabase() (string, error) { src := "./frpc-console.db" if _, err := os.Stat(src); os.IsNotExist(err) { - return "", nil // 数据库不存在,无需备份 + return "", nil } timestamp := time.Now().Format("20060102_150405") - dst := fmt.Sprintf("./frpc-console.db.pre-v%s.%s", SchemaVersion, timestamp) + dst := fmt.Sprintf("./frpc-console.db.pre-%s.%s", SchemaVersion, timestamp) srcFile, err := os.Open(src) if err != nil { @@ -250,68 +244,6 @@ func backupDatabase() (string, error) { return dst, nil } -// runMigrations 执行版本迁移 -func runMigrations() error { - currentVersion := getSchemaVersion() - log.Printf("📌 当前数据库 Schema: %s, 目标版本: %s", currentVersion, SchemaVersion) - - if currentVersion == SchemaVersion { - log.Println("✅ Schema 已是最新,无需迁移") - return nil - } - - log.Printf("🔄 检测到版本变更 (%s → %s),开始迁移...", currentVersion, SchemaVersion) - - // ---- 1. 备份数据库 ---- - backupPath, err := backupDatabase() - if err != nil { - return fmt.Errorf("备份数据库失败: %w", err) - } - if backupPath != "" { - log.Printf("📦 备份文件: %s", backupPath) - } else { - log.Println("📦 数据库为空,跳过备份") - } - - // ---- 2. 执行迁移 ---- - // 按照版本号逐个升级 - migrations := []struct { - from string - upgrade func() error - }{ - {"1.5.0", migrateFrom1_5_0}, - {"2.0.0", migrateFrom2_0_0}, // 预留,实际无操作 - } - - applied := 0 - for _, m := range migrations { - if currentVersion == m.from { - log.Printf(" 执行迁移: %s → %s", m.from, SchemaVersion) - if err := m.upgrade(); err != nil { - // 迁移失败,尝试恢复备份 - if backupPath != "" { - log.Printf("❌ 迁移失败,尝试恢复备份: %s", backupPath) - if restoreErr := restoreDatabase(backupPath); restoreErr != nil { - log.Printf("⚠️ 恢复备份失败: %v", restoreErr) - } - } - return fmt.Errorf("迁移失败: %w", err) - } - applied++ - currentVersion = SchemaVersion - break - } - } - - // ---- 3. 更新 Schema 版本 ---- - if err := setSchemaVersion(SchemaVersion); err != nil { - return fmt.Errorf("更新 Schema 版本失败: %w", err) - } - - log.Printf("✅ 迁移完成,应用了 %d 个迁移", applied) - return nil -} - // restoreDatabase 从备份恢复数据库 func restoreDatabase(backupPath string) error { srcFile, err := os.Open(backupPath) @@ -334,76 +266,176 @@ func restoreDatabase(backupPath string) error { return nil } -// ============================================================ -// 迁移函数(各版本) -// ============================================================ +// runMigrations 执行迁移 +func runMigrations() error { + currentVer := getCurrentSchemaVersion() + targetVer := SchemaVersion -// migrateFrom1_5_0: v1.5 → v2.0 -// v1.5 已经有 wire_protocol_v2 字段(灰标占位),v2.0 无需新增字段 -// 但需要确保字段存在(兼容从 v1.0 直接升级的场景) -func migrateFrom1_5_0() error { - log.Println(" 迁移: v1.5.0 → v2.0.0") + log.Printf("📌 当前数据库 Schema: %s, 目标版本: %s", currentVer, targetVer) - // 检查并补全 wire_protocol_v2 字段(兼容从 v1.0 直接升级的场景) - cols, err := getCurrentColumns("global_config") - if err != nil { - return fmt.Errorf("获取列信息失败: %w", err) + if currentVer == targetVer { + log.Println("✅ Schema 已是最新,无需迁移") + return nil } - if !contains(cols, "wire_protocol_v2") { - log.Println(" 添加字段: wire_protocol_v2") - _, err := DB.Exec("ALTER TABLE global_config ADD COLUMN wire_protocol_v2 INTEGER NOT NULL DEFAULT 0") - if err != nil { - return fmt.Errorf("添加 wire_protocol_v2 字段失败: %w", err) + log.Printf("🔄 检测到版本变更 (%s → %s),开始迁移...", currentVer, targetVer) + + backupPath, err := backupDatabase() + if err != nil { + return fmt.Errorf("备份数据库失败: %w", err) + } + if backupPath != "" { + log.Printf("📦 备份文件: %s", backupPath) + } + + currentSchema := getSchemaDef(currentVer) + targetSchema := getSchemaDef(targetVer) + + if targetSchema == nil { + return fmt.Errorf("目标 Schema 版本 %s 未在 schemaHistory 中定义", targetVer) + } + + if currentSchema == nil || schemaVersionsEqual(currentSchema, targetSchema) { + log.Println(" 迁移类型: 轻量复制(Schema 无变更)") + } else { + log.Println(" 迁移类型: 重型迁移(Schema 有变更,新建表 + 搬数据)") + if err := heavyMigration(currentSchema, targetSchema); err != nil { + if backupPath != "" { + log.Printf("❌ 迁移失败,尝试恢复备份: %s", backupPath) + if restoreErr := restoreDatabase(backupPath); restoreErr != nil { + log.Printf("⚠️ 恢复备份失败: %v", restoreErr) + } + } + return fmt.Errorf("重型迁移失败: %w", err) } } - log.Println(" ✅ v1.5.0 → v2.0.0 迁移完成") - return nil -} - -// migrateFrom2_0_0: 预留,v2.0 → 未来版本 -func migrateFrom2_0_0() error { - log.Println(" v2.0.0 已是当前版本,无需迁移") - return nil -} - -// ============================================================ -// 辅助函数 -// ============================================================ - -func getCurrentColumns(tableName string) ([]string, error) { - rows, err := DB.Query("PRAGMA table_info(" + tableName + ")") - if err != nil { - return nil, err + if err := setSchemaVersion(targetVer); err != nil { + return fmt.Errorf("更新 Schema 版本失败: %w", err) } - defer rows.Close() + log.Printf("✅ 迁移完成,当前 Schema: %s", targetVer) + return nil +} + +// heavyMigration 重型迁移 +func heavyMigration(oldDef, newDef *SchemaVersionDef) error { + if oldDef == nil { + return fmt.Errorf("旧 Schema 定义为空,无法执行重型迁移") + } + + oldTable := oldDef.TableName + newTable := oldTable + "_new" + + createSQL := buildCreateTableSQL(newTable, newDef) + log.Printf(" 创建新表: %s", newTable) + if _, err := DB.Exec(createSQL); err != nil { + return fmt.Errorf("创建新表失败: %w", err) + } + + insertSQL, err := buildInsertSQL(oldTable, newTable, oldDef, newDef) + if err != nil { + return fmt.Errorf("构建数据迁移 SQL 失败: %w", err) + } + log.Printf(" 迁移数据: %s → %s", oldTable, newTable) + if _, err := DB.Exec(insertSQL); err != nil { + return fmt.Errorf("数据迁移失败: %w", err) + } + + var oldCount, newCount int + DB.QueryRow(fmt.Sprintf("SELECT COUNT(*) FROM %s", oldTable)).Scan(&oldCount) + DB.QueryRow(fmt.Sprintf("SELECT COUNT(*) FROM %s", newTable)).Scan(&newCount) + if oldCount != newCount { + return fmt.Errorf("数据迁移不完整: 旧表 %d 行,新表 %d 行", oldCount, newCount) + } + log.Printf(" 数据迁移验证通过: %d 行", newCount) + + tempTable := oldTable + "_old_temp" + if _, err := DB.Exec(fmt.Sprintf("ALTER TABLE %s RENAME TO %s", oldTable, tempTable)); err != nil { + return fmt.Errorf("重命名旧表失败: %w", err) + } + if _, err := DB.Exec(fmt.Sprintf("ALTER TABLE %s RENAME TO %s", newTable, oldTable)); err != nil { + DB.Exec(fmt.Sprintf("ALTER TABLE %s RENAME TO %s", tempTable, oldTable)) + return fmt.Errorf("重命名新表失败: %w", err) + } + if _, err := DB.Exec(fmt.Sprintf("DROP TABLE %s", tempTable)); err != nil { + log.Printf("⚠️ 删除临时表失败(不影响使用): %v", err) + } + + log.Printf(" 表交换完成: %s (新表已生效)", oldTable) + return nil +} + +// buildCreateTableSQL 根据 Schema 定义生成 CREATE TABLE 语句 +func buildCreateTableSQL(tableName string, def *SchemaVersionDef) string { var cols []string - for rows.Next() { - var ( - cid int - name string - typ string - notNull int - dfltVal sql.NullString - pk int - ) - if err := rows.Scan(&cid, &name, &typ, ¬Null, &dfltVal, &pk); err != nil { - return nil, err - } - cols = append(cols, name) + var primaryKey string + + names := make([]string, 0, len(def.Columns)) + for name := range def.Columns { + names = append(names, name) } - return cols, rows.Err() + sort.Strings(names) + + for _, name := range names { + col := def.Columns[name] + parts := []string{name, col.Type} + if col.NotNull { + parts = append(parts, "NOT NULL") + } + if col.Default != "" { + parts = append(parts, "DEFAULT "+col.Default) + } + if col.Primary { + primaryKey = "PRIMARY KEY (" + name + ")" + } else { + cols = append(cols, strings.Join(parts, " ")) + } + } + + if primaryKey != "" { + cols = append(cols, primaryKey) + } + + return fmt.Sprintf("CREATE TABLE %s (\n %s\n)", tableName, strings.Join(cols, ",\n ")) } -func contains(slice []string, item string) bool { - for _, s := range slice { - if strings.EqualFold(s, item) { - return true +// buildInsertSQL 构建 INSERT INTO new_table SELECT ... FROM old_table +func buildInsertSQL(oldTable, newTable string, oldDef, newDef *SchemaVersionDef) (string, error) { + newCols := make([]string, 0, len(newDef.Columns)) + for name := range newDef.Columns { + newCols = append(newCols, name) + } + sort.Strings(newCols) + + var selectParts []string + var colNames []string + + for _, name := range newCols { + colNames = append(colNames, name) + if _, ok := oldDef.Columns[name]; ok { + selectParts = append(selectParts, name) + } else { + colDef := newDef.Columns[name] + if colDef.Default != "" { + selectParts = append(selectParts, colDef.Default+" AS "+name) + } else if colDef.Type == "INTEGER" { + selectParts = append(selectParts, "0 AS "+name) + } else if colDef.Type == "TEXT" { + selectParts = append(selectParts, "'' AS "+name) + } else { + selectParts = append(selectParts, "NULL AS "+name) + } } } - return false + + return fmt.Sprintf( + "INSERT INTO %s (%s) SELECT %s FROM %s", + newTable, + strings.Join(colNames, ", "), + strings.Join(selectParts, ", "), + oldTable, + ), nil } // ============================================================ @@ -504,10 +536,7 @@ func GetProxies() ([]Proxy, error) { } proxies = append(proxies, p) } - if err = rows.Err(); err != nil { - return nil, err - } - return proxies, nil + return proxies, rows.Err() } func GetProxy(id int) (*Proxy, error) { diff --git a/static/app.js b/static/app.js index a0e0c24..7d3082d 100644 --- a/static/app.js +++ b/static/app.js @@ -28,6 +28,14 @@ async function apiFetch(endpoint, options = {}) { ...options, headers: { ...headers, ...(options.headers || {}) }, }); + + // ---- 统一处理 401 认证失效 ---- + if (res.status === 401) { + clearAuthState(); + window.dispatchEvent(new CustomEvent("auth:expired")); + return { code: 401, msg: "认证已过期,请重新登录" }; + } + return res.json(); } @@ -286,8 +294,16 @@ const app = createApp({ let logTimer = null; let logFetching = false; + // ---- 认证过期处理 ---- + const handleAuthExpired = () => { + doLogout(); + }; + // ---- 初始化 ---- onMounted(() => { + // 监听认证过期事件 + window.addEventListener("auth:expired", handleAuthExpired); + const saved = loadAuthState(); if (saved.valid) { token.value = saved.token; @@ -351,6 +367,11 @@ const app = createApp({ logFetching = true; try { const result = await fetchLogsApi(); + // ---- 如果 API 返回 401,停止轮询(事件已触发登出) ---- + if (result.code === 401) { + logFetching = false; + return; + } logLines.value = result.lines; logTotal.value = result.total; logError.value = result.error; @@ -641,7 +662,6 @@ const app = createApp({ ); const showDefaultTip = computed(() => { - // 如果 serverAddr 还是默认值,或者为空,显示警告 const addr = globalConfig.serverAddr; return !addr || addr === "frp.example.com" || addr.trim() === ""; }); @@ -692,9 +712,8 @@ const app = createApp({ passwordChangeSuccess, changePassword, - showDefaultTip, + showDefaultTip, - // 日志相关 logLines, logTotal, logError, @@ -703,7 +722,6 @@ const app = createApp({ refreshLogs, scrollLogToBottom, - // Token 显示切换 showToken, }; },