feat(audit): F1+F2 告警静默规则标记完成 + 审计日志CSV导出实现
F1 告警静默规则 [done]: - 后端 alert_silence.go: CRUD + Toggle (onetime/daily/weekly 三模式) - 告警引擎 isSilenced 集成 (alert_engine.go) - 前端 SilenceRuleTab + SilenceDialog - 路由 /alerts/silences 已注册 F2 审计日志导出 [done]: - 重写 ExportAuditLogs: 真正的 CSV 导出 (encoding/csv + UTF-8 BOM) - 支持 module/action/username/startTime/endTime 筛选 - 导出限制 10000 条防溢出,自动记录审计日志 - JSON 格式向后兼容 (format=json) - 前端 exportLogs 传递查询参数 同时包含累积的前期工作: - 目录重构 internal/ -> controller/model/router (v0.55-v0.57) - 后端分层架构 dto/repository/service/utils - 审计模块全功能 (终端审计/操作审计/登录日志) - asciicast v2 录制格式 - SFTP 文件浏览器 - StatStrip 紧凑统计条 - 在线会话自动刷新 - LiveMonitor WebSocket 诊断修复 验证: go build ✅ | npm run build ✅ | Docker 部署 ✅ | 10 回归端点 ✅ | CSV 导出 ✅ | 静默 CRUD ✅
This commit is contained in:
1 parent
e57718998a
commit
77c234b060
314 files changed
+10050
-4065
No files matched your search
@@ -0,0 +1,130 @@
|
||||
// 终端审计实时流 Hub — 在线会话内容广播给审计查看者 // Logan.Li
|
||||
package controller
|
||||
|
||||
import (
|
||||
"log"
|
||||
"net/http"
|
||||
"sync"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/gorilla/websocket"
|
||||
)
|
||||
|
||||
// AuditHub 终端审计流中心 — 管理所有活跃会话的审计订阅者
|
||||
type AuditHub struct {
|
||||
mu sync.RWMutex
|
||||
watchers map[uint]map[chan []byte]struct{} // sessionID -> set of watcher channels
|
||||
}
|
||||
|
||||
// NewAuditHub 创建审计 Hub
|
||||
func NewAuditHub() *AuditHub {
|
||||
return &AuditHub{
|
||||
watchers: make(map[uint]map[chan []byte]struct{}),
|
||||
}
|
||||
}
|
||||
|
||||
// Subscribe 订阅某个会话的实时输出
|
||||
func (h *AuditHub) Subscribe(sessionID uint) chan []byte {
|
||||
ch := make(chan []byte, 256)
|
||||
h.mu.Lock()
|
||||
defer h.mu.Unlock()
|
||||
if h.watchers[sessionID] == nil {
|
||||
h.watchers[sessionID] = make(map[chan []byte]struct{})
|
||||
}
|
||||
h.watchers[sessionID][ch] = struct{}{}
|
||||
return ch
|
||||
}
|
||||
|
||||
// Unsubscribe 取消订阅
|
||||
func (h *AuditHub) Unsubscribe(sessionID uint, ch chan []byte) {
|
||||
h.mu.Lock()
|
||||
defer h.mu.Unlock()
|
||||
if watchers, ok := h.watchers[sessionID]; ok {
|
||||
delete(watchers, ch)
|
||||
close(ch)
|
||||
if len(watchers) == 0 {
|
||||
delete(h.watchers, sessionID)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Broadcast 向所有订阅者广播终端输出数据(非阻塞)
|
||||
func (h *AuditHub) Broadcast(sessionID uint, data []byte) {
|
||||
h.mu.RLock()
|
||||
watchers, ok := h.watchers[sessionID]
|
||||
h.mu.RUnlock()
|
||||
if !ok || len(watchers) == 0 {
|
||||
return
|
||||
}
|
||||
for ch := range watchers {
|
||||
select {
|
||||
case ch <- data:
|
||||
default:
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Cleanup 清理某个会话的所有订阅者
|
||||
func (h *AuditHub) Cleanup(sessionID uint) {
|
||||
h.mu.Lock()
|
||||
defer h.mu.Unlock()
|
||||
if watchers, ok := h.watchers[sessionID]; ok {
|
||||
for ch := range watchers {
|
||||
close(ch)
|
||||
}
|
||||
delete(h.watchers, sessionID)
|
||||
}
|
||||
}
|
||||
|
||||
var auditUpgrader = websocket.Upgrader{
|
||||
CheckOrigin: func(r *http.Request) bool { return true },
|
||||
ReadBufferSize: 1024,
|
||||
WriteBufferSize: 1024,
|
||||
}
|
||||
|
||||
// ServeAuditStream 处理 /ws/audit/:sessionId WebSocket — 实时推送终端输出
|
||||
func (h *AuditHub) ServeAuditStream(c *gin.Context, sessionID uint, historyData string, sessionInfo map[string]interface{}) {
|
||||
conn, err := auditUpgrader.Upgrade(c.Writer, c.Request, nil)
|
||||
if err != nil {
|
||||
log.Printf("[AuditStream] WS 升级失败 session=%d: %v", sessionID, err)
|
||||
return
|
||||
}
|
||||
defer conn.Close()
|
||||
|
||||
// 发送会话元信息
|
||||
conn.WriteJSON(map[string]interface{}{
|
||||
"type": "session_info",
|
||||
"payload": sessionInfo,
|
||||
})
|
||||
|
||||
// 发送历史数据(已录制的输出)
|
||||
if historyData != "" {
|
||||
conn.WriteJSON(map[string]interface{}{
|
||||
"type": "history",
|
||||
"data": historyData,
|
||||
})
|
||||
}
|
||||
|
||||
// 订阅实时流
|
||||
ch := h.Subscribe(sessionID)
|
||||
defer h.Unsubscribe(sessionID, ch)
|
||||
|
||||
// goroutine 读取关闭信号
|
||||
go func() {
|
||||
for {
|
||||
if _, _, err := conn.ReadMessage(); err != nil {
|
||||
break
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
// 推送实时数据
|
||||
for data := range ch {
|
||||
if err := conn.WriteJSON(map[string]interface{}{
|
||||
"type": "output",
|
||||
"data": string(data),
|
||||
}); err != nil {
|
||||
break
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user