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,120 @@
|
||||
package controller
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"log"
|
||||
"sync"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/gorilla/websocket"
|
||||
)
|
||||
|
||||
// MetricsUpdate 单个主机的指标更新 // Logan.Li
|
||||
type MetricsUpdate struct {
|
||||
Type string `json:"type"` // "metrics_update"
|
||||
Hostname string `json:"hostname"`
|
||||
CPU float64 `json:"cpu_usage"`
|
||||
Memory float64 `json:"memory_usage"`
|
||||
Disk float64 `json:"disk_usage"`
|
||||
Status string `json:"status"`
|
||||
TCPEstablished int `json:"tcp_established,omitempty"`
|
||||
TCPTimeWait int `json:"tcp_time_wait,omitempty"`
|
||||
NetRxBPS float64 `json:"net_rx_bps,omitempty"`
|
||||
NetTxBPS float64 `json:"net_tx_bps,omitempty"`
|
||||
DiskReadBPS float64 `json:"disk_read_bps,omitempty"`
|
||||
DiskWriteBPS float64 `json:"disk_write_bps,omitempty"`
|
||||
}
|
||||
|
||||
// MetricsHub 管理所有 WebSocket 客户端连接,广播指标更新 // Logan.Li
|
||||
type MetricsHub struct {
|
||||
mu sync.RWMutex
|
||||
clients map[*websocket.Conn]bool
|
||||
}
|
||||
|
||||
// NewMetricsHub 创建 MetricsHub 实例
|
||||
func NewMetricsHub() *MetricsHub {
|
||||
return &MetricsHub{
|
||||
clients: make(map[*websocket.Conn]bool),
|
||||
}
|
||||
}
|
||||
|
||||
// Register 注册新的 WebSocket 连接
|
||||
func (h *MetricsHub) Register(conn *websocket.Conn) {
|
||||
h.mu.Lock()
|
||||
h.clients[conn] = true
|
||||
h.mu.Unlock()
|
||||
log.Printf("[MetricsWS] 客户端已连接,当前连接数: %d", h.Count())
|
||||
}
|
||||
|
||||
// Unregister 移除断开的 WebSocket 连接
|
||||
func (h *MetricsHub) Unregister(conn *websocket.Conn) {
|
||||
h.mu.Lock()
|
||||
delete(h.clients, conn)
|
||||
h.mu.Unlock()
|
||||
log.Printf("[MetricsWS] 客户端已断开,当前连接数: %d", h.Count())
|
||||
}
|
||||
|
||||
// Count 返回当前连接数
|
||||
func (h *MetricsHub) Count() int {
|
||||
h.mu.RLock()
|
||||
defer h.mu.RUnlock()
|
||||
return len(h.clients)
|
||||
}
|
||||
|
||||
// Broadcast 向所有连接的客户端广播指标更新
|
||||
func (h *MetricsHub) Broadcast(update MetricsUpdate) {
|
||||
data, err := json.Marshal(update)
|
||||
if err != nil {
|
||||
log.Printf("[MetricsWS] JSON 序列化失败: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
h.mu.RLock()
|
||||
defer h.mu.RUnlock()
|
||||
|
||||
for conn := range h.clients {
|
||||
if err := conn.WriteMessage(websocket.TextMessage, data); err != nil {
|
||||
log.Printf("[MetricsWS] 发送失败: %v", err)
|
||||
go func(c *websocket.Conn) {
|
||||
h.Unregister(c)
|
||||
c.Close()
|
||||
}(conn)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// MetricsWSHandler 处理 /ws/metrics 的 WebSocket 连接 // Logan.Li
|
||||
type MetricsWSHandler struct {
|
||||
hub *MetricsHub
|
||||
}
|
||||
|
||||
// NewMetricsWSHandler 创建 WebSocket metrics 处理器
|
||||
func NewMetricsWSHandler(hub *MetricsHub) *MetricsWSHandler {
|
||||
return &MetricsWSHandler{hub: hub}
|
||||
}
|
||||
|
||||
// ServeWS 升级 HTTP → WebSocket,注册到 Hub,保持连接
|
||||
func (h *MetricsWSHandler) ServeWS(c *gin.Context) {
|
||||
conn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
|
||||
if err != nil {
|
||||
log.Printf("[MetricsWS] WebSocket 升级失败: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
h.hub.Register(conn)
|
||||
defer func() {
|
||||
h.hub.Unregister(conn)
|
||||
conn.Close()
|
||||
}()
|
||||
|
||||
// 保持连接,读取消息(客户端可能发 ping)
|
||||
for {
|
||||
_, _, err := conn.ReadMessage()
|
||||
if err != nil {
|
||||
if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseNormalClosure) {
|
||||
log.Printf("[MetricsWS] 连接异常关闭: %v", err)
|
||||
}
|
||||
break
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user