diff --git a/internal/infra/repository/alarm_repository.go b/internal/infra/repository/alarm_repository.go index 4a1893d..c8ec5ee 100644 --- a/internal/infra/repository/alarm_repository.go +++ b/internal/infra/repository/alarm_repository.go @@ -2,12 +2,30 @@ 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 实现。 @@ -23,3 +41,55 @@ func NewGormAlarmRepository(ctx context.Context, db *gorm.DB) AlarmRepository { 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 +}