重构进行中,preview通道尚未完全恢复全部功能

This commit is contained in:
2026-08-07 19:16:08 +08:00
parent 5bce105bd6
commit 3a381c85ed
21 changed files with 889 additions and 2416 deletions
+44 -34
View File
@@ -6,20 +6,26 @@ import (
"os"
"strings"
"time"
)
var (
// 全局进程管理器
processManager *ProcessManager
"frpc-console/internal/api"
"frpc-console/internal/db"
"frpc-console/internal/frp"
"frpc-console/internal/process"
)
func main() {
// 确保 data 目录存在
log.SetFlags(log.LstdFlags | log.Lshortfile)
// ============================================================
// 1. 初始化数据目录
// ============================================================
if err := os.MkdirAll("./data", 0755); err != nil {
log.Printf("⚠️ 创建 data 目录失败: %v", err)
log.Printf("⚠️ 创建 data 目录失败: %v", err)
}
// 读取 version.ini 显示版本
// ============================================================
// 2. 显示版本信息
// ============================================================
if data, err := os.ReadFile("./data/version.ini"); err == nil {
version := strings.TrimSpace(string(data))
log.Printf("📌 版本: %s", version)
@@ -27,37 +33,39 @@ func main() {
log.Printf("📌 版本: (未记录)")
}
if err := InitDB(); err != nil {
// ============================================================
// 3. 初始化数据库
// ============================================================
if err := db.InitDB(); err != nil {
log.Fatal("❌ 数据库初始化失败:", err)
}
if err := GenerateFrpcConfig(); err != nil {
log.Println("⚠️ 生成配置文件失败:", err)
// ============================================================
// 4. 生成 frpc 配置文件
// ============================================================
if err := frp.GenerateConfig(); err != nil {
log.Println("⚠️ 生成配置文件失败:", err)
}
// ============================================================
// 2.6-preview: 使用 ProcessManager 管理 frpc 进程
// 5. 创建进程管理器并启动 frpc
// ============================================================
processManager = NewProcessManager(
"./data", // 数据目录
"./data/frpc.toml", // 配置文件路径
"./data/frpc", // frpc 二进制路径
)
pm := process.NewManager("./data", "./data/frpc.toml")
process.SetGlobalManager(pm)
// 加载 admin_port 配置
if err := processManager.LoadConfig(); err != nil {
log.Printf("⚠️ 加载 admin_port 配置失败: %v (将使用 PID 文件模式)", err)
if err := pm.LoadConfig(); err != nil {
log.Printf("⚠️ 加载 admin_port 配置失败: %v (将使用 PID 文件模式)", err)
} else {
log.Printf("📌 admin_port: %d", processManager.adminPort)
log.Printf("📌 admin_port: %d", pm.AdminPort())
}
// 启动 frpc (幂等)
ctx := context.Background()
if err := processManager.Start(ctx); err != nil {
log.Printf("⚠️ 启动 frpc 失败: %v", err)
if err := pm.Start(ctx); err != nil {
log.Printf("⚠️ 启动 frpc 失败: %v", err)
} else {
// 查询并显示启动状态
if status, err := processManager.Status(); err == nil {
if status, err := pm.Status(); err == nil {
log.Printf("✅ frpc 状态: %s", status.State)
if status.PID > 0 {
log.Printf(" PID: %d, 端口: %d", status.PID, status.Port)
@@ -66,40 +74,42 @@ func main() {
}
// ============================================================
// 启动看门狗 (使用 ProcessManager 检测状态)
// 6. 启动看门狗 (自动恢复)
// ============================================================
go startWatchdog()
go startWatchdog(pm)
r := SetupRouter()
// ============================================================
// 7. 启动 HTTP 服务
// ============================================================
router := api.SetupRouter()
log.Println("🚀 frpc-console 启动成功!")
log.Println("📍 访问地址: http://localhost:9300")
log.Println("📍 API 地址: http://localhost:9300/api")
if err := r.Run(":9300"); err != nil {
if err := router.Run(":9300"); err != nil {
log.Fatal("❌ 服务启动失败:", err)
}
}
// ================================================================
// 看门狗 (使用 ProcessManager)
// 看门狗
// ================================================================
func startWatchdog() {
func startWatchdog(pm *process.Manager) {
ticker := time.NewTicker(30 * time.Second)
defer ticker.Stop()
for range ticker.C {
// 使用 ProcessManager 检查状态
status, err := processManager.Status()
status, err := pm.Status()
if err != nil {
log.Printf("⚠️ 检查 frpc 状态失败: %v", err)
log.Printf("⚠️ 检查 frpc 状态失败: %v", err)
continue
}
if status.State != "running" {
log.Printf("⚠️ frpc 进程已停止 (状态: %s),自动重启...", status.State)
log.Printf("⚠️ frpc 进程已停止 (状态: %s),自动重启...", status.State)
ctx := context.Background()
if err := processManager.Start(ctx); err != nil {
if err := pm.Start(ctx); err != nil {
log.Printf("❌ 自动重启 frpc 失败: %v", err)
} else {
log.Println("✅ frpc 已自动重启")