tcjlj/OpenFlare
1
0
Fork 0
mirror of https://github.com/Rain-kl/OpenFlare.git synced 2026-10-07 08:06:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: implement heartbeat and sync services with error handling in Runner

Browse Source
This commit is contained in:
ryan
2026-03-10 14:44:05 +08:00
parent 7be2da0c19
commit f396c8c74e
3 changed files with 449 additions and 248 deletions
Download Patch File Download Diff File Expand all files Collapse all files
atsf_agent/internal/agent/runner.go
+39 -10
View File
@@ -2,20 +2,29 @@ package agent
import (
"context"
"log"
"time"
"atsflare-agent/internal/config"
"atsflare-agent/internal/heartbeat"
"atsflare-agent/internal/protocol"
"atsflare-agent/internal/state"
syncservice "atsflare-agent/internal/sync"
)
type HeartbeatService interface {
Register(ctx context.Context, payload protocol.NodePayload) error
Heartbeat(ctx context.Context, payload protocol.NodePayload) error
}
type SyncService interface {
SyncOnStartup(ctx context.Context) error
SyncOnce(ctx context.Context) error
}
type Runner struct {
Config *config.Config
StateStore *state.Store
HeartbeatService *heartbeat.Service
SyncService *syncservice.Service
HeartbeatService HeartbeatService
SyncService SyncService
}
func (r *Runner) Run(ctx context.Context) error {
@@ -23,11 +32,15 @@ func (r *Runner) Run(ctx context.Context) error {
if err != nil {
return err
}
if err = r.SyncService.SyncOnStartup(ctx); err != nil {
return err
}
if err = r.HeartbeatService.Register(ctx, r.nodePayload(nodeID)); err != nil {
return err
log.Printf("agent register failed: %v", err)
}
if err = r.SyncService.SyncOnStartup(ctx); err != nil {
r.recordSyncError(err)
log.Printf("agent startup sync failed: %v", err)
}
if err = r.HeartbeatService.Heartbeat(ctx, r.nodePayload(nodeID)); err != nil {
log.Printf("agent startup heartbeat failed: %v", err)
}
heartbeatTicker := time.NewTicker(r.Config.HeartbeatInterval)
@@ -41,16 +54,32 @@ func (r *Runner) Run(ctx context.Context) error {
return ctx.Err()
case <-heartbeatTicker.C:
if err = r.HeartbeatService.Heartbeat(ctx, r.nodePayload(nodeID)); err != nil {
return err
log.Printf("agent heartbeat failed: %v", err)
}
case <-syncTicker.C:
if err = r.SyncService.SyncOnce(ctx); err != nil {
return err
r.recordSyncError(err)
log.Printf("agent sync failed: %v", err)
}
}
}
}
func (r *Runner) recordSyncError(err error) {
if err == nil || r.StateStore == nil {
return
}
snapshot, loadErr := r.StateStore.Load()
if loadErr != nil {
log.Printf("load state before recording sync error failed: %v", loadErr)
return
}
snapshot.LastError = err.Error()
if saveErr := r.StateStore.Save(snapshot); saveErr != nil {
log.Printf("save state after sync error failed: %v", saveErr)
}
}
func (r *Runner) nodePayload(nodeID string) protocol.NodePayload {
snapshot, _ := r.StateStore.Load()
return protocol.NodePayload{