package alarm

import (
	"context"
	"github.com/gogf/gf/v2/frame/g"
	"time"
)

// 报警列表
func (h Alarm) HatAlarmList(ctx context.Context, req *HatAlarmListReq) (res *HatAlarmListRes, err error) {
	res = new(HatAlarmListRes)
	offset := (req.Page - 1) * req.PageSize
	var hatAlarms []HatAlarm
	model := g.Model("hat_alarm")
	count, err := model.Ctx(ctx).Count()
	if err != nil {
		return nil, err
	}
	model.Where("project_id = ?", req.ProjectId)
	err = model.Offset(int(offset)).Limit(int(req.PageSize)).Scan(&hatAlarms)
	if err != nil {
		return nil, err
	}
	res.HatAlarms = hatAlarms
	res.Total = int64(count)
	return res, nil
}

// 处理报警
func (h Alarm) HandleAlarm(ctx context.Context, req *HandleAlarmReq) (res *HandleAlarmRes, err error) {
	res = new(HandleAlarmRes)
	data := g.Map{
		"is_handle": 1,
		"handle_at": time.Now(),
	}
	g.Model("hat_alarm").Ctx(ctx).Data(data).Where("id", req.Id).Update()
	return res, nil
}