96 lines
3.0 KiB
Go
96 lines
3.0 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
|
|
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// ActiveAlarmListOptions 定义了查询活跃告警列表时的可选参数
|
|
type ActiveAlarmListOptions struct {
|
|
SourceType *models.AlarmSourceType // 按告警来源类型过滤
|
|
SourceID *uint // 按告警来源ID过滤
|
|
Level *models.SeverityLevel // 按告警严重性等级过滤
|
|
IsIgnored *bool // 按是否被忽略过滤
|
|
TriggerTime *time.Time // 告警触发时间范围 - 开始时间
|
|
EndTime *time.Time // 告警触发时间范围 - 结束时间
|
|
OrderBy string // 排序字段，例如 "trigger_time DESC"
|
|
}
|
|
|
|
// AlarmRepository 定义了对告警模型的数据库操作接口
|
|
type AlarmRepository interface {
|
|
// ListActiveAlarms 支持分页和过滤的活跃告警列表查询。
|
|
// 返回活跃告警列表、总记录数和错误。
|
|
ListActiveAlarms(ctx context.Context, opts ActiveAlarmListOptions, page, pageSize int) ([]models.ActiveAlarm, int64, error)
|
|
}
|
|
|
|
// gormAlarmRepository 是 AlarmRepository 的 GORM 实现。
|
|
type gormAlarmRepository struct {
|
|
ctx context.Context
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewGormAlarmRepository 创建一个新的 AlarmRepository GORM 实现实例。
|
|
func NewGormAlarmRepository(ctx context.Context, db *gorm.DB) AlarmRepository {
|
|
return &gormAlarmRepository{
|
|
ctx: ctx,
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
// ListActiveAlarms 实现了分页和过滤查询活跃告警记录的功能
|
|
func (r *gormAlarmRepository) ListActiveAlarms(ctx context.Context, opts ActiveAlarmListOptions, page, pageSize int) ([]models.ActiveAlarm, int64, error) {
|
|
repoCtx := logs.AddFuncName(ctx, r.ctx, "ListActiveAlarms")
|
|
// --- 校验分页参数 ---
|
|
if page <= 0 || pageSize <= 0 {
|
|
return nil, 0, ErrInvalidPagination // 复用已定义的错误
|
|
}
|
|
|
|
var results []models.ActiveAlarm
|
|
var total int64
|
|
|
|
query := r.db.WithContext(repoCtx).Model(&models.ActiveAlarm{})
|
|
|
|
// --- 应用过滤条件 ---
|
|
if opts.SourceType != nil {
|
|
query = query.Where("source_type = ?", *opts.SourceType)
|
|
}
|
|
if opts.SourceID != nil {
|
|
query = query.Where("source_id = ?", *opts.SourceID)
|
|
}
|
|
if opts.Level != nil {
|
|
query = query.Where("level = ?", *opts.Level)
|
|
}
|
|
if opts.IsIgnored != nil {
|
|
query = query.Where("is_ignored = ?", *opts.IsIgnored)
|
|
}
|
|
if opts.TriggerTime != nil {
|
|
query = query.Where("trigger_time >= ?", *opts.TriggerTime)
|
|
}
|
|
if opts.EndTime != nil {
|
|
query = query.Where("trigger_time <= ?", *opts.EndTime)
|
|
}
|
|
|
|
// --- 计算总数 ---
|
|
if err := query.Count(&total).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
// --- 应用排序条件 ---
|
|
orderBy := "trigger_time DESC" // 默认按触发时间倒序
|
|
if opts.OrderBy != "" {
|
|
orderBy = opts.OrderBy
|
|
}
|
|
query = query.Order(orderBy)
|
|
|
|
// --- 分页 ---
|
|
offset := (page - 1) * pageSize
|
|
err := query.Limit(pageSize).Offset(offset).Find(&results).Error
|
|
|
|
return results, total, err
|
|
}
|