部分代码上新

2026-08-07 08:30:17 +08:00
parent d93e74bc77
commit fafe07baf8
4 changed files with 1030 additions and 20 deletions
@@ -28,6 +28,10 @@ var (
 	frpcPathMutex  sync.Mutex
 )

+// ================================================================
+// frpc 二进制提取 (保持不变)
+// ================================================================
+
 func getFrpcPath() (string, error) {
 	frpcPathMutex.Lock()
 	defer frpcPathMutex.Unlock()
@@ -89,6 +93,10 @@ func getFrpcPath() (string, error) {
 	return "", fmt.Errorf("未找到 frpc 文件")
 }

+// ================================================================
+// 配置生成 (保持不变)
+// ================================================================
+
 func GenerateFrpcConfig() error {
 	cfg, err := GetGlobalConfig()
 	if err != nil {
@@ -155,7 +163,28 @@ func GenerateFrpcConfig() error {
 	return nil
 }

+// ================================================================
+// 进程管理 (兼容层，实际调用 ProcessManager)
+// 2.6-preview: 所有进程管理逻辑迁移到 process_manager.go
+// ================================================================
+
+// isFrpcRunning 检查 frpc 是否在运行
+// 优先使用 ProcessManager，降级到 PID 文件
 func isFrpcRunning() bool {
+	// 如果 ProcessManager 已初始化，使用它
+	if processManager != nil {
+		status, err := processManager.Status()
+		if err != nil {
+			log.Printf("[WARN] ProcessManager.Status() 失败: %v，降级到 PID 文件", err)
+			return isFrpcRunningLegacy()
+		}
+		return status.State == "running"
+	}
+	return isFrpcRunningLegacy()
+}
+
+// isFrpcRunningLegacy 旧版 PID 文件检测 (降级方案)
+func isFrpcRunningLegacy() bool {
 	pidData, err := os.ReadFile("./data/frpc.pid")
 	if err != nil {
 		return false
@@ -181,7 +210,19 @@ func isFrpcRunning() bool {
 	return process.Signal(syscall.Signal(0)) == nil
 }

+// StartFrpc 启动 frpc (幂等)
+// 优先使用 ProcessManager，降级到旧逻辑
 func StartFrpc() error {
+	if processManager != nil {
+		log.Println("[INFO] 使用 ProcessManager 启动 frpc")
+		return processManager.Start(nil)
+	}
+	log.Println("[WARN] ProcessManager 未初始化，使用兼容模式启动 frpc")
+	return startFrpcLegacy()
+}
+
+// startFrpcLegacy 旧版启动逻辑 (降级方案)
+func startFrpcLegacy() error {
 	frpcPath, err := getFrpcPath()
 	if err != nil {
 		return fmt.Errorf("获取 frpc 路径失败: %w", err)
@@ -197,7 +238,7 @@ func StartFrpc() error {
 		}
 	}

-	if isFrpcRunning() {
+	if isFrpcRunningLegacy() {
 		return nil
 	}

@@ -218,12 +259,10 @@ func StartFrpc() error {
 		return fmt.Errorf("启动 frpc 失败: %w", err)
 	}

-	// 回收子进程（防止僵尸进程）
 	go func() {
 		if err := cmd.Wait(); err != nil {
 			log.Printf("frpc 子进程退出: %v", err)
 		}
-		// 子进程退出后清理 PID 文件
 		os.Remove("./data/frpc.pid")
 	}()

@@ -234,7 +273,19 @@ func StartFrpc() error {
 	return nil
 }

+// StopFrpc 停止 frpc (幂等)
+// 优先使用 ProcessManager，降级到旧逻辑
 func StopFrpc() error {
+	if processManager != nil {
+		log.Println("[INFO] 使用 ProcessManager 停止 frpc")
+		return processManager.Stop(nil)
+	}
+	log.Println("[WARN] ProcessManager 未初始化，使用兼容模式停止 frpc")
+	return stopFrpcLegacy()
+}
+
+// stopFrpcLegacy 旧版停止逻辑 (降级方案)
+func stopFrpcLegacy() error {
 	if runtime.GOOS == "windows" {
 		cmd := exec.Command("taskkill", "/F", "/IM", "frpc.exe")
 		if err := cmd.Run(); err != nil && !strings.Contains(err.Error(), "not found") {
@@ -268,14 +319,90 @@ func StopFrpc() error {
 	return nil
 }

-func GetFrpcStatus() (bool, error) {
-	return isFrpcRunning(), nil
+// RestartFrpc 重启 frpc (原子操作)
+// 优先使用 ProcessManager，降级到旧逻辑
+func RestartFrpc() error {
+	if processManager != nil {
+		log.Println("[INFO] 使用 ProcessManager 重启 frpc")
+		return processManager.Restart(nil)
+	}
+	log.Println("[WARN] ProcessManager 未初始化，使用兼容模式重启 frpc")
+	if err := stopFrpcLegacy(); err != nil {
+		return err
+	}
+	return startFrpcLegacy()
 }

+// GetFrpcStatus 获取 frpc 详细状态 (供 API 调用)
+// 优先使用 ProcessManager，降级到旧逻辑
+func GetFrpcStatus() (map[string]interface{}, error) {
+	if processManager != nil {
+		status, err := processManager.Status()
+		if err != nil {
+			return nil, err
+		}
+		return map[string]interface{}{
+			"state": status.State,
+			"pid":   status.PID,
+			"port":  status.Port,
+		}, nil
+	}
+
+	// 降级: 返回简单布尔值
+	running := isFrpcRunningLegacy()
+	return map[string]interface{}{
+		"state":   map[bool]string{true: "running", false: "stopped"}[running],
+		"pid":     0,
+		"port":    0,
+		"legacy":  true,
+	}, nil
+}
+
+// ReloadFrpc 热加载 frpc 配置
+// 优先尝试 reload，失败则降级到重启
 func ReloadFrpc() error {
-	running := isFrpcRunning()
-	if !running {
-		return StartFrpc()
+	// 如果 ProcessManager 未初始化，降级到旧逻辑
+	if processManager == nil {
+		return reloadFrpcLegacy()
+	}
+
+	// 检查 frpc 是否在运行
+	status, err := processManager.Status()
+	if err != nil {
+		return fmt.Errorf("获取 frpc 状态失败: %w", err)
+	}
+
+	if status.State != "running" {
+		// 未运行，直接启动
+		return processManager.Start(nil)
+	}
+
+	// 尝试热加载 (使用 frpc reload 命令)
+	frpcPath, err := getFrpcPath()
+	if err != nil {
+		return fmt.Errorf("获取 frpc 路径失败: %w", err)
+	}
+
+	cmd := exec.Command(frpcPath, "reload", "-c", "./data/frpc.toml")
+	output, err := cmd.CombinedOutput()
+	if err != nil {
+		log.Printf("⚠️  热加载失败 (%v)，降级为重启 frpc", err)
+		log.Printf("   reload 输出: %s", string(output))
+		// 降级: 重启
+		if err := processManager.Restart(nil); err != nil {
+			return fmt.Errorf("重启 frpc 失败: %w", err)
+		}
+		return nil
+	}
+
+	log.Printf("✅ frpc 热加载成功")
+	return nil
+}
+
+// reloadFrpcLegacy 旧版热加载 (降级方案)
+func reloadFrpcLegacy() error {
+	if !isFrpcRunningLegacy() {
+		return startFrpcLegacy()
 	}

 	frpcPath, err := getFrpcPath()
@@ -284,20 +411,25 @@ func ReloadFrpc() error {
 	}

 	cmd := exec.Command(frpcPath, "reload", "-c", "./data/frpc.toml")
-	_, err = cmd.CombinedOutput()
+	output, err := cmd.CombinedOutput()
 	if err != nil {
-		log.Printf("⚠️  热加载失败 (%v)，自动降级为重启 frpc", err)
-		if stopErr := StopFrpc(); stopErr != nil {
+		log.Printf("⚠️  热加载失败 (%v)，降级为重启 frpc", err)
+		if stopErr := stopFrpcLegacy(); stopErr != nil {
 			return fmt.Errorf("停止 frpc 失败: %w", stopErr)
 		}
-		if startErr := StartFrpc(); startErr != nil {
+		if startErr := startFrpcLegacy(); startErr != nil {
 			return fmt.Errorf("启动 frpc 失败: %w", startErr)
 		}
 		return nil
 	}
+	log.Printf("✅ frpc 热加载成功 (兼容模式)")
 	return nil
 }

+// ================================================================
+// 日志读取 (保持不变)
+// ================================================================
+
 // readTailLog 读取文件末尾 n 行
 func readTailLog(filePath string, n int) ([]string, error) {
 	file, err := os.Open(filePath)
@@ -370,4 +502,4 @@ func readTailLog(filePath string, n int) ([]string, error) {
 	}

 	return lines, nil
-}
+}