mirror of
https://github.com/Awuqing/BackupX.git
synced 2026-05-07 04:52:52 +08:00
757b0fa5ed
基础修复： - 新增节点离线检测：每 15s 扫描，超 45s 未心跳的远程节点自动置离线 - 节点删除前检查关联任务，避免孤立备份任务 - BackupTaskRepository 新增 CountByNodeID/ListByNodeID Master 端 Agent 协议： - 新增 AgentCommand 模型与命令队列仓储（pending/dispatched/succeeded/failed/timeout） - 新增 AgentService：任务下发、命令轮询、结果回收、超时扫描 - 新增专用 Agent HTTP API（X-Agent-Token 认证）： /api/agent/heartbeat /api/agent/commands/poll /api/agent/commands/:id/result /api/agent/tasks/:id /api/agent/records/:id - BackupExecutionService 支持 node 路由：task.NodeID 指向远程节点时自动入队派发 Agent CLI（backupx agent 子命令）： - 配置：YAML 文件 / 环境变量 / CLI 参数，优先级 CLI > 文件 > 环境 - 心跳循环 + 命令轮询循环 + 优雅退出 - 本地复用 BackupRunner 与 storage registry 执行备份并直接上传 - 支持 run_task 和 list_dir 两种命令 远程目录浏览： - NodeService 支持通过 Agent RPC 列出远程节点目录（15s 超时） 前端： - NodesPage 添加节点后展示 Agent 启动命令和环境变量配置 文档： - README 中英文重写"多节点集群"章节，含架构图、步骤、限制、CLI 参考
71 lines
1.9 KiB
Go
71 lines
1.9 KiB
Go
package main
|
||
|
||
import (
|
||
"context"
|
||
"flag"
|
||
"fmt"
|
||
"os"
|
||
"os/signal"
|
||
"syscall"
|
||
|
||
"backupx/server/internal/agent"
|
||
)
|
||
|
||
// runAgent 是 `backupx agent` 子命令入口。
|
||
//
|
||
// 用法：
|
||
//
|
||
// backupx agent --master http://master:8340 --token <token>
|
||
// backupx agent --config /etc/backupx-agent.yaml
|
||
//
|
||
// 配置优先级：CLI 参数 > 配置文件 > 环境变量
|
||
func runAgent(args []string) {
|
||
fs := flag.NewFlagSet("agent", flag.ExitOnError)
|
||
configPath := fs.String("config", "", "path to agent config YAML (optional)")
|
||
master := fs.String("master", "", "master URL, e.g. http://master.example.com:8340")
|
||
token := fs.String("token", "", "agent authentication token")
|
||
tempDir := fs.String("temp-dir", "", "local temp directory for backup artifacts")
|
||
insecureTLS := fs.Bool("insecure-tls", false, "skip TLS verification (testing only)")
|
||
|
||
if err := fs.Parse(args); err != nil {
|
||
os.Exit(2)
|
||
}
|
||
|
||
cfg, err := loadAgentConfig(*configPath)
|
||
if err != nil {
|
||
fmt.Fprintf(os.Stderr, "agent: load config: %v\n", err)
|
||
os.Exit(2)
|
||
}
|
||
cfg.MergeWithFlags(*master, *token, *tempDir)
|
||
if *insecureTLS {
|
||
cfg.InsecureSkipTLSVerify = true
|
||
}
|
||
if err := cfg.Validate(); err != nil {
|
||
fmt.Fprintf(os.Stderr, "agent: %v\n", err)
|
||
os.Exit(2)
|
||
}
|
||
|
||
a, err := agent.New(cfg, version)
|
||
if err != nil {
|
||
fmt.Fprintf(os.Stderr, "agent: init: %v\n", err)
|
||
os.Exit(1)
|
||
}
|
||
|
||
ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
||
defer stop()
|
||
|
||
fmt.Fprintf(os.Stderr, "backupx agent %s starting (master=%s)\n", version, cfg.Master)
|
||
if err := a.Run(ctx); err != nil && err != context.Canceled {
|
||
fmt.Fprintf(os.Stderr, "agent: %v\n", err)
|
||
os.Exit(1)
|
||
}
|
||
}
|
||
|
||
// loadAgentConfig 按优先级加载配置：如果提供了 --config 就用文件，否则走环境变量。
|
||
func loadAgentConfig(configPath string) (*agent.Config, error) {
|
||
if configPath != "" {
|
||
return agent.LoadConfigFile(configPath)
|
||
}
|
||
return agent.LoadConfigFromEnv()
|
||
}
|