mirror of
https://github.com/Awuqing/BackupX.git
synced 2026-08-31 21:16:46 +08:00
Wu Qing
f6bd185b9f
- fix bare-metal Agent install config and executor path handling - support release package layout in deploy/install.sh and release workflow - add regression tests for Agent execution and deploy install script behavior
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 备份临时目录，默认 /var/lib/backupx-agent/tmp
|
|
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 = "/var/lib/backupx-agent/tmp"
|
|
}
|
|
cfg.Master = strings.TrimRight(strings.TrimSpace(cfg.Master), "/")
|
|
return cfg, nil
|
|
}
|