aiops/backend/controller/alert.go

package controller

import (
	"encoding/json"
	"net/http"
	"strconv"

	"ezops/model"

	"github.com/gin-gonic/gin"
	"gorm.io/gorm"
)

// 告警事件处理 + AlertHandler 定义（规则见 alert_rules.go，渠道见 alert_channels.go） // Logan.Li

// AlertHandler 告警处理器
type AlertHandler struct {
	db *gorm.DB
}

// NewAlertHandler 创建告警处理器
func NewAlertHandler(db *gorm.DB) *AlertHandler {
	return &AlertHandler{db: db}
}

// ListAlerts 获取告警事件列表
// @Summary 获取告警事件列表
// @Description 获取所有告警事件列表
// @Tags 告警
// @Produce json
// @Security Bearer
// @Param page query int false "页码"
// @Param page_size query int false "每页数量"
// @Param status query string false "状态过滤"
// @Param severity query string false "级别过滤"
// @Success 200 {object} model.Response{data=model.ListResponse}
// @Router /alerts [get]
func (h *AlertHandler) ListAlerts(c *gin.Context) {
	var query model.PageQuery
	c.ShouldBindQuery(&query)

	db := h.db.Model(&model.AlertEvent{}).Preload("Rule")

	// 状态过滤
	if status := c.Query("status"); status != "" {
		db = db.Where("status = ?", status)
	}

	// 级别过滤
	if severity := c.Query("severity"); severity != "" {
		db = db.Where("severity = ?", severity)
	}

	var total int64
	db.Count(&total)

	var alerts []model.AlertEvent
	db.Order("created_at DESC").
		Offset(query.GetOffset()).
		Limit(query.GetLimit()).
		Find(&alerts)

	c.JSON(http.StatusOK, model.Response{
		Code:    200,
		Message: "success",
		Data: model.ListResponse{
			List:  alerts,
			Total: total,
		},
	})
}

// AcknowledgeAlert 确认告警
// @Summary 确认告警
// @Description 确认告警事件
// @Tags 告警
// @Accept json
// @Produce json
// @Security Bearer
// @Param id path int true "告警ID"
// @Param request body model.AcknowledgeAlertRequest true "确认请求"
// @Success 200 {object} model.Response{data=model.AlertEvent}
// @Router /alerts/{id}/ack [post]
func (h *AlertHandler) AcknowledgeAlert(c *gin.Context) {
	id, _ := strconv.Atoi(c.Param("id"))

	var req model.AcknowledgeAlertRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(http.StatusBadRequest, model.Response{
			Code:    400,
			Message: "请求参数错误",
		})
		return
	}

	var alert model.AlertEvent
	if err := h.db.First(&alert, id).Error; err != nil {
		c.JSON(http.StatusNotFound, model.Response{
			Code:    404,
			Message: "告警不存在",
		})
		return
	}

	alert.Acknowledged = true
	// alert.AcknowledgedBy = middleware.GetUserID(c)
	// now := time.Now()
	// alert.AcknowledgedAt = &now

	if err := h.db.Save(&alert).Error; err != nil {
		c.JSON(http.StatusInternalServerError, model.Response{
			Code:    500,
			Message: "确认失败",
		})
		return
	}

	c.JSON(http.StatusOK, model.Response{
		Code:    200,
		Message: "确认成功",
		Data:    alert,
	})
}

// toJSON serializes a value to JSON string. Returns empty string on error.
// 包级辅助函数，供 alert_rules.go 等同包文件复用 // Logan.Li
func toJSON(v interface{}) string {
	b, err := json.Marshal(v)
	if err != nil {
		return ""
	}
	return string(b)
}