mirror of
https://github.com/Awuqing/BackupX.git
synced 2026-05-11 09:59:56 +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 参考
106 lines
3.6 KiB
Go
106 lines
3.6 KiB
Go
// Package agent 实现 BackupX 远程 Agent。
|
||
//
|
||
// Agent 是一个独立的 Go 进程，部署在远程服务器上，通过 HTTP 轮询的方式
|
||
// 与 Master 通信：定期上报心跳、拉取 Master 下发的命令、本地执行备份、
|
||
// 把执行结果和日志回报给 Master。
|
||
//
|
||
// 通信协议见 server/internal/http/agent_handler.go。
|
||
package agent
|
||
|
||
import (
|
||
"errors"
|
||
"fmt"
|
||
"os"
|
||
"strings"
|
||
|
||
"gopkg.in/yaml.v3"
|
||
)
|
||
|
||
// Config 是 Agent 的运行时配置。
|
||
type Config struct {
|
||
// Master BackupX Master 的 HTTP 基础地址，例如 http://master.example.com:8340
|
||
Master string `yaml:"master"`
|
||
// Token 节点认证令牌（在 Master 创建节点时生成）
|
||
Token string `yaml:"token"`
|
||
// HeartbeatInterval 心跳间隔，默认 15s
|
||
HeartbeatInterval string `yaml:"heartbeatInterval"`
|
||
// PollInterval 命令轮询间隔，默认 5s
|
||
PollInterval string `yaml:"pollInterval"`
|
||
// TempDir 备份临时目录，默认 /tmp/backupx-agent
|
||
TempDir string `yaml:"tempDir"`
|
||
// InsecureSkipTLSVerify 测试环境允许跳过 TLS 证书校验
|
||
InsecureSkipTLSVerify bool `yaml:"insecureSkipTlsVerify"`
|
||
}
|
||
|
||
// LoadConfigFile 从 YAML 文件加载 Agent 配置。
|
||
func LoadConfigFile(path string) (*Config, error) {
|
||
data, err := os.ReadFile(path)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("read agent config: %w", err)
|
||
}
|
||
var cfg Config
|
||
if err := yaml.Unmarshal(data, &cfg); err != nil {
|
||
return nil, fmt.Errorf("parse agent config: %w", err)
|
||
}
|
||
return applyConfigDefaults(&cfg)
|
||
}
|
||
|
||
// LoadConfigFromEnv 从环境变量加载 Agent 配置。优先级低于 --config 文件。
|
||
//
|
||
// 支持的环境变量：
|
||
// - BACKUPX_AGENT_MASTER Master URL
|
||
// - BACKUPX_AGENT_TOKEN 节点认证令牌
|
||
// - BACKUPX_AGENT_HEARTBEAT 心跳间隔（如 15s）
|
||
// - BACKUPX_AGENT_POLL 命令轮询间隔（如 5s）
|
||
// - BACKUPX_AGENT_TEMP_DIR 临时目录
|
||
// - BACKUPX_AGENT_INSECURE_TLS true / 1 跳过 TLS 校验
|
||
func LoadConfigFromEnv() (*Config, error) {
|
||
cfg := &Config{
|
||
Master: strings.TrimSpace(os.Getenv("BACKUPX_AGENT_MASTER")),
|
||
Token: strings.TrimSpace(os.Getenv("BACKUPX_AGENT_TOKEN")),
|
||
HeartbeatInterval: strings.TrimSpace(os.Getenv("BACKUPX_AGENT_HEARTBEAT")),
|
||
PollInterval: strings.TrimSpace(os.Getenv("BACKUPX_AGENT_POLL")),
|
||
TempDir: strings.TrimSpace(os.Getenv("BACKUPX_AGENT_TEMP_DIR")),
|
||
InsecureSkipTLSVerify: strings.EqualFold(os.Getenv("BACKUPX_AGENT_INSECURE_TLS"), "true") || os.Getenv("BACKUPX_AGENT_INSECURE_TLS") == "1",
|
||
}
|
||
return applyConfigDefaults(cfg)
|
||
}
|
||
|
||
// MergeWithFlags 把命令行覆盖值合并入配置（非空覆盖）。
|
||
func (c *Config) MergeWithFlags(master, token, tempDir string) {
|
||
if strings.TrimSpace(master) != "" {
|
||
c.Master = master
|
||
}
|
||
if strings.TrimSpace(token) != "" {
|
||
c.Token = token
|
||
}
|
||
if strings.TrimSpace(tempDir) != "" {
|
||
c.TempDir = tempDir
|
||
}
|
||
}
|
||
|
||
// Validate 校验必填字段。
|
||
func (c *Config) Validate() error {
|
||
if strings.TrimSpace(c.Master) == "" {
|
||
return errors.New("master url is required (set via --master, BACKUPX_AGENT_MASTER or config file)")
|
||
}
|
||
if strings.TrimSpace(c.Token) == "" {
|
||
return errors.New("token is required (set via --token, BACKUPX_AGENT_TOKEN or config file)")
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func applyConfigDefaults(cfg *Config) (*Config, error) {
|
||
if cfg.HeartbeatInterval == "" {
|
||
cfg.HeartbeatInterval = "15s"
|
||
}
|
||
if cfg.PollInterval == "" {
|
||
cfg.PollInterval = "5s"
|
||
}
|
||
if cfg.TempDir == "" {
|
||
cfg.TempDir = "/tmp/backupx-agent"
|
||
}
|
||
cfg.Master = strings.TrimRight(strings.TrimSpace(cfg.Master), "/")
|
||
return cfg, nil
|
||
}
|