feat(audit): Live-Stream im Dashboard via WebSocket
Recent-Activity-Karte zeigt neue audit_log-Events jetzt sofort statt
in 15s-Polls.
internal/services/audit/audit.go:
- Repo bekommt Subscribe()-Methode mit fan-out-channel (Buffer 32,
non-blocking-send — langsame Clients droppen Events statt die
Pipeline zu blockieren).
- Log() macht jetzt INSERT … RETURNING id, created_at und broadcastet
den fertigen Entry an alle Subscribers. Broadcast nur nach
erfolgreichem INSERT — failed inserts erscheinen nicht.
internal/handlers/audit.go:
- Neuer GET /api/v1/audit/live (WebSocket): sendet beim Connect die
letzten 50 Einträge (oldest→newest), danach Live-Stream aus
Subscribe-Channel. 30s-Ping gegen HAProxy-Tunnel-Timeout.
- Recent (Poll-Endpoint) bleibt für Fallbacks erhalten.
UI Dashboard:
- useAuditLive(keep=15)-Hook ersetzt das 15s-useQuery-Poll.
- WebSocket auf wss://<host>/api/v1/audit/live; Auto-Reconnect alle
2s nach Drop.
- dedupe per id (Snapshot + erste live-Events können sich kurz
überschneiden während des Subscribe-Race).
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -6,6 +6,7 @@ package audit
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/jackc/pgx/v5/pgxpool"
|
||||
@@ -13,9 +14,48 @@ import (
|
||||
|
||||
type Repo struct {
|
||||
Pool *pgxpool.Pool
|
||||
|
||||
subsMu sync.RWMutex
|
||||
subs map[chan Entry]struct{}
|
||||
}
|
||||
|
||||
func New(pool *pgxpool.Pool) *Repo { return &Repo{Pool: pool} }
|
||||
func New(pool *pgxpool.Pool) *Repo {
|
||||
return &Repo{Pool: pool, subs: map[chan Entry]struct{}{}}
|
||||
}
|
||||
|
||||
// Subscribe gibt einen Channel für Live-Audit-Events zurück + ein
|
||||
// Unsubscribe-Cleanup. Channel-Buffer 32 — bei stehenden Clients
|
||||
// werden Events gedropt (non-blocking-send).
|
||||
func (r *Repo) Subscribe() (<-chan Entry, func()) {
|
||||
c := make(chan Entry, 32)
|
||||
r.subsMu.Lock()
|
||||
if r.subs == nil {
|
||||
r.subs = map[chan Entry]struct{}{}
|
||||
}
|
||||
r.subs[c] = struct{}{}
|
||||
r.subsMu.Unlock()
|
||||
return c, func() {
|
||||
r.subsMu.Lock()
|
||||
delete(r.subs, c)
|
||||
r.subsMu.Unlock()
|
||||
close(c)
|
||||
}
|
||||
}
|
||||
|
||||
func (r *Repo) broadcast(e Entry) {
|
||||
r.subsMu.RLock()
|
||||
subs := make([]chan Entry, 0, len(r.subs))
|
||||
for c := range r.subs {
|
||||
subs = append(subs, c)
|
||||
}
|
||||
r.subsMu.RUnlock()
|
||||
for _, c := range subs {
|
||||
select {
|
||||
case c <- e:
|
||||
default:
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Entry mirrors one audit_log row — ListRecent returns these for
|
||||
// the dashboard's recent-activity card.
|
||||
@@ -82,9 +122,34 @@ func (r *Repo) Log(ctx context.Context, actor, action, subject string, detail an
|
||||
if nodeID == "" {
|
||||
nodeArg = nil
|
||||
}
|
||||
_, err := r.Pool.Exec(ctx,
|
||||
`INSERT INTO audit_log (actor, action, subject, detail, node_id)
|
||||
VALUES ($1, $2, $3, $4, $5)`,
|
||||
actor, action, subjectArg, detailJSON, nodeArg)
|
||||
return err
|
||||
// RETURNING id+created_at damit der Subscribe-Channel direkt einen
|
||||
// vollständigen Entry verteilen kann — Subscriber müssen nicht
|
||||
// erneut die DB hitten für die Anzeige.
|
||||
var e Entry
|
||||
e.Actor = actor
|
||||
e.Action = action
|
||||
if subject != "" {
|
||||
s := subject
|
||||
e.Subject = &s
|
||||
}
|
||||
if len(detailJSON) > 0 {
|
||||
e.Detail = json.RawMessage(detailJSON)
|
||||
}
|
||||
if nodeID != "" {
|
||||
n := nodeID
|
||||
e.NodeID = &n
|
||||
}
|
||||
err := r.Pool.QueryRow(ctx, `
|
||||
INSERT INTO audit_log (actor, action, subject, detail, node_id)
|
||||
VALUES ($1, $2, $3, $4, $5)
|
||||
RETURNING id, created_at`,
|
||||
actor, action, subjectArg, detailJSON, nodeArg).
|
||||
Scan(&e.ID, &e.CreatedAt)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
// Broadcast nach erfolgreichem INSERT — wenn DB ablehnt, sollen
|
||||
// Subscribers das Event auch nicht sehen.
|
||||
r.broadcast(e)
|
||||
return nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user