mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 05:56:38 +08:00
ryan
63cd906cfc
- Merge all files inside openflare-server to the repository root directory. - Relocate agent, relay, and flared subprojects from internal/ to internal/apps/. - Combine docker-compose files and update build context paths to root. - Update GitHub workflows and Dockerfiles to refer to new directories and package names. - Rewrite Go package imports across all files. - Resolve database renew test race condition and clean up docs.
122 lines
4.3 KiB
Go
122 lines
4.3 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"log/slog"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"github.com/Rain-kl/Wavelet/internal/apps/agent/agent"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/agent/config"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/agent/geoipupdate"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/agent/heartbeat"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/agent/httpclient"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/agent/logging"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/agent/nginx"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/agent/state"
|
|
syncservice "github.com/Rain-kl/Wavelet/internal/apps/agent/sync"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/agent/updater"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/agent/wsclient"
|
|
)
|
|
|
|
func main() {
|
|
logging.Setup()
|
|
|
|
configPath := flag.String("config", "./agent.json", "agent config path")
|
|
flag.Parse()
|
|
|
|
cfg, err := config.Load(*configPath)
|
|
if err != nil {
|
|
slog.Error("load agent config failed", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
cfg.ExtVersion = nginx.DetectVersion(
|
|
context.Background(),
|
|
nginx.ExecutorOptions{
|
|
NginxPath: cfg.OpenrestyPath,
|
|
MainConfigPath: cfg.MainConfigPath,
|
|
RouteConfigPath: cfg.RouteConfigPath,
|
|
CertDir: cfg.CertDir,
|
|
NginxCertDir: cfg.OpenrestyCertDir,
|
|
LuaDir: cfg.LuaDir,
|
|
NginxLuaDir: cfg.OpenrestyLuaDir,
|
|
OpenrestyObservabilityPort: cfg.OpenrestyObservabilityPort,
|
|
},
|
|
)
|
|
slog.Info("agent config loaded",
|
|
"server", cfg.ServerURL,
|
|
"node", cfg.NodeName,
|
|
"ip", cfg.NodeIP,
|
|
"heartbeat_interval", cfg.HeartbeatInterval,
|
|
"route_config", cfg.RouteConfigPath,
|
|
"access_log", cfg.AccessLogPath,
|
|
"cert_dir", cfg.CertDir,
|
|
"lua_dir", cfg.LuaDir,
|
|
"runtime_config_dir", cfg.RuntimeConfigDir,
|
|
"mmdb_path", cfg.MMDBPath,
|
|
)
|
|
|
|
client := httpclient.New(cfg.ServerURL, cfg.InitialAuthToken(), cfg.RequestTimeout.Duration())
|
|
wsClient := wsclient.New(cfg.ServerURL, cfg.InitialAuthToken(), cfg.RequestTimeout.Duration())
|
|
stateStore := state.NewStore(cfg.StatePath)
|
|
observabilityBuffer := state.NewObservabilityBufferStore(cfg.ObservabilityBufferPath)
|
|
runtimeManager := &nginx.Manager{
|
|
MainConfigPath: cfg.MainConfigPath,
|
|
RouteConfigPath: cfg.RouteConfigPath,
|
|
AccessLogPath: cfg.AccessLogPath,
|
|
CertDir: cfg.CertDir,
|
|
NginxCertDir: cfg.OpenrestyCertDir,
|
|
LuaDir: cfg.LuaDir,
|
|
NginxLuaDir: cfg.OpenrestyLuaDir,
|
|
RuntimeConfigDir: cfg.RuntimeConfigDir,
|
|
PagesDir: cfg.PagesDir,
|
|
OpenrestyObservabilityListen: nginx.ObservabilityListenAddress(cfg.OpenrestyObservabilityPort),
|
|
OpenrestyObservabilityPort: cfg.OpenrestyObservabilityPort,
|
|
OpenrestyResolverDirective: "",
|
|
Executor: nginx.NewExecutor(nginx.ExecutorOptions{
|
|
NginxPath: cfg.OpenrestyPath,
|
|
MainConfigPath: cfg.MainConfigPath,
|
|
RouteConfigPath: cfg.RouteConfigPath,
|
|
CertDir: cfg.CertDir,
|
|
NginxCertDir: cfg.OpenrestyCertDir,
|
|
LuaDir: cfg.LuaDir,
|
|
NginxLuaDir: cfg.OpenrestyLuaDir,
|
|
OpenrestyObservabilityPort: cfg.OpenrestyObservabilityPort,
|
|
}),
|
|
}
|
|
if err = runtimeManager.EnsureLuaAssets(); err != nil {
|
|
slog.Error("ensure managed lua assets failed", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
syncService := syncservice.New(client, runtimeManager, stateStore)
|
|
syncService.SetPagesDir(cfg.PagesDir)
|
|
runner := &agent.Runner{
|
|
Config: cfg,
|
|
StateStore: stateStore,
|
|
ObservabilityBuffer: observabilityBuffer,
|
|
HeartbeatService: heartbeat.New(client),
|
|
SyncService: syncService,
|
|
Updater: updater.New(),
|
|
RuntimeManager: runtimeManager,
|
|
WebSocketService: wsClient,
|
|
}
|
|
|
|
ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
|
defer stop()
|
|
geoIPUpdater := &geoipupdate.Updater{
|
|
MMDBPath: cfg.MMDBPath,
|
|
DownloadURL: cfg.MMDBDownloadURL,
|
|
UpdateInterval: cfg.MMDBUpdateInterval.Duration(),
|
|
}
|
|
go geoIPUpdater.Run(ctx)
|
|
slog.Info("agent process started")
|
|
|
|
if err = runner.Run(ctx); err != nil && err != context.Canceled {
|
|
slog.Error("agent process exited with error", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
slog.Info("agent process stopped")
|
|
}
|