lerko
70a83a1da9
Every Store interface method (except Close) now takes context.Context as first parameter. All 54 db.Query/Exec/QueryRow calls in SQLStore replaced with their *Context variants. DB operations now respect cancellation and deadlines. Context sources by caller: - Engine dbWriter/poll/pruner: engine ctx from Start() - HTTP handlers: r.Context() - config.Apply/Export: caller-provided ctx - TUI/main.go init: context.Background() RunCheck and all sub-checks (HTTP/ping/port/DNS) accept parent ctx. HTTP checks now inherit shutdown cancellation instead of rooting in context.Background(). dbWrite.exec takes ctx so the writer goroutine can cancel stuck DB operations. DeleteSite/ImportData use BeginTx(ctx) instead of Begin().
87 lines
3.6 KiB
Go
87 lines
3.6 KiB
Go
package store
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"gitea.lerkolabs.com/lerkolabs/uptop/internal/models"
|
|
)
|
|
|
|
type Store interface {
|
|
Init(ctx context.Context) error
|
|
|
|
// Sites
|
|
GetSites(ctx context.Context) ([]models.Site, error)
|
|
AddSite(ctx context.Context, site models.Site) error
|
|
UpdateSite(ctx context.Context, site models.Site) error
|
|
UpdateSitePaused(ctx context.Context, id int, paused bool) error
|
|
DeleteSite(ctx context.Context, id int) error
|
|
|
|
// Alerts
|
|
GetAllAlerts(ctx context.Context) ([]models.AlertConfig, error)
|
|
GetAlert(ctx context.Context, id int) (models.AlertConfig, error)
|
|
AddAlert(ctx context.Context, name, aType string, settings map[string]string) error
|
|
UpdateAlert(ctx context.Context, id int, name, aType string, settings map[string]string) error
|
|
DeleteAlert(ctx context.Context, id int) error
|
|
|
|
// Declarative config support
|
|
GetSiteByName(ctx context.Context, name string) (models.Site, error)
|
|
GetAlertByName(ctx context.Context, name string) (models.AlertConfig, error)
|
|
AddSiteReturningID(ctx context.Context, site models.Site) (int, error)
|
|
AddAlertReturningID(ctx context.Context, name, aType string, settings map[string]string) (int, error)
|
|
|
|
// Users
|
|
GetAllUsers(ctx context.Context) ([]models.User, error)
|
|
AddUser(ctx context.Context, username, publicKey, role string) error
|
|
UpdateUser(ctx context.Context, id int, username, publicKey, role string) error
|
|
DeleteUser(ctx context.Context, id int) error
|
|
|
|
// History
|
|
SaveCheck(ctx context.Context, siteID int, latencyNs int64, isUp bool) error
|
|
SaveCheckFromNode(ctx context.Context, siteID int, nodeID string, latencyNs int64, isUp bool) error
|
|
LoadAllHistory(ctx context.Context, limit int) (map[int][]models.CheckRecord, error)
|
|
PruneCheckHistory(ctx context.Context) error
|
|
|
|
// State Changes
|
|
SaveStateChange(ctx context.Context, siteID int, fromStatus, toStatus, errorReason string) error
|
|
GetStateChanges(ctx context.Context, siteID int, limit int) ([]models.StateChange, error)
|
|
GetStateChangesSince(ctx context.Context, siteID int, since time.Time) ([]models.StateChange, error)
|
|
PruneStateChanges(ctx context.Context) error
|
|
|
|
// Nodes
|
|
RegisterNode(ctx context.Context, node models.ProbeNode) error
|
|
GetNode(ctx context.Context, id string) (models.ProbeNode, error)
|
|
GetAllNodes(ctx context.Context) ([]models.ProbeNode, error)
|
|
UpdateNodeLastSeen(ctx context.Context, id string) error
|
|
DeleteNode(ctx context.Context, id string) error
|
|
|
|
// Alert Health
|
|
LoadAlertHealth(ctx context.Context) (map[int]models.AlertHealthRecord, error)
|
|
SaveAlertHealth(ctx context.Context, h models.AlertHealthRecord) error
|
|
|
|
// Logs
|
|
SaveLog(ctx context.Context, message string) error
|
|
LoadLogs(ctx context.Context, limit int) ([]string, error)
|
|
PruneLogs(ctx context.Context) error
|
|
|
|
// Maintenance Windows
|
|
GetActiveMaintenanceWindows(ctx context.Context) ([]models.MaintenanceWindow, error)
|
|
GetAllMaintenanceWindows(ctx context.Context, limit int) ([]models.MaintenanceWindow, error)
|
|
AddMaintenanceWindow(ctx context.Context, mw models.MaintenanceWindow) error
|
|
EndMaintenanceWindow(ctx context.Context, id int) error
|
|
DeleteMaintenanceWindow(ctx context.Context, id int) error
|
|
PruneExpiredMaintenanceWindows(ctx context.Context, retention time.Duration) (int64, error)
|
|
IsMonitorInMaintenance(ctx context.Context, monitorID int) (bool, error)
|
|
|
|
// Preferences
|
|
GetPreference(ctx context.Context, key string) (string, error)
|
|
SetPreference(ctx context.Context, key, value string) error
|
|
|
|
// Backup & Restore
|
|
ExportData(ctx context.Context) (models.Backup, error)
|
|
ImportData(ctx context.Context, data models.Backup) error
|
|
|
|
// Lifecycle
|
|
Close() error
|
|
}
|