[优化] Phase4

openflared/internal/sync/service.go

@@ -0,0 +1,92 @@
+package sync
+
+import (
+	"context"
+	"log/slog"
+	"time"
+
+	"openflare-flared/internal/config"
+	"openflare-flared/internal/frpc"
+	"openflare-flared/internal/httpclient"
+	"openflare/service"
+)
+
+type Service struct {
+	client      *httpclient.Client
+	frpcManager *frpc.Manager
+	config      *config.Config
+	triggerCh   chan struct{}
+}
+
+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),
+	}
+}
+
+func (s *Service) Trigger() {
+	select {
+	case s.triggerCh <- struct{}{}:
+	default:
+	}
+}
+
+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
+	}
+
+	if s.frpcManager.GetCurrentConfigVersion() == configResp.Version &&
+		s.frpcManager.GetCurrentConfigChecksum() == configResp.Checksum {
+		slog.Debug("tunnel config is up to date", "version", configResp.Version)
+		return
+	}
+
+	err = s.frpcManager.UpdateConfig(ctx, configResp)
+
+	result := "success"
+	message := "apply success"
+	if err != nil {
+		result = "failed"
+		message = err.Error()
+		slog.Error("failed to apply tunnel config", "error", err)
+	} else {
+		slog.Info("tunnel config applied successfully", "version", configResp.Version)
+	}
+
+	// Report apply log
+	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)
+	}
+}