mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-30 22:26:38 +08:00
ryan
75 lines
2.3 KiB
Go
75 lines
2.3 KiB
Go
// Package flared implements the tunnel client daemon runtime loop.
|
|
package flared
|
|
|
|
import (
|
|
"context"
|
|
"log/slog"
|
|
|
|
edgerunner "github.com/Rain-kl/Wavelet/internal/apps/edge/runner"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/flared/config"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/flared/frpc"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/flared/heartbeat"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/flared/httpclient"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/flared/sync"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/flared/wsclient"
|
|
)
|
|
|
|
// Runner is the top-level orchestrator for the flared agent. It wires together
|
|
// heartbeat, sync, frpc management, and the WebSocket control-plane connection.
|
|
type Runner struct {
|
|
Config *config.Config
|
|
HeartbeatService *heartbeat.Service
|
|
FrpcManager *frpc.Manager
|
|
SyncService *sync.Service
|
|
WebSocketService *wsclient.Client
|
|
HTTPClient *httpclient.Client
|
|
}
|
|
|
|
// Run starts all background services and enters the WebSocket reconnect loop.
|
|
// It blocks until ctx is cancelled or an unrecoverable error occurs.
|
|
func (r *Runner) Run(ctx context.Context) error {
|
|
go r.HeartbeatService.Run(ctx)
|
|
go r.SyncService.Run(ctx)
|
|
|
|
return edgerunner.RunWSReconnectLoop(ctx, edgerunner.WSReconnectConfig{
|
|
ComponentName: "flared",
|
|
OnShutdown: r.FrpcManager.Stop,
|
|
}, func(ctx context.Context) (edgerunner.WSConnection, error) {
|
|
return r.WebSocketService.Connect(ctx)
|
|
}, func(ctx context.Context, conn edgerunner.WSConnection) {
|
|
r.handleConnection(ctx, conn)
|
|
})
|
|
}
|
|
|
|
type flaredWSHandler struct {
|
|
runner *Runner
|
|
}
|
|
|
|
func (h *flaredWSHandler) OnConnect(_ context.Context) error {
|
|
return nil
|
|
}
|
|
|
|
func (h *flaredWSHandler) HandleMessage(_ context.Context, msg wsclient.WSMessage) error {
|
|
switch msg.Type {
|
|
case "active_config":
|
|
slog.Info("received config update notification from server")
|
|
h.runner.SyncService.Trigger()
|
|
default:
|
|
slog.Debug("ignored unknown ws message type", "type", msg.Type)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (h *flaredWSHandler) OnClose(err error) {
|
|
slog.Error("flared ws receive failed", "error", err)
|
|
}
|
|
|
|
func (r *Runner) handleConnection(ctx context.Context, conn edgerunner.WSConnection) {
|
|
wsConn, ok := conn.(*wsclient.Connection)
|
|
if !ok {
|
|
slog.Error("flared ws connection has unexpected type")
|
|
return
|
|
}
|
|
_ = wsConn.RunReceiveLoop(ctx, &flaredWSHandler{runner: r})
|
|
}
|