RomanNum3ral
/
Cybersecurity-Projects
mirror of https://github.com/CarterPerez-dev/Cybersecurity-Projects.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat(monitor/main): wire bus + hub + snapshot + heartbeat under errgroup; expose /v1/snapshot, /v1/ws, /v1/healthz

This commit is contained in:
CarterPerez-dev 2026-05-01 13:53:16 -04:00
parent a7865533cd
commit a67ef20680
1 changed files with 41 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
PROJECTS/advanced/monitor-the-situation-dashboard/backend/cmd/api/main.go
View File

@ -5,6 +5,7 @@ package main
import ( import (
"context" "context"
"errors"
"flag" "flag"
"log/slog" "log/slog"
"os" "os"
@ -13,15 +14,20 @@ import (
"time" "time"
"github.com/go-chi/chi/v5" "github.com/go-chi/chi/v5"
"golang.org/x/sync/errgroup"
"github.com/carterperez-dev/monitor-the-situation/backend/internal/admin" "github.com/carterperez-dev/monitor-the-situation/backend/internal/admin"
"github.com/carterperez-dev/monitor-the-situation/backend/internal/auth" "github.com/carterperez-dev/monitor-the-situation/backend/internal/auth"
"github.com/carterperez-dev/monitor-the-situation/backend/internal/bus"
"github.com/carterperez-dev/monitor-the-situation/backend/internal/collectors/heartbeat"
"github.com/carterperez-dev/monitor-the-situation/backend/internal/config" "github.com/carterperez-dev/monitor-the-situation/backend/internal/config"
"github.com/carterperez-dev/monitor-the-situation/backend/internal/core" "github.com/carterperez-dev/monitor-the-situation/backend/internal/core"
"github.com/carterperez-dev/monitor-the-situation/backend/internal/health" "github.com/carterperez-dev/monitor-the-situation/backend/internal/health"
"github.com/carterperez-dev/monitor-the-situation/backend/internal/middleware" "github.com/carterperez-dev/monitor-the-situation/backend/internal/middleware"
"github.com/carterperez-dev/monitor-the-situation/backend/internal/server" "github.com/carterperez-dev/monitor-the-situation/backend/internal/server"
"github.com/carterperez-dev/monitor-the-situation/backend/internal/snapshot"
"github.com/carterperez-dev/monitor-the-situation/backend/internal/user" "github.com/carterperez-dev/monitor-the-situation/backend/internal/user"
"github.com/carterperez-dev/monitor-the-situation/backend/internal/ws"
) )
const ( const (
@ -117,6 +123,29 @@ func run(configPath string) error {
RedisPing: redis.Ping, RedisPing: redis.Ping,
}) })
snapStore := snapshot.NewStore(redis.Client)
snapHandler := snapshot.NewHandler(snapStore)
hub := ws.NewHub(ws.HubConfig{Logger: logger})
wsHandler := ws.NewHandler(hub)
eventBus := bus.New(bus.Config{
BufferSize: 512,
Persister: snapshot.StorePersister{Store: snapStore},
Broadcaster: ws.HubBroadcaster{Hub: hub},
Logger: logger,
})
beat := heartbeat.New(heartbeat.Config{
Interval: 5 * time.Second,
Emitter: eventBus,
})
collectorGroup, collectorCtx := errgroup.WithContext(ctx)
collectorGroup.Go(func() error { return eventBus.Run(collectorCtx) })
collectorGroup.Go(func() error { return beat.Run(collectorCtx) })
logger.Info("collectors started", "count", 1)
srv := server.New(server.Config{ srv := server.New(server.Config{
ServerConfig: cfg.Server, ServerConfig: cfg.Server,
HealthHandler: healthHandler, HealthHandler: healthHandler,
@ -147,6 +176,12 @@ func run(configPath string) error {
adminOnly := middleware.RequireAdmin adminOnly := middleware.RequireAdmin
router.Route("/v1", func(r chi.Router) { router.Route("/v1", func(r chi.Router) {
r.Get("/healthz", healthHandler.Liveness)
r.Get("/readyz", healthHandler.Readiness)
r.Get("/snapshot", snapHandler.ServeHTTP)
r.Get("/ws", wsHandler.ServeHTTP)
authHandler.RegisterRoutes(r, authenticator) authHandler.RegisterRoutes(r, authenticator)
r.Post("/users", authHandler.Register) r.Post("/users", authHandler.Register)
@ -184,6 +219,12 @@ func run(configPath string) error {
} }
} }
if err := collectorGroup.Wait(); err != nil &&
!errors.Is(err, context.Canceled) &&
!errors.Is(err, context.DeadlineExceeded) {
logger.Error("collector group exit", "error", err)
}
if err := redis.Close(); err != nil { if err := redis.Close(); err != nil {
logger.Error("redis close error", "error", err) logger.Error("redis close error", "error", err)
} }