55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
from flask import Flask, request, jsonify
|
|
import pymysql
|
|
|
|
app = Flask(__name__)
|
|
|
|
def get_db():
|
|
return pymysql.connect(
|
|
host="127.0.0.1",
|
|
user="root",
|
|
password="hp93000",
|
|
database="alert_mail_stat",
|
|
charset="utf8mb4"
|
|
)
|
|
|
|
@app.route("/alert", methods=["POST"])
|
|
def alert_receive():
|
|
try:
|
|
data = request.get_json()
|
|
alerts = data.get("alerts", [])
|
|
insert_num = 0
|
|
update_num = 0
|
|
db = get_db()
|
|
cur = db.cursor()
|
|
|
|
for alert in alerts:
|
|
fp = alert["fingerprint"]
|
|
alert_name = alert["labels"].get("alertname", "")
|
|
instance = alert["labels"].get("instance", "")
|
|
severity = alert["labels"].get("severity", "warning")
|
|
status = alert["status"]
|
|
start_at = alert.get("startsAt", "")
|
|
end_at = alert.get("endsAt")
|
|
alert_type = 2 if status == "resolved" else 1
|
|
content_str = str(alert)
|
|
|
|
sql = """
|
|
INSERT INTO alert_log(mail_uid,alert_type,alert_name,instance,severity,starts_at,ends_at,content,receive_time)
|
|
VALUES(%s,%s,%s,%s,%s,%s,%s,%s,NOW())
|
|
"""
|
|
cur.execute(sql, (fp, alert_type, alert_name, instance, severity, start_at, end_at, content_str))
|
|
insert_num += 1
|
|
|
|
db.commit()
|
|
cur.close()
|
|
db.close()
|
|
return jsonify({"code": 200, "insert": insert_num, "update": update_num})
|
|
|
|
except Exception as e:
|
|
if "db" in locals():
|
|
db.rollback()
|
|
db.close()
|
|
return jsonify({"code": 500, "msg": str(e)}), 200
|
|
|
|
if __name__ == "__main__":
|
|
app.run(host="0.0.0.0", port=909, debug=False) |