lxh2875931338/frps-console
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 2 Wiki Activity

frps端同步frpc2.4-LTS的全套代码库的设计逻辑

Browse Source
This commit is contained in:
不安的亡命樱
2026-08-01 19:31:03 +08:00
parent ccdd5b0725
commit 3214215032
10 changed files with 988 additions and 455 deletions
Download Patch File Download Diff File Expand all files Collapse all files
db.go
+212 -75
View File
@@ -8,6 +8,8 @@ import (
"io"
"log"
"os"
"sort"
"strings"
"time"
_ "modernc.org/sqlite"
@@ -15,17 +17,9 @@ import (
var DB *sql.DB
// ============================================================
// 数据库版本常量
// ============================================================
const SchemaVersion = "v2"
const SchemaVersion = "2.0.0"
// ============================================================
// 数据模型
// ============================================================
// GlobalConfig frps 全局配置
// GlobalConfig 对应 frps 配置
type GlobalConfig struct {
ID int `json:"id"`
BindPort int `json:"bindPort"`
@@ -35,7 +29,15 @@ type GlobalConfig struct {
TcpMux bool `json:"tcpMux"`
}
// Client 客户端连接记录
// User 保持不变
type User struct {
ID int `json:"id"`
Username string `json:"username"`
PasswordHash string `json:"-"`
CreatedAt string `json:"createdAt"`
}
// Client 记录（从 Dashboard 同步，可选）
type Client struct {
ID int `json:"id"`
Name string `json:"name"`
@@ -47,21 +49,15 @@ type Client struct {
UpdatedAt string `json:"updatedAt"`
}
// User 用户表
type User struct {
ID int `json:"id"`
Username string `json:"username"`
PasswordHash string `json:"-"`
CreatedAt string `json:"createdAt"`
}
// ============================================================
// 数据库初始化
// ============================================================
func InitDB() error {
// 确保 data 目录存在
if err := os.MkdirAll("./data", 0755); err != nil {
return fmt.Errorf("创建数据目录失败: %w", err)
}
dbPath := "./data/frps-console.db"
var err error
DB, err = sql.Open("sqlite", "./frps-console.db")
DB, err = sql.Open("sqlite", dbPath)
if err != nil {
return err
}
@@ -78,16 +74,12 @@ func InitDB() error {
return err
}
log.Println("✅ 数据库初始化完成 (Schema v" + SchemaVersion + ")")
log.Println("✅ 数据库初始化完成 (Schema " + SchemaVersion + ")")
return nil
}
// ============================================================
// 建表
// ============================================================
func createTables() error {
// 用户表
// users 表
_, err := DB.Exec(`
CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -101,7 +93,7 @@ func createTables() error {
return err
}
// 全局配置表（frps 专用）
// global_config 表（frps 专用）
_, err = DB.Exec(`
CREATE TABLE IF NOT EXISTS global_config (
id INTEGER PRIMARY KEY CHECK (id = 1),
@@ -117,7 +109,7 @@ func createTables() error {
return err
}
// 客户端记录表
// clients 表（用于存储从 Dashboard 拉取的客户端，可选）
_, err = DB.Exec(`
CREATE TABLE IF NOT EXISTS clients (
id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -134,7 +126,7 @@ func createTables() error {
return err
}
// 应用配置表
// app_config 表
_, err = DB.Exec(`
CREATE TABLE IF NOT EXISTS app_config (
key TEXT PRIMARY KEY,
@@ -151,9 +143,8 @@ func createTables() error {
DB.QueryRow("SELECT COUNT(*) FROM global_config").Scan(&count)
if count == 0 {
_, err = DB.Exec(`
INSERT INTO global_config (
id, bind_port, token, log_level, log_max_days, tcp_mux
) VALUES (1, 9358, 'CHANGE_ME', 'info', 3, 1)
INSERT INTO global_config (id, bind_port, token, log_level, log_max_days, tcp_mux)
VALUES (1, 9358, 'CHANGE_ME', 'info', 3, 1)
`)
if err != nil {
return err
@@ -164,23 +155,23 @@ func createTables() error {
return nil
}
// ============================================================
// 迁移引擎
// ============================================================
// ========== 迁移引擎（复用 frpc 的 db-history.go） ==========
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 {
// 尝试判断是否为旧版本
var count int
DB.QueryRow("SELECT COUNT(*) FROM users").Scan(&count)
if count > 0 {
return "1.0.0"
return "v1"
}
return SchemaVersion
}
return "1.0.0"
log.Printf("⚠️ 读取 Schema 版本失败: %v", err)
return "v1"
}
return version
}
@@ -194,13 +185,13 @@ func setSchemaVersion(version string) error {
}
func backupDatabase() (string, error) {
src := "./frps-console.db"
src := "./data/frps-console.db"
if _, err := os.Stat(src); os.IsNotExist(err) {
return "", nil
}
timestamp := time.Now().Format("20060102_150405")
dst := fmt.Sprintf("./frps-console.db.pre-v%s.%s", SchemaVersion, timestamp)
dst := fmt.Sprintf("./data/frps-console.db.pre-%s.%s", SchemaVersion, timestamp)
srcFile, err := os.Open(src)
if err != nil {
@@ -229,7 +220,7 @@ func restoreDatabase(backupPath string) error {
}
defer srcFile.Close()
dstFile, err := os.Create("./frps-console.db")
dstFile, err := os.Create("./data/frps-console.db")
if err != nil {
return err
}
@@ -244,35 +235,187 @@ func restoreDatabase(backupPath string) error {
}
func runMigrations() error {
currentVersion := getSchemaVersion()
log.Printf("📌 当前数据库 Schema: %s, 目标版本: %s", currentVersion, SchemaVersion)
currentVer := getCurrentSchemaVersion()
targetVer := SchemaVersion
if currentVersion != SchemaVersion {
log.Printf("🔄 检测到版本变更 (%s → %s)，开始迁移...", currentVersion, SchemaVersion)
log.Printf("📌 当前数据库 Schema: %s, 目标版本: %s", currentVer, targetVer)
backupPath, err := backupDatabase()
if err != nil {
return fmt.Errorf("备份数据库失败: %w", err)
if currentVer == targetVer {
var userCount int
err := DB.QueryRow("SELECT COUNT(*) FROM users").Scan(&userCount)
if err != nil || userCount == 0 {
log.Println(" 数据库为空或无效，无需迁移，直接初始化")
return nil
}
if backupPath != "" {
log.Printf("📦 备份文件: %s", backupPath)
}
if err := setSchemaVersion(SchemaVersion); err != nil {
return fmt.Errorf("更新 Schema 版本失败: %w", err)
}
log.Printf("✅ 版本迁移完成")
} else {
log.Println("✅ Schema 已是最新，无需迁移")
log.Println("✅ Schema 已是最新，数据库有效")
return nil
}
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 无变更）")
var userCount int
err := DB.QueryRow("SELECT COUNT(*) FROM users").Scan(&userCount)
if err != nil || userCount == 0 {
log.Println(" 数据库为空或无效，跳过迁移，直接初始化")
return nil
}
log.Println(" 数据库有效，继续使用")
} 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)
}
}
if err := setSchemaVersion(targetVer); err != nil {
return fmt.Errorf("更新 Schema 版本失败: %w", err)
}
log.Printf("✅ 迁移完成，当前 Schema: %s", targetVer)
return nil
}
// ============================================================
// JWT 密钥管理
// ============================================================
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
}
func buildCreateTableSQL(tableName string, def *SchemaVersionDef) string {
var cols []string
var primaryKey string
names := make([]string, 0, len(def.Columns))
for name := range def.Columns {
names = append(names, 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")
}
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 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 fmt.Sprintf(
"INSERT INTO %s (%s) SELECT %s FROM %s",
newTable,
strings.Join(colNames, ", "),
strings.Join(selectParts, ", "),
oldTable,
), nil
}
// ========== JWT 密钥 ==========
func ensureJwtSecret() error {
var value string
@@ -306,9 +449,7 @@ func GetJwtSecret() (string, error) {
return secret, nil
}
// ============================================================
// 全局配置 CRUD
// ============================================================
// ========== GlobalConfig CRUD ==========
func GetGlobalConfig() (*GlobalConfig, error) {
var cfg GlobalConfig
@@ -338,9 +479,7 @@ func UpdateGlobalConfig(cfg *GlobalConfig) error {
return err
}
// ============================================================
// 客户端记录 CRUD
// ============================================================
// ========== Clients CRUD（用于同步 Dashboard） ==========
func SaveClient(client *Client) error {
_, err := DB.Exec(`
@@ -372,9 +511,7 @@ func GetClients() ([]Client, error) {
return clients, rows.Err()
}
// ============================================================
// 用户 CRUD
// ============================================================
// ========== User CRUD ==========
func GetUserByUsername(username string) (*User, error) {
var u User