重构进行中,preview通道尚未完全恢复全部功能
This commit is contained in:
@@ -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 已自动重启")
|
||||
|
||||
Reference in New Issue
Block a user