mirror of
https://github.com/Awuqing/BackupX.git
synced 2026-05-28 04:59:36 +08:00
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
|
||
}
|