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}
102 lines
2.8 KiB
Go
102 lines
2.8 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// Package sync periodically fetches and applies the active tunnel configuration.
|
|
package sync
|
|
|
|
import (
|
|
"context"
|
|
"log/slog"
|
|
"time"
|
|
|
|
"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"
|
|
service "github.com/Rain-kl/Wavelet/pkg/protocol"
|
|
)
|
|
|
|
// Service synchronizes tunnel configuration from the control plane to the local frpc manager.
|
|
type Service struct {
|
|
client *httpclient.Client
|
|
frpcManager *frpc.Manager
|
|
config *config.Config
|
|
triggerCh chan struct{}
|
|
}
|
|
|
|
// New creates a sync 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,
|
|
triggerCh: make(chan struct{}, 1),
|
|
}
|
|
}
|
|
|
|
// Trigger requests an immediate configuration sync without waiting for the next interval.
|
|
func (s *Service) Trigger() {
|
|
select {
|
|
case s.triggerCh <- struct{}{}:
|
|
default:
|
|
}
|
|
}
|
|
|
|
// Run starts the sync loop until ctx is canceled.
|
|
func (s *Service) Run(ctx context.Context) {
|
|
ticker := time.NewTicker(s.config.SyncInterval.Duration())
|
|
defer ticker.Stop()
|
|
|
|
// initial sync
|
|
s.doSync(ctx)
|
|
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return
|
|
case <-ticker.C:
|
|
s.doSync(ctx)
|
|
case <-s.triggerCh:
|
|
s.doSync(ctx)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (s *Service) doSync(ctx context.Context) {
|
|
slog.Debug("fetching active tunnel config")
|
|
configResp, err := s.client.GetActiveConfig(ctx)
|
|
if err != nil {
|
|
slog.Error("failed to fetch active tunnel config", "error", err)
|
|
return
|
|
}
|
|
|
|
// 不在 sync 层做版本早退，由 frpcManager.UpdateConfig 负责判断。
|
|
// 原因：重启后进程全部消失，即使版本/checksum 未变，仍需重新拉起 frpc 进程。
|
|
configChanged, err := s.frpcManager.UpdateConfig(ctx, configResp)
|
|
if err != nil {
|
|
slog.Error("failed to apply tunnel config", "error", err)
|
|
s.reportApplyLog(ctx, configResp, "failed", err.Error())
|
|
return
|
|
}
|
|
if configChanged {
|
|
slog.Info("tunnel config applied successfully", "version", configResp.Version)
|
|
s.reportApplyLog(ctx, configResp, "success", "apply success")
|
|
return
|
|
}
|
|
slog.Debug("tunnel config unchanged, skipping apply log report", "version", configResp.Version)
|
|
}
|
|
|
|
func (s *Service) reportApplyLog(ctx context.Context, configResp *service.FlaredTunnelConfigResponse, result string, message string) {
|
|
if configResp == nil {
|
|
return
|
|
}
|
|
logPayload := service.ApplyLogPayload{
|
|
Version: configResp.Version,
|
|
Result: result,
|
|
Message: message,
|
|
Checksum: configResp.Checksum,
|
|
}
|
|
if reportErr := s.client.ReportApplyLog(ctx, logPayload); reportErr != nil {
|
|
slog.Error("failed to report apply log", "error", reportErr)
|
|
}
|
|
}
|