mirror of
https://gitee.com/XM-GO/PandaX.git
synced 2026-04-23 02:48:34 +08:00
105 lines
3.0 KiB
Go
105 lines
3.0 KiB
Go
package services
|
|
|
|
import (
|
|
"github.com/PandaXGO/PandaKit/biz"
|
|
"pandax/apps/device/entity"
|
|
"pandax/pkg/global"
|
|
"pandax/pkg/tool"
|
|
)
|
|
|
|
type (
|
|
DeviceAlarmModel interface {
|
|
Insert(data entity.DeviceAlarm) error
|
|
FindOne(id string) *entity.DeviceAlarm
|
|
FindOneByType(deviceId, ty, state string) *entity.DeviceAlarm
|
|
FindListPage(page, pageSize int, data entity.DeviceAlarmForm) (*[]entity.DeviceAlarm, int64)
|
|
Update(data entity.DeviceAlarm) error
|
|
Delete(ids []string)
|
|
FindAlarmCount() entity.DeviceCount
|
|
}
|
|
|
|
alarmModelImpl struct {
|
|
table string
|
|
}
|
|
)
|
|
|
|
var DeviceAlarmModelDao DeviceAlarmModel = &alarmModelImpl{
|
|
table: `device_alarms`,
|
|
}
|
|
|
|
func (m *alarmModelImpl) Insert(data entity.DeviceAlarm) error {
|
|
err := global.Db.Table(m.table).Create(&data).Error
|
|
return err
|
|
}
|
|
|
|
func (m *alarmModelImpl) FindOne(id string) *entity.DeviceAlarm {
|
|
resData := new(entity.DeviceAlarm)
|
|
db := global.Db.Table(m.table).Where("id = ?", id)
|
|
err := db.First(resData).Error
|
|
biz.ErrIsNil(err, "查询设备告警失败")
|
|
return resData
|
|
}
|
|
|
|
func (m *alarmModelImpl) FindOneByType(deviceId, ty, state string) *entity.DeviceAlarm {
|
|
resData := new(entity.DeviceAlarm)
|
|
db := global.Db.Table(m.table).Where("device_id = ?", deviceId).Where("type = ? ", ty).Where("state = ? ", state)
|
|
err := db.First(resData).Error
|
|
biz.ErrIsNil(err, "查询设备告警失败")
|
|
return resData
|
|
}
|
|
|
|
func (m *alarmModelImpl) FindListPage(page, pageSize int, data entity.DeviceAlarmForm) (*[]entity.DeviceAlarm, int64) {
|
|
list := make([]entity.DeviceAlarm, 0)
|
|
var total int64 = 0
|
|
offset := pageSize * (page - 1)
|
|
db := global.Db.Table(m.table)
|
|
if data.DeviceId != "" {
|
|
db = db.Where("device_id = ?", data.DeviceId)
|
|
}
|
|
if data.ProductId != "" {
|
|
db = db.Where("product_id = ?", data.ProductId)
|
|
}
|
|
if data.Level != "" {
|
|
db = db.Where("level = ?", data.Level)
|
|
}
|
|
if data.Type != "" {
|
|
db = db.Where("type like ?", "%"+data.Type+"%")
|
|
}
|
|
if data.State != "" {
|
|
db = db.Where("state = ?", data.State)
|
|
}
|
|
if data.StartTime != "" {
|
|
db = db.Where("time > ?", data.StartTime)
|
|
}
|
|
if data.EndTime != "" {
|
|
db = db.Where("time < ?", data.EndTime)
|
|
}
|
|
// 组织数据访问权限
|
|
tool.OrgAuthSet(db, data.RoleId, data.Owner)
|
|
|
|
err := db.Count(&total).Error
|
|
err = db.Order("time").Limit(pageSize).Offset(offset).Find(&list).Error
|
|
biz.ErrIsNil(err, "查询设备告警分页列表失败")
|
|
return &list, total
|
|
}
|
|
|
|
func (m *alarmModelImpl) Update(data entity.DeviceAlarm) error {
|
|
err := global.Db.Table(m.table).
|
|
Where("type = ?", data.Type).
|
|
Where("device_id = ?", data.DeviceId).
|
|
Updates(&data).Error
|
|
return err
|
|
}
|
|
|
|
func (m *alarmModelImpl) Delete(id []string) {
|
|
biz.ErrIsNil(global.Db.Table(m.table).Delete(&entity.DeviceAlarm{}, "id in (?)", id).Error, "删除设备告警失败")
|
|
}
|
|
|
|
// 获取告警数量统计
|
|
func (m *alarmModelImpl) FindAlarmCount() (count entity.DeviceCount) {
|
|
sql := `SELECT COUNT(*) AS total, (SELECT COUNT(*) FROM device_alarms WHERE DATE(time) = CURDATE()) AS today FROM device_alarms`
|
|
err := global.Db.Raw(sql).Scan(&count).Error
|
|
biz.ErrIsNil(err, "获取告警统计总数失败")
|
|
return
|
|
}
|