mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-28 05:46: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}
80 lines
2.5 KiB
Go
80 lines
2.5 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// Package heartbeat runs the periodic flared heartbeat loop against the control plane.
|
|
package heartbeat
|
|
|
|
import (
|
|
"context"
|
|
"log/slog"
|
|
|
|
edgeheartbeat "github.com/Rain-kl/Wavelet/internal/apps/edge/heartbeat"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/edge/nodeip"
|
|
"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/httpclient"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/flared/updater"
|
|
service "github.com/Rain-kl/Wavelet/pkg/protocol"
|
|
)
|
|
|
|
// Service sends periodic heartbeat payloads and applies tunnel settings from responses.
|
|
type Service struct {
|
|
client *httpclient.Client
|
|
frpcManager *frpc.Manager
|
|
config *config.Config
|
|
updater *updater.Service
|
|
}
|
|
|
|
// New creates a heartbeat service with the given client, frpc manager, and config.
|
|
func New(client *httpclient.Client, manager *frpc.Manager, cfg *config.Config) *Service {
|
|
return &Service{
|
|
client: client,
|
|
frpcManager: manager,
|
|
config: cfg,
|
|
updater: updater.New(),
|
|
}
|
|
}
|
|
|
|
// Run starts the heartbeat loop until ctx is canceled.
|
|
func (s *Service) Run(ctx context.Context) {
|
|
edgeheartbeat.RunLoop(ctx, s.config.HeartbeatInterval.Duration(), s.doHeartbeat)
|
|
}
|
|
|
|
func (s *Service) doHeartbeat(ctx context.Context) {
|
|
slog.Debug("sending flared heartbeat")
|
|
|
|
payload := service.FlaredHeartbeatPayload{
|
|
ClientVersion: config.Version,
|
|
FrpVersion: s.frpcManager.GetVersion(ctx),
|
|
IP: nodeip.DetectWithContext(ctx),
|
|
TunnelStatus: "running",
|
|
ConnectedRelays: s.frpcManager.GetConnectedRelays(),
|
|
CurrentVersion: s.frpcManager.GetCurrentConfigVersion(),
|
|
CurrentChecksum: s.frpcManager.GetCurrentConfigChecksum(),
|
|
}
|
|
|
|
resp, err := s.client.Heartbeat(ctx, payload)
|
|
if err != nil {
|
|
slog.Error("flared heartbeat failed", "error", err)
|
|
return
|
|
}
|
|
slog.Debug("flared heartbeat succeeded")
|
|
|
|
if resp != nil && resp.TunnelSettings != nil {
|
|
edgeheartbeat.TryAutoUpdate(ctx, s.updater, tunnelSettingsToAutoUpdate(resp.TunnelSettings), "flared")
|
|
}
|
|
}
|
|
|
|
func tunnelSettingsToAutoUpdate(settings *service.RelaySettings) *edgeheartbeat.AutoUpdateSettings {
|
|
if settings == nil {
|
|
return nil
|
|
}
|
|
return &edgeheartbeat.AutoUpdateSettings{
|
|
AutoUpdate: settings.AutoUpdate,
|
|
UpdateNow: settings.UpdateNow,
|
|
UpdateRepo: settings.UpdateRepo,
|
|
UpdateChannel: settings.UpdateChannel,
|
|
UpdateTag: settings.UpdateTag,
|
|
}
|
|
}
|