forked from iti-framework/iTi-Flask
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
from sqlalchemy.orm import noload
|
|
from sqlalchemy.sql._typing import ColumnExpressionArgument
|
|
from typing import List, Dict, Any, Optional, Set
|
|
from iti.applications.extensions import db
|
|
from flask import current_app
|
|
from iti.applications.models import (
|
|
IotAlertLog,
|
|
IotAlertRule,
|
|
IotAlertPush,
|
|
IotEndpoint,
|
|
)
|
|
from sqlalchemy import select, distinct
|
|
|
|
def add_endpoint_alert_log(endpoint_id: int):
|
|
"""
|
|
添加采集端告警日志
|
|
Args:
|
|
endpoint_id: 采集端ID
|
|
"""
|
|
endpoint = db.session.scalar(select(IotEndpoint).filter_by(id=endpoint_id))
|
|
if not endpoint:
|
|
return "采集端不存在"
|
|
alert_tag = f"ep{endpoint.id}_nd0"
|
|
alert_log = db.session.scalar(select(IotAlertLog).filter_by(alert_tag=alert_tag))
|
|
if alert_log:
|
|
alert_log.trigger_count += 1
|
|
if alert_log.trigger_count >= 3:
|
|
alert_log.status = 1
|
|
push_alert(alert_log)
|
|
else:
|
|
dict_data = dict(
|
|
alert_tag=alert_tag,
|
|
alert_target_name=f"{endpoint.endpoint_name}({endpoint.endpoint_number})",
|
|
alert_level=1,
|
|
status=0,
|
|
trigger_count=1,
|
|
alert_content=f"采集端 {endpoint.endpoint_name}({endpoint.endpoint_number}) 网络不可达",
|
|
)
|
|
alert_log = IotAlertLog(**dict_data)
|
|
|
|
db.session.add(alert_log)
|
|
db.session.commit()
|
|
return ""
|
|
|
|
|
|
def push_alert(alert_log: IotAlertLog):
|
|
"""
|
|
推送告警消息
|
|
Args:
|
|
alert_log: 告警日志
|
|
"""
|
|
|
|
level = f"{alert_log.alert_level}"
|
|
alert_push_list = db.session.scalars(select(IotAlertPush).filter(IotAlertPush.alert_level.contains(level))).all()
|
|
if alert_push_list:
|
|
for alert_push in alert_push_list:
|
|
# TODO 调用地址发送告警消息
|
|
current_app.logger.info(f"推送告警消息到 {alert_push.push_url}，内容：{alert_log.alert_content}") |