部分代码上新

main.go

@@ -1,12 +1,18 @@
 package main
 
 import (
+	"context"
 	"log"
 	"os"
 	"strings"
 	"time"
 )
 
+var (
+	// 全局进程管理器
+	processManager *ProcessManager
+)
+
 func main() {
 	// 确保 data 目录存在
 	if err := os.MkdirAll("./data", 0755); err != nil {
@@ -29,10 +35,39 @@ func main() {
 		log.Println("⚠️  生成配置文件失败:", err)
 	}
 
-	if err := StartFrpc(); err != nil {
-		log.Println("⚠️  启动 frpc 失败:", err)
+	// ============================================================
+	// 2.6-preview: 使用 ProcessManager 管理 frpc 进程
+	// ============================================================
+	processManager = NewProcessManager(
+		"./data",          // 数据目录
+		"./data/frpc.toml", // 配置文件路径
+		"./data/frpc",      // frpc 二进制路径
+	)
+
+	// 加载 admin_port 配置
+	if err := processManager.LoadConfig(); err != nil {
+		log.Printf("⚠️  加载 admin_port 配置失败: %v (将使用 PID 文件模式)", err)
+	} else {
+		log.Printf("📌 admin_port: %d", processManager.adminPort)
 	}
 
+	// 启动 frpc (幂等)
+	ctx := context.Background()
+	if err := processManager.Start(ctx); err != nil {
+		log.Printf("⚠️  启动 frpc 失败: %v", err)
+	} else {
+		// 查询并显示启动状态
+		if status, err := processManager.Status(); err == nil {
+			log.Printf("✅ frpc 状态: %s", status.State)
+			if status.PID > 0 {
+				log.Printf("   PID: %d, 端口: %d", status.PID, status.Port)
+			}
+		}
+	}
+
+	// ============================================================
+	// 启动看门狗 (使用 ProcessManager 检测状态)
+	// ============================================================
 	go startWatchdog()
 
 	r := SetupRouter()
@@ -45,18 +80,30 @@ func main() {
 	}
 }
 
+// ================================================================
+// 看门狗 (使用 ProcessManager)
+// ================================================================
+
 func startWatchdog() {
 	ticker := time.NewTicker(30 * time.Second)
 	defer ticker.Stop()
 
 	for range ticker.C {
-		if !isFrpcRunning() {
-			log.Println("⚠️  frpc 进程已停止，自动重启...")
-			if err := StartFrpc(); err != nil {
+		// 使用 ProcessManager 检查状态
+		status, err := processManager.Status()
+		if err != nil {
+			log.Printf("⚠️  检查 frpc 状态失败: %v", err)
+			continue
+		}
+
+		if status.State != "running" {
+			log.Printf("⚠️  frpc 进程已停止 (状态: %s)，自动重启...", status.State)
+			ctx := context.Background()
+			if err := processManager.Start(ctx); err != nil {
 				log.Printf("❌ 自动重启 frpc 失败: %v", err)
 			} else {
 				log.Println("✅ frpc 已自动重启")
 			}
 		}
 	}
-}
+}