mirror of
https://github.com/MeshCore-Beacon/beacon-server.git
synced 2026-09-01 16:48:19 +00:00
94 lines
2.8 KiB
Go
94 lines
2.8 KiB
Go
// Copyright 2026 Beacon Contributors
|
|
// SPDX-License-Identifier: AGPL-3.0-or-later
|
|
|
|
package background
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"time"
|
|
|
|
"github.com/MeshCore-Beacon/beacon-server/db"
|
|
)
|
|
|
|
// ViewRefreshTask returns a Task that refreshes all materialized views.
|
|
func ViewRefreshTask(store *db.Store, interval time.Duration) Task {
|
|
return Task{
|
|
Name: "view_refresh",
|
|
Interval: interval,
|
|
Run: func(ctx context.Context) error {
|
|
if err := store.RefreshHourlyStats(ctx); err != nil {
|
|
log.Printf("background[view_refresh]: hourly stats: %v", err)
|
|
}
|
|
if err := store.RefreshTopNodes(ctx); err != nil {
|
|
log.Printf("background[view_refresh]: top nodes: %v", err)
|
|
}
|
|
if err := store.RefreshTopObservers(ctx); err != nil {
|
|
log.Printf("background[view_refresh]: top observers: %v", err)
|
|
}
|
|
if err := store.RefreshPayloadBreakdown(ctx); err != nil {
|
|
log.Printf("background[view_refresh]: payload breakdown: %v", err)
|
|
}
|
|
if err := store.RefreshTopTalkers(ctx); err != nil {
|
|
log.Printf("background[view_refresh]: top talkers: %v", err)
|
|
}
|
|
if err := store.RefreshTopAdvertisers(ctx); err != nil {
|
|
log.Printf("background[view_refresh]: top advertisers: %v", err)
|
|
}
|
|
if err := store.RefreshRadioPresets(ctx); err != nil {
|
|
log.Printf("background[view_refresh]: radio presets: %v", err)
|
|
}
|
|
return nil
|
|
},
|
|
}
|
|
}
|
|
|
|
// CleanupTask returns a Task that prunes old telemetry, packet, and node rows.
|
|
func CleanupTask(store *db.Store, telemetryRetention, packetRetention, nodeDeleteAfter, interval time.Duration) Task {
|
|
return Task{
|
|
Name: "cleanup",
|
|
Interval: interval,
|
|
Run: func(ctx context.Context) error {
|
|
if err := store.DeleteOldTelemetry(ctx, time.Now().Add(-telemetryRetention)); err != nil {
|
|
return err
|
|
}
|
|
// One cutoff for all three so the IATA tables stay in step
|
|
// with the packets they mirror.
|
|
cutoff := time.Now().Add(-packetRetention)
|
|
if err := store.DeleteOldPackets(ctx, cutoff); err != nil {
|
|
return err
|
|
}
|
|
if err := store.DeleteOldChannelIATAs(ctx, cutoff); err != nil {
|
|
return err
|
|
}
|
|
if err := store.DeleteOldTraceIATAs(ctx, cutoff); err != nil {
|
|
return err
|
|
}
|
|
if err := store.DeleteOldNodes(ctx, time.Now().Add(-nodeDeleteAfter)); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
},
|
|
}
|
|
}
|
|
|
|
// ReconfirmTask returns a Task that prunes stale and ambiguous resolved paths
|
|
// and neighbors. Runs after routes to ensure neighbors are cleaned against
|
|
// already-reconfirmed path data.
|
|
func ReconfirmTask(store *db.Store, interval time.Duration) Task {
|
|
return Task{
|
|
Name: "reconfirm",
|
|
Interval: interval,
|
|
Run: func(ctx context.Context) error {
|
|
if err := store.ReconfirmRoutes(ctx); err != nil {
|
|
return fmt.Errorf("routes: %w", err)
|
|
}
|
|
if err := store.ReconfirmNeighbors(ctx); err != nil {
|
|
return fmt.Errorf("neighbors: %w", err)
|
|
}
|
|
return nil
|
|
},
|
|
}
|
|
}
|