2.2的部分代码新增实现

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
+}

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 版本
+// getCurrentSchemaVersion 读取当前数据库的 Schema 版本
-func getSchemaVersion() string {
+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 "v1"
-				return "1.5.0"
 			}
-			// 全新安装 → 直接标记为当前版本
 			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
+	log.Printf("📌 当前数据库 Schema: %s, 目标版本: %s", currentVer, targetVer)
-// v1.5 已经有 wire_protocol_v2 字段（灰标占位），v2.0 无需新增字段
-// 但需要确保字段存在（兼容从 v1.0 直接升级的场景）
-func migrateFrom1_5_0() error {
-	log.Println("  迁移: v1.5.0 → v2.0.0")
 
-	// 检查并补全 wire_protocol_v2 字段（兼容从 v1.0 直接升级的场景）
+	if currentVer == targetVer {
-	cols, err := getCurrentColumns("global_config")
+		log.Println("✅ Schema 已是最新，无需迁移")
+		return nil
+	}
+
+	log.Printf("🔄 检测到版本变更 (%s → %s)，开始迁移...", currentVer, targetVer)
+
+	backupPath, err := backupDatabase()
 	if err != nil {
-		return fmt.Errorf("获取列信息失败: %w", err)
+		return fmt.Errorf("备份数据库失败: %w", err)
+	}
+	if backupPath != "" {
+		log.Printf("📦 备份文件: %s", backupPath)
 	}
 
-	if !contains(cols, "wire_protocol_v2") {
+	currentSchema := getSchemaDef(currentVer)
-		log.Println("  添加字段: wire_protocol_v2")
+	targetSchema := getSchemaDef(targetVer)
-		_, err := DB.Exec("ALTER TABLE global_config ADD COLUMN wire_protocol_v2 INTEGER NOT NULL DEFAULT 0")
+
-		if err != nil {
+	if targetSchema == nil {
-			return fmt.Errorf("添加 wire_protocol_v2 字段失败: %w", err)
+		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 迁移完成")
+	if err := setSchemaVersion(targetVer); err != nil {
+		return fmt.Errorf("更新 Schema 版本失败: %w", err)
+	}
+
+	log.Printf("✅ 迁移完成，当前 Schema: %s", targetVer)
 	return nil
 }
 
-// migrateFrom2_0_0: 预留，v2.0 → 未来版本
+// heavyMigration 重型迁移
-func migrateFrom2_0_0() error {
+func heavyMigration(oldDef, newDef *SchemaVersionDef) error {
-	log.Println("  v2.0.0 已是当前版本，无需迁移")
+	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 {
-// ============================================================
-
-func getCurrentColumns(tableName string) ([]string, error) {
-	rows, err := DB.Query("PRAGMA table_info(" + tableName + ")")
-	if err != nil {
-		return nil, err
-	}
-	defer rows.Close()
-
 	var cols []string
-	for rows.Next() {
+	var primaryKey string
-		var (
+
-			cid     int
+	names := make([]string, 0, len(def.Columns))
-			name    string
+	for name := range def.Columns {
-			typ     string
+		names = append(names, name)
-			notNull int
-			dfltVal sql.NullString
-			pk      int
-		)
-		if err := rows.Scan(&cid, &name, &typ, &notNull, &dfltVal, &pk); err != nil {
-			return nil, err
 	}
-		cols = append(cols, name)
+	sort.Strings(names)
+
+	for _, name := range names {
+		col := def.Columns[name]
+		parts := []string{name, col.Type}
+		if col.NotNull {
+			parts = append(parts, "NOT NULL")
 		}
-	return cols, rows.Err()
+		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 {
+// buildInsertSQL 构建 INSERT INTO new_table SELECT ... FROM old_table
-	for _, s := range slice {
+func buildInsertSQL(oldTable, newTable string, oldDef, newDef *SchemaVersionDef) (string, error) {
-		if strings.EqualFold(s, item) {
+	newCols := make([]string, 0, len(newDef.Columns))
-			return true
+	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 proxies, rows.Err()
-		return nil, err
-	}
-	return proxies, nil
 }
 
 func GetProxy(id int) (*Proxy, error) {

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() === "";
         });
@@ -694,7 +714,6 @@ const app = createApp({
 
             showDefaultTip,
 
-            // 日志相关
             logLines,
             logTotal,
             logError,
@@ -703,7 +722,6 @@ const app = createApp({
             refreshLogs,
             scrollLogToBottom,
 
-            // Token 显示切换
             showToken,
         };
     },