jpetree331
ec0d880036
- Fix migration 024 self-insert and bad column reference (F-B1-1, F-B1-2) Uses existing enabled/auto_run columns instead of non-existent deprecated - Abort server on migration failure instead of warning (F-B1-11) main.go now calls log.Fatalf, prints [INFO] only on success - Fix migration 018 scanner_config filename suffix (F-B1-3) Renumbered to 027 with .up.sql suffix - Remove GRANT to non-existent role in scanner_config (F-B1-4) - Resolve duplicate migration numbers 009 and 012 (F-B1-13) Renamed to 009b and 012b for unique lexical sorting - Add IF NOT EXISTS to all non-idempotent migrations (F-B1-15) Fixed: 011, 012, 017, 023, 023a - Replace N+1 dashboard stats loop with GetAllUpdateStats (F-B1-6) Single aggregate query replaces per-agent loop - Add composite index on agent_commands(status, sent_at) (F-B1-5) New migration 028 with partial index for timeout service - Add background refresh token cleanup goroutine (F-B1-10) 24-hour ticker calls CleanupExpiredTokens - ETHOS log format in migration runner (no emojis) All 55 tests pass (41 server + 14 agent). No regressions. See docs/B1_Fix_Implementation.md and DEV-025 through DEV-028. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
74 lines
2.3 KiB
Go
74 lines
2.3 KiB
Go
package handlers
|
|
|
|
import (
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/Fimeg/RedFlag/aggregator-server/internal/database/queries"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// StatsHandler handles statistics for the dashboard
|
|
type StatsHandler struct {
|
|
agentQueries *queries.AgentQueries
|
|
updateQueries *queries.UpdateQueries
|
|
}
|
|
|
|
// NewStatsHandler creates a new stats handler
|
|
func NewStatsHandler(agentQueries *queries.AgentQueries, updateQueries *queries.UpdateQueries) *StatsHandler {
|
|
return &StatsHandler{
|
|
agentQueries: agentQueries,
|
|
updateQueries: updateQueries,
|
|
}
|
|
}
|
|
|
|
// DashboardStats represents dashboard statistics
|
|
type DashboardStats struct {
|
|
TotalAgents int `json:"total_agents"`
|
|
OnlineAgents int `json:"online_agents"`
|
|
OfflineAgents int `json:"offline_agents"`
|
|
PendingUpdates int `json:"pending_updates"`
|
|
FailedUpdates int `json:"failed_updates"`
|
|
CriticalUpdates int `json:"critical_updates"`
|
|
ImportantUpdates int `json:"high_updates"`
|
|
ModerateUpdates int `json:"medium_updates"`
|
|
LowUpdates int `json:"low_updates"`
|
|
UpdatesByType map[string]int `json:"updates_by_type"`
|
|
}
|
|
|
|
// GetDashboardStats returns dashboard statistics using aggregate queries (F-B1-6 fix)
|
|
func (h *StatsHandler) GetDashboardStats(c *gin.Context) {
|
|
// Get all agents for online/offline count
|
|
agents, err := h.agentQueries.ListAgents("", "")
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "failed to get agents"})
|
|
return
|
|
}
|
|
|
|
stats := DashboardStats{
|
|
TotalAgents: len(agents),
|
|
UpdatesByType: make(map[string]int),
|
|
}
|
|
|
|
// Count online/offline agents
|
|
for _, agent := range agents {
|
|
if time.Since(agent.LastSeen) <= 10*time.Minute {
|
|
stats.OnlineAgents++
|
|
} else {
|
|
stats.OfflineAgents++
|
|
}
|
|
}
|
|
|
|
// Single aggregate query for all update stats (replaces N+1 per-agent loop)
|
|
updateStats, err := h.updateQueries.GetAllUpdateStats()
|
|
if err == nil {
|
|
stats.PendingUpdates = updateStats.PendingUpdates
|
|
stats.FailedUpdates = updateStats.FailedUpdates
|
|
stats.CriticalUpdates = updateStats.CriticalUpdates
|
|
stats.ImportantUpdates = updateStats.ImportantUpdates
|
|
stats.ModerateUpdates = updateStats.ModerateUpdates
|
|
stats.LowUpdates = updateStats.LowUpdates
|
|
}
|
|
|
|
c.JSON(http.StatusOK, stats)
|
|
} |