mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-10-01 06:36:38 +08:00
ryan
2f60329886
代码质量全量清理，零行为变化：golangci 扩展集 13 类 linter（gosec/modernize/perfsprint/canonicalheader/usestdlibvars/wastedassign/intrange/errorlint/forcetypeassert/recvcheck/exhaustive/unparam）全量修复，测试代码质量（testifylint/thelper/usetesting）25→0，frpc 进程生命周期真 bug（进程组击杀）、全仓 go test -race 6 类数据竞争（含 1 个生产竞争）、SPDX license 头补齐 131 文件、前端测试套件 next-intl 迁移后 44 失败→全绿、过期 swagger 文档重新生成、pnpm-workspace 构建审批。 Experiments: #2-#17, #18, #20, #21, #23 Metric: total_issues 108 → 8 (-92.6%)
78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// Command relay runs the OpenFlare relay node daemon.
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"flag"
|
|
"log/slog"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
edgelogging "github.com/Rain-kl/Wavelet/internal/apps/edge/logging"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/relay/config"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/relay/frps"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/relay/heartbeat"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/relay/httpclient"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/relay/relay"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/relay/state"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/relay/wsclient"
|
|
)
|
|
|
|
func main() {
|
|
edgelogging.Setup(edgelogging.Options{})
|
|
|
|
configPath := flag.String("config", "./relay.json", "relay config path")
|
|
flag.Parse()
|
|
|
|
cfg, err := config.Load(*configPath)
|
|
if err != nil {
|
|
slog.Error("load relay config failed", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
slog.Info("relay config loaded",
|
|
"server", cfg.ServerURL,
|
|
"node", cfg.NodeName,
|
|
"ip", cfg.NodeIP,
|
|
"frps_path", cfg.FrpsPath,
|
|
"data_dir", cfg.DataDir,
|
|
"heartbeat_interval", cfg.HeartbeatInterval,
|
|
)
|
|
|
|
stateStore := state.NewStore(cfg.StatePath)
|
|
_ = stateStore // In the future we may use stateStore for auth caching
|
|
|
|
frpsManager := frps.NewManager(cfg.FrpsPath, cfg.DataDir, cfg.InitialAuthToken())
|
|
|
|
slog.Info("detected frps version", "version", frpsManager.GetVersion(context.Background()))
|
|
|
|
httpClient := httpclient.New(cfg.ServerURL, cfg.InitialAuthToken(), cfg.RequestTimeout.Duration())
|
|
wsClient := wsclient.New(cfg.ServerURL, cfg.InitialAuthToken(), cfg.RequestTimeout.Duration())
|
|
|
|
runner := &relay.Runner{
|
|
Config: cfg,
|
|
StateStore: stateStore,
|
|
FrpsManager: frpsManager,
|
|
HTTPClient: httpClient,
|
|
WebSocketService: wsClient,
|
|
HeartbeatService: heartbeat.New(httpClient, frpsManager, cfg, stateStore),
|
|
}
|
|
|
|
ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
|
|
|
slog.Info("relay process started")
|
|
|
|
if err := runner.Run(ctx); err != nil && !errors.Is(err, context.Canceled) {
|
|
slog.Error("relay process exited with error", "error", err)
|
|
stop()
|
|
os.Exit(1)
|
|
}
|
|
stop()
|
|
slog.Info("relay process stopped")
|
|
}
|