gormAlarmRepository.ListActiveAlarms

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
+}