mirror of
https://github.com/Awuqing/BackupX.git
synced 2026-08-15 17:34:15 +08:00
5827074334
支持受限网络、正向代理、私有 CA 与 SSH 堡垒机部署 Agent。 加固 Docker、systemd、Nginx、安装器、Release 校验与可信代理边界，并完善命令队列索引、前端安装向导及中英文运维文档。
85 lines
2.4 KiB
Go
85 lines
2.4 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")
|
||
tokenFile := fs.String("token-file", "", "read the agent authentication token from a file")
|
||
tempDir := fs.String("temp-dir", "", "local temp directory for backup artifacts")
|
||
proxyURL := fs.String("proxy-url", "", "HTTP(S) or SOCKS5 proxy used to reach the master")
|
||
caCertFile := fs.String("ca-cert", "", "PEM CA certificate used to verify the master")
|
||
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.ApplyOverrides(agent.Overrides{
|
||
Master: *master,
|
||
Token: *token,
|
||
TokenFile: *tokenFile,
|
||
TempDir: *tempDir,
|
||
ProxyURL: *proxyURL,
|
||
CACertFile: *caCertFile,
|
||
})
|
||
if *insecureTLS {
|
||
cfg.InsecureSkipTLSVerify = true
|
||
}
|
||
if err := cfg.ResolveToken(); err != nil {
|
||
fmt.Fprintf(os.Stderr, "agent: %v\n", err)
|
||
os.Exit(2)
|
||
}
|
||
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()
|
||
}
|