mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-30 14:26:36 +08:00
ryan
aa4faddade
Result: {"status":"keep","total_issues":8,"golint_canonicalheader":0,"golint_errname":0,"golint_errorlint":1,"golint_exhaustive":0,"golint_forcetypeassert":0,"golint_gosec":0,"golint_intrange":0,"golint_modernize":3,"golint_nilnil":3,"golint_perfsprint":0,"golint_prealloc":0,"golint_recvcheck":1,"golint_usestdlibvars":0,"golint_wastedassign":0,"golint_total":8,"golint_test_testifylint":0,"golint_test_thelper":0,"golint_test_usetesting":0,"golint_test_total":0,"golint_vetx_total":0,"eslint_problems":0,"eslint_errors":0,"eslint_warnings":0,"tsc_errors":0,"vitest_failed":0,"vitest_total":116,"measure_s":81}
78 lines
2.4 KiB
Go
78 lines
2.4 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// 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})
|
|
}
|